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 818c7501 2019-07-11 stsp #include <ctype.h>
29 99437157 2018-11-11 stsp #include <signal.h>
30 5c860e29 2018-03-12 stsp #include <stdio.h>
31 5c860e29 2018-03-12 stsp #include <stdlib.h>
32 5c860e29 2018-03-12 stsp #include <string.h>
33 5c860e29 2018-03-12 stsp #include <unistd.h>
34 c09a553d 2018-03-12 stsp #include <libgen.h>
35 c0768b0f 2018-06-10 stsp #include <time.h>
36 33ad4cbe 2019-05-12 jcs #include <paths.h>
37 5c860e29 2018-03-12 stsp
38 f42b1b34 2018-03-12 stsp #include "got_error.h"
39 f42b1b34 2018-03-12 stsp #include "got_object.h"
40 5261c201 2018-04-01 stsp #include "got_reference.h"
41 f42b1b34 2018-03-12 stsp #include "got_repository.h"
42 1dd54920 2019-05-11 stsp #include "got_path.h"
43 c09a553d 2018-03-12 stsp #include "got_worktree.h"
44 79109fed 2018-03-27 stsp #include "got_diff.h"
45 372ccdbb 2018-06-10 stsp #include "got_commit_graph.h"
46 404c43c4 2018-06-21 stsp #include "got_blame.h"
47 63219cd2 2019-01-04 stsp #include "got_privsep.h"
48 793c30b5 2019-05-13 stsp #include "got_opentemp.h"
49 5c860e29 2018-03-12 stsp
50 5c860e29 2018-03-12 stsp #ifndef nitems
51 5c860e29 2018-03-12 stsp #define nitems(_a) (sizeof((_a)) / sizeof((_a)[0]))
52 5c860e29 2018-03-12 stsp #endif
53 99437157 2018-11-11 stsp
54 99437157 2018-11-11 stsp static volatile sig_atomic_t sigint_received;
55 99437157 2018-11-11 stsp static volatile sig_atomic_t sigpipe_received;
56 99437157 2018-11-11 stsp
57 99437157 2018-11-11 stsp static void
58 99437157 2018-11-11 stsp catch_sigint(int signo)
59 99437157 2018-11-11 stsp {
60 99437157 2018-11-11 stsp sigint_received = 1;
61 99437157 2018-11-11 stsp }
62 99437157 2018-11-11 stsp
63 99437157 2018-11-11 stsp static void
64 99437157 2018-11-11 stsp catch_sigpipe(int signo)
65 99437157 2018-11-11 stsp {
66 99437157 2018-11-11 stsp sigpipe_received = 1;
67 99437157 2018-11-11 stsp }
68 5c860e29 2018-03-12 stsp
69 99437157 2018-11-11 stsp
70 8cfb4057 2019-07-09 stsp struct got_cmd {
71 5c860e29 2018-03-12 stsp const char *cmd_name;
72 d7d4f210 2018-03-12 stsp const struct got_error *(*cmd_main)(int, char *[]);
73 1b6b95a8 2018-03-12 stsp void (*cmd_usage)(void);
74 97b3a7be 2019-07-09 stsp const char *cmd_alias;
75 5c860e29 2018-03-12 stsp };
76 5c860e29 2018-03-12 stsp
77 ce5b7c56 2019-07-09 stsp __dead static void usage(int);
78 2c7829a4 2019-06-17 stsp __dead static void usage_init(void);
79 3ce1b845 2019-07-15 stsp __dead static void usage_import(void);
80 4ed7e80c 2018-05-20 stsp __dead static void usage_checkout(void);
81 507dc3bb 2018-12-29 stsp __dead static void usage_update(void);
82 4ed7e80c 2018-05-20 stsp __dead static void usage_log(void);
83 4ed7e80c 2018-05-20 stsp __dead static void usage_diff(void);
84 404c43c4 2018-06-21 stsp __dead static void usage_blame(void);
85 5de5890b 2018-10-18 stsp __dead static void usage_tree(void);
86 6bad629b 2019-02-04 stsp __dead static void usage_status(void);
87 d0eebce4 2019-03-11 stsp __dead static void usage_ref(void);
88 4e759de4 2019-06-26 stsp __dead static void usage_branch(void);
89 d00136be 2019-03-26 stsp __dead static void usage_add(void);
90 648e4ef7 2019-07-09 stsp __dead static void usage_remove(void);
91 a129376b 2019-03-28 stsp __dead static void usage_revert(void);
92 c4296144 2019-05-09 stsp __dead static void usage_commit(void);
93 234035bc 2019-06-01 stsp __dead static void usage_cherrypick(void);
94 5ef14e63 2019-06-02 stsp __dead static void usage_backout(void);
95 818c7501 2019-07-11 stsp __dead static void usage_rebase(void);
96 5c860e29 2018-03-12 stsp
97 2c7829a4 2019-06-17 stsp static const struct got_error* cmd_init(int, char *[]);
98 3ce1b845 2019-07-15 stsp static const struct got_error* cmd_import(int, char *[]);
99 4ed7e80c 2018-05-20 stsp static const struct got_error* cmd_checkout(int, char *[]);
100 507dc3bb 2018-12-29 stsp static const struct got_error* cmd_update(int, char *[]);
101 4ed7e80c 2018-05-20 stsp static const struct got_error* cmd_log(int, char *[]);
102 4ed7e80c 2018-05-20 stsp static const struct got_error* cmd_diff(int, char *[]);
103 404c43c4 2018-06-21 stsp static const struct got_error* cmd_blame(int, char *[]);
104 5de5890b 2018-10-18 stsp static const struct got_error* cmd_tree(int, char *[]);
105 4ed7e80c 2018-05-20 stsp static const struct got_error* cmd_status(int, char *[]);
106 d0eebce4 2019-03-11 stsp static const struct got_error* cmd_ref(int, char *[]);
107 4e759de4 2019-06-26 stsp static const struct got_error* cmd_branch(int, char *[]);
108 d00136be 2019-03-26 stsp static const struct got_error* cmd_add(int, char *[]);
109 648e4ef7 2019-07-09 stsp static const struct got_error* cmd_remove(int, char *[]);
110 a129376b 2019-03-28 stsp static const struct got_error* cmd_revert(int, char *[]);
111 c4296144 2019-05-09 stsp static const struct got_error* cmd_commit(int, char *[]);
112 234035bc 2019-06-01 stsp static const struct got_error* cmd_cherrypick(int, char *[]);
113 5ef14e63 2019-06-02 stsp static const struct got_error* cmd_backout(int, char *[]);
114 818c7501 2019-07-11 stsp static const struct got_error* cmd_rebase(int, char *[]);
115 5c860e29 2018-03-12 stsp
116 8cfb4057 2019-07-09 stsp static struct got_cmd got_commands[] = {
117 97b3a7be 2019-07-09 stsp { "init", cmd_init, usage_init, "" },
118 3ce1b845 2019-07-15 stsp { "import", cmd_import, usage_import, "" },
119 97b3a7be 2019-07-09 stsp { "checkout", cmd_checkout, usage_checkout, "co" },
120 97b3a7be 2019-07-09 stsp { "update", cmd_update, usage_update, "up" },
121 97b3a7be 2019-07-09 stsp { "log", cmd_log, usage_log, "" },
122 97b3a7be 2019-07-09 stsp { "diff", cmd_diff, usage_diff, "" },
123 97b3a7be 2019-07-09 stsp { "blame", cmd_blame, usage_blame, "" },
124 97b3a7be 2019-07-09 stsp { "tree", cmd_tree, usage_tree, "" },
125 97b3a7be 2019-07-09 stsp { "status", cmd_status, usage_status, "st" },
126 97b3a7be 2019-07-09 stsp { "ref", cmd_ref, usage_ref, "" },
127 97b3a7be 2019-07-09 stsp { "branch", cmd_branch, usage_branch, "br" },
128 97b3a7be 2019-07-09 stsp { "add", cmd_add, usage_add, "" },
129 648e4ef7 2019-07-09 stsp { "remove", cmd_remove, usage_remove, "rm" },
130 97b3a7be 2019-07-09 stsp { "revert", cmd_revert, usage_revert, "rv" },
131 97b3a7be 2019-07-09 stsp { "commit", cmd_commit, usage_commit, "ci" },
132 016477fd 2019-07-09 stsp { "cherrypick", cmd_cherrypick, usage_cherrypick, "cy" },
133 97b3a7be 2019-07-09 stsp { "backout", cmd_backout, usage_backout, "bo" },
134 818c7501 2019-07-11 stsp { "rebase", cmd_rebase, usage_rebase, "rb" },
135 5c860e29 2018-03-12 stsp };
136 ce5b7c56 2019-07-09 stsp
137 ce5b7c56 2019-07-09 stsp static void
138 ce5b7c56 2019-07-09 stsp list_commands(void)
139 ce5b7c56 2019-07-09 stsp {
140 ce5b7c56 2019-07-09 stsp int i;
141 ce5b7c56 2019-07-09 stsp
142 ce5b7c56 2019-07-09 stsp fprintf(stderr, "commands:");
143 ce5b7c56 2019-07-09 stsp for (i = 0; i < nitems(got_commands); i++) {
144 ce5b7c56 2019-07-09 stsp struct got_cmd *cmd = &got_commands[i];
145 ce5b7c56 2019-07-09 stsp fprintf(stderr, " %s", cmd->cmd_name);
146 ce5b7c56 2019-07-09 stsp }
147 ce5b7c56 2019-07-09 stsp fputc('\n', stderr);
148 ce5b7c56 2019-07-09 stsp }
149 5c860e29 2018-03-12 stsp
150 5c860e29 2018-03-12 stsp int
151 5c860e29 2018-03-12 stsp main(int argc, char *argv[])
152 5c860e29 2018-03-12 stsp {
153 8cfb4057 2019-07-09 stsp struct got_cmd *cmd;
154 5c860e29 2018-03-12 stsp unsigned int i;
155 5c860e29 2018-03-12 stsp int ch;
156 1b6b95a8 2018-03-12 stsp int hflag = 0;
157 5c860e29 2018-03-12 stsp
158 289e3cbf 2019-02-04 stsp setlocale(LC_CTYPE, "");
159 5c860e29 2018-03-12 stsp
160 1b6b95a8 2018-03-12 stsp while ((ch = getopt(argc, argv, "h")) != -1) {
161 5c860e29 2018-03-12 stsp switch (ch) {
162 1b6b95a8 2018-03-12 stsp case 'h':
163 1b6b95a8 2018-03-12 stsp hflag = 1;
164 1b6b95a8 2018-03-12 stsp break;
165 5c860e29 2018-03-12 stsp default:
166 ce5b7c56 2019-07-09 stsp usage(hflag);
167 5c860e29 2018-03-12 stsp /* NOTREACHED */
168 5c860e29 2018-03-12 stsp }
169 5c860e29 2018-03-12 stsp }
170 5c860e29 2018-03-12 stsp
171 5c860e29 2018-03-12 stsp argc -= optind;
172 5c860e29 2018-03-12 stsp argv += optind;
173 1e70621d 2018-03-27 stsp optind = 0;
174 5c860e29 2018-03-12 stsp
175 5c860e29 2018-03-12 stsp if (argc <= 0)
176 ce5b7c56 2019-07-09 stsp usage(hflag);
177 5c860e29 2018-03-12 stsp
178 99437157 2018-11-11 stsp signal(SIGINT, catch_sigint);
179 99437157 2018-11-11 stsp signal(SIGPIPE, catch_sigpipe);
180 99437157 2018-11-11 stsp
181 5c860e29 2018-03-12 stsp for (i = 0; i < nitems(got_commands); i++) {
182 d7d4f210 2018-03-12 stsp const struct got_error *error;
183 d7d4f210 2018-03-12 stsp
184 5c860e29 2018-03-12 stsp cmd = &got_commands[i];
185 5c860e29 2018-03-12 stsp
186 97b3a7be 2019-07-09 stsp if (strcmp(cmd->cmd_name, argv[0]) != 0 &&
187 97b3a7be 2019-07-09 stsp strcmp(cmd->cmd_alias, argv[0]) != 0)
188 5c860e29 2018-03-12 stsp continue;
189 5c860e29 2018-03-12 stsp
190 1b6b95a8 2018-03-12 stsp if (hflag)
191 1b6b95a8 2018-03-12 stsp got_commands[i].cmd_usage();
192 1b6b95a8 2018-03-12 stsp
193 d7d4f210 2018-03-12 stsp error = got_commands[i].cmd_main(argc, argv);
194 80d5f134 2018-11-11 stsp if (error && !(sigint_received || sigpipe_received)) {
195 d7d4f210 2018-03-12 stsp fprintf(stderr, "%s: %s\n", getprogname(), error->msg);
196 d7d4f210 2018-03-12 stsp return 1;
197 d7d4f210 2018-03-12 stsp }
198 d7d4f210 2018-03-12 stsp
199 d7d4f210 2018-03-12 stsp return 0;
200 5c860e29 2018-03-12 stsp }
201 5c860e29 2018-03-12 stsp
202 20ecf764 2018-03-12 stsp fprintf(stderr, "%s: unknown command '%s'\n", getprogname(), argv[0]);
203 ce5b7c56 2019-07-09 stsp list_commands();
204 5c860e29 2018-03-12 stsp return 1;
205 5c860e29 2018-03-12 stsp }
206 5c860e29 2018-03-12 stsp
207 4ed7e80c 2018-05-20 stsp __dead static void
208 ce5b7c56 2019-07-09 stsp usage(int hflag)
209 5c860e29 2018-03-12 stsp {
210 5e070240 2019-06-22 stsp fprintf(stderr, "usage: %s [-h] command [arg ...]\n", getprogname());
211 ce5b7c56 2019-07-09 stsp if (hflag)
212 ce5b7c56 2019-07-09 stsp list_commands();
213 5c860e29 2018-03-12 stsp exit(1);
214 5c860e29 2018-03-12 stsp }
215 5c860e29 2018-03-12 stsp
216 0266afb7 2019-01-04 stsp static const struct got_error *
217 0ee7065d 2019-05-13 stsp get_editor(char **abspath)
218 e2ba3d07 2019-05-13 stsp {
219 0ee7065d 2019-05-13 stsp const struct got_error *err = NULL;
220 e2ba3d07 2019-05-13 stsp const char *editor;
221 e2ba3d07 2019-05-13 stsp
222 e2ba3d07 2019-05-13 stsp editor = getenv("VISUAL");
223 e2ba3d07 2019-05-13 stsp if (editor == NULL)
224 e2ba3d07 2019-05-13 stsp editor = getenv("EDITOR");
225 e2ba3d07 2019-05-13 stsp
226 0ee7065d 2019-05-13 stsp if (editor) {
227 0ee7065d 2019-05-13 stsp err = got_path_find_prog(abspath, editor);
228 0ee7065d 2019-05-13 stsp if (err)
229 0ee7065d 2019-05-13 stsp return err;
230 0ee7065d 2019-05-13 stsp }
231 e2ba3d07 2019-05-13 stsp
232 0ee7065d 2019-05-13 stsp if (*abspath == NULL) {
233 0ee7065d 2019-05-13 stsp *abspath = strdup("/bin/ed");
234 0ee7065d 2019-05-13 stsp if (*abspath == NULL)
235 0ee7065d 2019-05-13 stsp return got_error_from_errno("strdup");
236 0ee7065d 2019-05-13 stsp }
237 0ee7065d 2019-05-13 stsp
238 e2ba3d07 2019-05-13 stsp return NULL;
239 e2ba3d07 2019-05-13 stsp }
240 e2ba3d07 2019-05-13 stsp
241 e2ba3d07 2019-05-13 stsp static const struct got_error *
242 d0eebce4 2019-03-11 stsp apply_unveil(const char *repo_path, int repo_read_only,
243 314a6357 2019-05-13 stsp const char *worktree_path, int create_worktree)
244 0266afb7 2019-01-04 stsp {
245 163ce85a 2019-05-13 stsp const struct got_error *err;
246 0266afb7 2019-01-04 stsp
247 a0937847 2019-05-11 stsp if (create_worktree) {
248 a0937847 2019-05-11 stsp /* Pre-create work tree path to avoid unveiling its parents. */
249 163ce85a 2019-05-13 stsp err = got_path_mkdir(worktree_path);
250 3c45a30a 2019-05-12 jcs
251 3c45a30a 2019-05-12 jcs if (errno == EEXIST) {
252 280f921b 2019-05-12 stsp if (got_path_dir_is_empty(worktree_path)) {
253 3c45a30a 2019-05-12 jcs errno = 0;
254 163ce85a 2019-05-13 stsp err = NULL;
255 3c45a30a 2019-05-12 jcs } else {
256 df056ada 2019-05-15 stsp err = got_error_path(worktree_path,
257 df056ada 2019-05-15 stsp GOT_ERR_DIR_NOT_EMPTY);
258 3c45a30a 2019-05-12 jcs }
259 3c45a30a 2019-05-12 jcs }
260 3c45a30a 2019-05-12 jcs
261 163ce85a 2019-05-13 stsp if (err && (err->code != GOT_ERR_ERRNO || errno != EISDIR))
262 163ce85a 2019-05-13 stsp return err;
263 a0937847 2019-05-11 stsp }
264 a0937847 2019-05-11 stsp
265 d0eebce4 2019-03-11 stsp if (repo_path && unveil(repo_path, repo_read_only ? "r" : "rwc") != 0)
266 638f9024 2019-05-13 stsp return got_error_from_errno2("unveil", repo_path);
267 0266afb7 2019-01-04 stsp
268 0266afb7 2019-01-04 stsp if (worktree_path && unveil(worktree_path, "rwc") != 0)
269 638f9024 2019-05-13 stsp return got_error_from_errno2("unveil", worktree_path);
270 0266afb7 2019-01-04 stsp
271 f12d0dbe 2019-01-04 stsp if (unveil("/tmp", "rwc") != 0)
272 638f9024 2019-05-13 stsp return got_error_from_errno2("unveil", "/tmp");
273 0266afb7 2019-01-04 stsp
274 163ce85a 2019-05-13 stsp err = got_privsep_unveil_exec_helpers();
275 163ce85a 2019-05-13 stsp if (err != NULL)
276 163ce85a 2019-05-13 stsp return err;
277 0266afb7 2019-01-04 stsp
278 0266afb7 2019-01-04 stsp if (unveil(NULL, NULL) != 0)
279 638f9024 2019-05-13 stsp return got_error_from_errno("unveil");
280 0266afb7 2019-01-04 stsp
281 0266afb7 2019-01-04 stsp return NULL;
282 2c7829a4 2019-06-17 stsp }
283 2c7829a4 2019-06-17 stsp
284 2c7829a4 2019-06-17 stsp __dead static void
285 2c7829a4 2019-06-17 stsp usage_init(void)
286 2c7829a4 2019-06-17 stsp {
287 2c7829a4 2019-06-17 stsp fprintf(stderr, "usage: %s init path\n", getprogname());
288 2c7829a4 2019-06-17 stsp exit(1);
289 2c7829a4 2019-06-17 stsp }
290 2c7829a4 2019-06-17 stsp
291 2c7829a4 2019-06-17 stsp static const struct got_error *
292 2c7829a4 2019-06-17 stsp cmd_init(int argc, char *argv[])
293 2c7829a4 2019-06-17 stsp {
294 2c7829a4 2019-06-17 stsp const struct got_error *error = NULL;
295 2c7829a4 2019-06-17 stsp char *repo_path = NULL;
296 2c7829a4 2019-06-17 stsp int ch;
297 2c7829a4 2019-06-17 stsp
298 2c7829a4 2019-06-17 stsp while ((ch = getopt(argc, argv, "")) != -1) {
299 2c7829a4 2019-06-17 stsp switch (ch) {
300 2c7829a4 2019-06-17 stsp default:
301 2c7829a4 2019-06-17 stsp usage_init();
302 2c7829a4 2019-06-17 stsp /* NOTREACHED */
303 2c7829a4 2019-06-17 stsp }
304 2c7829a4 2019-06-17 stsp }
305 2c7829a4 2019-06-17 stsp
306 2c7829a4 2019-06-17 stsp argc -= optind;
307 2c7829a4 2019-06-17 stsp argv += optind;
308 2c7829a4 2019-06-17 stsp
309 2c7829a4 2019-06-17 stsp #ifndef PROFILE
310 2c7829a4 2019-06-17 stsp if (pledge("stdio rpath wpath cpath unveil", NULL) == -1)
311 2c7829a4 2019-06-17 stsp err(1, "pledge");
312 2c7829a4 2019-06-17 stsp #endif
313 2c7829a4 2019-06-17 stsp if (argc != 1)
314 bc20e173 2019-06-17 stsp usage_init();
315 2c7829a4 2019-06-17 stsp
316 2c7829a4 2019-06-17 stsp repo_path = strdup(argv[0]);
317 2c7829a4 2019-06-17 stsp if (repo_path == NULL)
318 2c7829a4 2019-06-17 stsp return got_error_from_errno("strdup");
319 2c7829a4 2019-06-17 stsp
320 2c7829a4 2019-06-17 stsp got_path_strip_trailing_slashes(repo_path);
321 2c7829a4 2019-06-17 stsp
322 2c7829a4 2019-06-17 stsp error = got_path_mkdir(repo_path);
323 2c7829a4 2019-06-17 stsp if (error &&
324 2c7829a4 2019-06-17 stsp !(error->code == GOT_ERR_ERRNO && errno == EEXIST))
325 2c7829a4 2019-06-17 stsp goto done;
326 2c7829a4 2019-06-17 stsp
327 2c7829a4 2019-06-17 stsp error = apply_unveil(repo_path, 0, NULL, 0);
328 2c7829a4 2019-06-17 stsp if (error)
329 2c7829a4 2019-06-17 stsp goto done;
330 2c7829a4 2019-06-17 stsp
331 2c7829a4 2019-06-17 stsp error = got_repo_init(repo_path);
332 2c7829a4 2019-06-17 stsp if (error != NULL)
333 3ce1b845 2019-07-15 stsp goto done;
334 3ce1b845 2019-07-15 stsp
335 3ce1b845 2019-07-15 stsp done:
336 3ce1b845 2019-07-15 stsp free(repo_path);
337 3ce1b845 2019-07-15 stsp return error;
338 3ce1b845 2019-07-15 stsp }
339 3ce1b845 2019-07-15 stsp
340 3ce1b845 2019-07-15 stsp __dead static void
341 3ce1b845 2019-07-15 stsp usage_import(void)
342 3ce1b845 2019-07-15 stsp {
343 3ce1b845 2019-07-15 stsp fprintf(stderr, "usage: %s import [-b branch] [-m message] "
344 3ce1b845 2019-07-15 stsp "[-r repository-path] [-I pattern] path\n", getprogname());
345 3ce1b845 2019-07-15 stsp exit(1);
346 3ce1b845 2019-07-15 stsp }
347 3ce1b845 2019-07-15 stsp
348 3ce1b845 2019-07-15 stsp int
349 3ce1b845 2019-07-15 stsp spawn_editor(const char *editor, const char *file)
350 3ce1b845 2019-07-15 stsp {
351 3ce1b845 2019-07-15 stsp pid_t pid;
352 3ce1b845 2019-07-15 stsp sig_t sighup, sigint, sigquit;
353 3ce1b845 2019-07-15 stsp int st = -1;
354 3ce1b845 2019-07-15 stsp
355 3ce1b845 2019-07-15 stsp sighup = signal(SIGHUP, SIG_IGN);
356 3ce1b845 2019-07-15 stsp sigint = signal(SIGINT, SIG_IGN);
357 3ce1b845 2019-07-15 stsp sigquit = signal(SIGQUIT, SIG_IGN);
358 3ce1b845 2019-07-15 stsp
359 3ce1b845 2019-07-15 stsp switch (pid = fork()) {
360 3ce1b845 2019-07-15 stsp case -1:
361 3ce1b845 2019-07-15 stsp goto doneediting;
362 3ce1b845 2019-07-15 stsp case 0:
363 3ce1b845 2019-07-15 stsp execl(editor, editor, file, (char *)NULL);
364 3ce1b845 2019-07-15 stsp _exit(127);
365 3ce1b845 2019-07-15 stsp }
366 3ce1b845 2019-07-15 stsp
367 3ce1b845 2019-07-15 stsp while (waitpid(pid, &st, 0) == -1)
368 3ce1b845 2019-07-15 stsp if (errno != EINTR)
369 3ce1b845 2019-07-15 stsp break;
370 3ce1b845 2019-07-15 stsp
371 3ce1b845 2019-07-15 stsp doneediting:
372 3ce1b845 2019-07-15 stsp (void)signal(SIGHUP, sighup);
373 3ce1b845 2019-07-15 stsp (void)signal(SIGINT, sigint);
374 3ce1b845 2019-07-15 stsp (void)signal(SIGQUIT, sigquit);
375 3ce1b845 2019-07-15 stsp
376 3ce1b845 2019-07-15 stsp if (!WIFEXITED(st)) {
377 3ce1b845 2019-07-15 stsp errno = EINTR;
378 3ce1b845 2019-07-15 stsp return -1;
379 3ce1b845 2019-07-15 stsp }
380 3ce1b845 2019-07-15 stsp
381 3ce1b845 2019-07-15 stsp return WEXITSTATUS(st);
382 3ce1b845 2019-07-15 stsp }
383 3ce1b845 2019-07-15 stsp
384 3ce1b845 2019-07-15 stsp static const struct got_error *
385 3ce1b845 2019-07-15 stsp edit_logmsg(char **logmsg, const char *editor, const char *logmsg_path,
386 3ce1b845 2019-07-15 stsp const char *initial_content)
387 3ce1b845 2019-07-15 stsp {
388 3ce1b845 2019-07-15 stsp const struct got_error *err = NULL;
389 3ce1b845 2019-07-15 stsp char buf[1024];
390 3ce1b845 2019-07-15 stsp struct stat st, st2;
391 3ce1b845 2019-07-15 stsp FILE *fp;
392 3ce1b845 2019-07-15 stsp int content_changed = 0;
393 3ce1b845 2019-07-15 stsp size_t len;
394 3ce1b845 2019-07-15 stsp
395 3ce1b845 2019-07-15 stsp *logmsg = NULL;
396 3ce1b845 2019-07-15 stsp
397 3ce1b845 2019-07-15 stsp if (stat(logmsg_path, &st) == -1)
398 3ce1b845 2019-07-15 stsp return got_error_from_errno2("stat", logmsg_path);
399 3ce1b845 2019-07-15 stsp
400 3ce1b845 2019-07-15 stsp if (spawn_editor(editor, logmsg_path) == -1)
401 3ce1b845 2019-07-15 stsp return got_error_from_errno("failed spawning editor");
402 3ce1b845 2019-07-15 stsp
403 3ce1b845 2019-07-15 stsp if (stat(logmsg_path, &st2) == -1)
404 3ce1b845 2019-07-15 stsp return got_error_from_errno("stat");
405 3ce1b845 2019-07-15 stsp
406 3ce1b845 2019-07-15 stsp if (st.st_mtime == st2.st_mtime && st.st_size == st2.st_size)
407 3ce1b845 2019-07-15 stsp return got_error_msg(GOT_ERR_COMMIT_MSG_EMPTY,
408 3ce1b845 2019-07-15 stsp "no changes made to commit message, aborting");
409 3ce1b845 2019-07-15 stsp
410 3ce1b845 2019-07-15 stsp *logmsg = malloc(st2.st_size + 1);
411 3ce1b845 2019-07-15 stsp if (*logmsg == NULL)
412 3ce1b845 2019-07-15 stsp return got_error_from_errno("malloc");
413 3ce1b845 2019-07-15 stsp (*logmsg)[0] = '\0';
414 3ce1b845 2019-07-15 stsp len = 0;
415 3ce1b845 2019-07-15 stsp
416 3ce1b845 2019-07-15 stsp fp = fopen(logmsg_path, "r");
417 3ce1b845 2019-07-15 stsp if (fp == NULL) {
418 3ce1b845 2019-07-15 stsp err = got_error_from_errno("fopen");
419 3ce1b845 2019-07-15 stsp goto done;
420 3ce1b845 2019-07-15 stsp }
421 3ce1b845 2019-07-15 stsp while (fgets(buf, sizeof(buf), fp) != NULL) {
422 3ce1b845 2019-07-15 stsp if (!content_changed && strcmp(buf, initial_content) != 0)
423 3ce1b845 2019-07-15 stsp content_changed = 1;
424 3ce1b845 2019-07-15 stsp if (buf[0] == '#' || (len == 0 && buf[0] == '\n'))
425 3ce1b845 2019-07-15 stsp continue; /* remove comments and leading empty lines */
426 3ce1b845 2019-07-15 stsp len = strlcat(*logmsg, buf, st2.st_size);
427 3ce1b845 2019-07-15 stsp }
428 3ce1b845 2019-07-15 stsp fclose(fp);
429 3ce1b845 2019-07-15 stsp
430 3ce1b845 2019-07-15 stsp while (len > 0 && (*logmsg)[len - 1] == '\n') {
431 3ce1b845 2019-07-15 stsp (*logmsg)[len - 1] = '\0';
432 3ce1b845 2019-07-15 stsp len--;
433 3ce1b845 2019-07-15 stsp }
434 3ce1b845 2019-07-15 stsp
435 3ce1b845 2019-07-15 stsp if (len == 0 || !content_changed)
436 3ce1b845 2019-07-15 stsp err = got_error_msg(GOT_ERR_COMMIT_MSG_EMPTY,
437 3ce1b845 2019-07-15 stsp "commit message cannot be empty, aborting");
438 3ce1b845 2019-07-15 stsp done:
439 3ce1b845 2019-07-15 stsp if (err) {
440 3ce1b845 2019-07-15 stsp free(*logmsg);
441 3ce1b845 2019-07-15 stsp *logmsg = NULL;
442 3ce1b845 2019-07-15 stsp }
443 3ce1b845 2019-07-15 stsp return err;
444 3ce1b845 2019-07-15 stsp }
445 3ce1b845 2019-07-15 stsp
446 3ce1b845 2019-07-15 stsp static const struct got_error *
447 3ce1b845 2019-07-15 stsp collect_import_msg(char **logmsg, const char *editor, const char *path_dir,
448 3ce1b845 2019-07-15 stsp const char *branch_name)
449 3ce1b845 2019-07-15 stsp {
450 3ce1b845 2019-07-15 stsp char *initial_content = NULL, *logmsg_path = NULL;
451 3ce1b845 2019-07-15 stsp const struct got_error *err = NULL;
452 3ce1b845 2019-07-15 stsp int fd;
453 3ce1b845 2019-07-15 stsp
454 3ce1b845 2019-07-15 stsp if (asprintf(&initial_content,
455 3ce1b845 2019-07-15 stsp "\n# %s to be imported to branch %s\n", path_dir,
456 3ce1b845 2019-07-15 stsp branch_name) == -1)
457 3ce1b845 2019-07-15 stsp return got_error_from_errno("asprintf");
458 3ce1b845 2019-07-15 stsp
459 3ce1b845 2019-07-15 stsp err = got_opentemp_named_fd(&logmsg_path, &fd, "/tmp/got-importmsg");
460 3ce1b845 2019-07-15 stsp if (err)
461 3ce1b845 2019-07-15 stsp goto done;
462 3ce1b845 2019-07-15 stsp
463 3ce1b845 2019-07-15 stsp dprintf(fd, initial_content);
464 3ce1b845 2019-07-15 stsp close(fd);
465 3ce1b845 2019-07-15 stsp
466 3ce1b845 2019-07-15 stsp err = edit_logmsg(logmsg, editor, logmsg_path, initial_content);
467 3ce1b845 2019-07-15 stsp done:
468 3ce1b845 2019-07-15 stsp free(initial_content);
469 3ce1b845 2019-07-15 stsp free(logmsg_path);
470 3ce1b845 2019-07-15 stsp return err;
471 3ce1b845 2019-07-15 stsp }
472 3ce1b845 2019-07-15 stsp
473 3ce1b845 2019-07-15 stsp static const struct got_error *
474 3ce1b845 2019-07-15 stsp import_progress(void *arg, const char *path)
475 3ce1b845 2019-07-15 stsp {
476 3ce1b845 2019-07-15 stsp printf("A %s\n", path);
477 3ce1b845 2019-07-15 stsp return NULL;
478 3ce1b845 2019-07-15 stsp }
479 3ce1b845 2019-07-15 stsp
480 3ce1b845 2019-07-15 stsp static const struct got_error *
481 3ce1b845 2019-07-15 stsp cmd_import(int argc, char *argv[])
482 3ce1b845 2019-07-15 stsp {
483 3ce1b845 2019-07-15 stsp const struct got_error *error = NULL;
484 3ce1b845 2019-07-15 stsp char *path_dir = NULL, *repo_path = NULL, *logmsg = NULL;
485 3ce1b845 2019-07-15 stsp char *editor = NULL;
486 3ce1b845 2019-07-15 stsp const char *got_author = getenv("GOT_AUTHOR");
487 3ce1b845 2019-07-15 stsp const char *branch_name = "master";
488 3ce1b845 2019-07-15 stsp char *refname = NULL, *id_str = NULL;
489 3ce1b845 2019-07-15 stsp struct got_repository *repo = NULL;
490 3ce1b845 2019-07-15 stsp struct got_reference *branch_ref = NULL, *head_ref = NULL;
491 3ce1b845 2019-07-15 stsp struct got_object_id *new_commit_id = NULL;
492 3ce1b845 2019-07-15 stsp int ch;
493 3ce1b845 2019-07-15 stsp struct got_pathlist_head ignores;
494 3ce1b845 2019-07-15 stsp struct got_pathlist_entry *pe;
495 3ce1b845 2019-07-15 stsp
496 3ce1b845 2019-07-15 stsp TAILQ_INIT(&ignores);
497 3ce1b845 2019-07-15 stsp
498 3ce1b845 2019-07-15 stsp while ((ch = getopt(argc, argv, "b:m:r:I:")) != -1) {
499 3ce1b845 2019-07-15 stsp switch (ch) {
500 3ce1b845 2019-07-15 stsp case 'b':
501 3ce1b845 2019-07-15 stsp branch_name = optarg;
502 3ce1b845 2019-07-15 stsp break;
503 3ce1b845 2019-07-15 stsp case 'm':
504 3ce1b845 2019-07-15 stsp logmsg = strdup(optarg);
505 3ce1b845 2019-07-15 stsp if (logmsg == NULL) {
506 3ce1b845 2019-07-15 stsp error = got_error_from_errno("strdup");
507 3ce1b845 2019-07-15 stsp goto done;
508 3ce1b845 2019-07-15 stsp }
509 3ce1b845 2019-07-15 stsp break;
510 3ce1b845 2019-07-15 stsp case 'r':
511 3ce1b845 2019-07-15 stsp repo_path = realpath(optarg, NULL);
512 3ce1b845 2019-07-15 stsp if (repo_path == NULL) {
513 3ce1b845 2019-07-15 stsp error = got_error_from_errno("realpath");
514 3ce1b845 2019-07-15 stsp goto done;
515 3ce1b845 2019-07-15 stsp }
516 3ce1b845 2019-07-15 stsp break;
517 3ce1b845 2019-07-15 stsp case 'I':
518 3ce1b845 2019-07-15 stsp if (optarg[0] == '\0')
519 3ce1b845 2019-07-15 stsp break;
520 3ce1b845 2019-07-15 stsp error = got_pathlist_insert(&pe, &ignores, optarg,
521 3ce1b845 2019-07-15 stsp NULL);
522 3ce1b845 2019-07-15 stsp if (error)
523 3ce1b845 2019-07-15 stsp goto done;
524 3ce1b845 2019-07-15 stsp break;
525 3ce1b845 2019-07-15 stsp default:
526 3ce1b845 2019-07-15 stsp usage_init();
527 3ce1b845 2019-07-15 stsp /* NOTREACHED */
528 3ce1b845 2019-07-15 stsp }
529 3ce1b845 2019-07-15 stsp }
530 3ce1b845 2019-07-15 stsp
531 3ce1b845 2019-07-15 stsp argc -= optind;
532 3ce1b845 2019-07-15 stsp argv += optind;
533 3ce1b845 2019-07-15 stsp
534 3ce1b845 2019-07-15 stsp #ifndef PROFILE
535 3ce1b845 2019-07-15 stsp if (pledge("stdio rpath wpath cpath fattr flock proc exec unveil",
536 3ce1b845 2019-07-15 stsp NULL) == -1)
537 3ce1b845 2019-07-15 stsp err(1, "pledge");
538 3ce1b845 2019-07-15 stsp #endif
539 3ce1b845 2019-07-15 stsp if (argc != 1)
540 3ce1b845 2019-07-15 stsp usage_import();
541 3ce1b845 2019-07-15 stsp
542 3ce1b845 2019-07-15 stsp if (got_author == NULL) {
543 3ce1b845 2019-07-15 stsp /* TODO: Look current user up in password database */
544 3ce1b845 2019-07-15 stsp error = got_error(GOT_ERR_COMMIT_NO_AUTHOR);
545 2c7829a4 2019-06-17 stsp goto done;
546 3ce1b845 2019-07-15 stsp }
547 2c7829a4 2019-06-17 stsp
548 3ce1b845 2019-07-15 stsp if (repo_path == NULL) {
549 3ce1b845 2019-07-15 stsp repo_path = getcwd(NULL, 0);
550 3ce1b845 2019-07-15 stsp if (repo_path == NULL)
551 3ce1b845 2019-07-15 stsp return got_error_from_errno("getcwd");
552 3ce1b845 2019-07-15 stsp }
553 3ce1b845 2019-07-15 stsp got_path_strip_trailing_slashes(repo_path);
554 3ce1b845 2019-07-15 stsp error = got_repo_open(&repo, repo_path);
555 3ce1b845 2019-07-15 stsp if (error)
556 3ce1b845 2019-07-15 stsp goto done;
557 3ce1b845 2019-07-15 stsp
558 3ce1b845 2019-07-15 stsp if (asprintf(&refname, "refs/heads/%s", branch_name) == -1) {
559 3ce1b845 2019-07-15 stsp error = got_error_from_errno("asprintf");
560 3ce1b845 2019-07-15 stsp goto done;
561 3ce1b845 2019-07-15 stsp }
562 3ce1b845 2019-07-15 stsp
563 3ce1b845 2019-07-15 stsp error = got_ref_open(&branch_ref, repo, refname, 0);
564 3ce1b845 2019-07-15 stsp if (error) {
565 3ce1b845 2019-07-15 stsp if (error->code != GOT_ERR_NOT_REF)
566 3ce1b845 2019-07-15 stsp goto done;
567 3ce1b845 2019-07-15 stsp } else {
568 3ce1b845 2019-07-15 stsp error = got_error_msg(GOT_ERR_BRANCH_EXISTS,
569 3ce1b845 2019-07-15 stsp "import target branch already exists");
570 3ce1b845 2019-07-15 stsp goto done;
571 3ce1b845 2019-07-15 stsp }
572 3ce1b845 2019-07-15 stsp
573 3ce1b845 2019-07-15 stsp path_dir = realpath(argv[0], NULL);
574 3ce1b845 2019-07-15 stsp if (path_dir == NULL) {
575 3ce1b845 2019-07-15 stsp error = got_error_from_errno("realpath");
576 3ce1b845 2019-07-15 stsp goto done;
577 3ce1b845 2019-07-15 stsp }
578 3ce1b845 2019-07-15 stsp got_path_strip_trailing_slashes(path_dir);
579 3ce1b845 2019-07-15 stsp
580 3ce1b845 2019-07-15 stsp /*
581 3ce1b845 2019-07-15 stsp * unveil(2) traverses exec(2); if an editor is used we have
582 3ce1b845 2019-07-15 stsp * to apply unveil after the log message has been written.
583 3ce1b845 2019-07-15 stsp */
584 3ce1b845 2019-07-15 stsp if (logmsg == NULL || strlen(logmsg) == 0) {
585 3ce1b845 2019-07-15 stsp error = get_editor(&editor);
586 3ce1b845 2019-07-15 stsp if (error)
587 3ce1b845 2019-07-15 stsp goto done;
588 3ce1b845 2019-07-15 stsp error = collect_import_msg(&logmsg, editor, path_dir, refname);
589 3ce1b845 2019-07-15 stsp if (error)
590 3ce1b845 2019-07-15 stsp goto done;
591 3ce1b845 2019-07-15 stsp }
592 3ce1b845 2019-07-15 stsp
593 3ce1b845 2019-07-15 stsp if (unveil(path_dir, "r") != 0)
594 3ce1b845 2019-07-15 stsp return got_error_from_errno2("unveil", path_dir);
595 3ce1b845 2019-07-15 stsp
596 3ce1b845 2019-07-15 stsp error = apply_unveil(got_repo_get_path(repo), 0, NULL, 0);
597 3ce1b845 2019-07-15 stsp if (error)
598 3ce1b845 2019-07-15 stsp goto done;
599 3ce1b845 2019-07-15 stsp
600 3ce1b845 2019-07-15 stsp error = got_repo_import(&new_commit_id, path_dir, logmsg,
601 3ce1b845 2019-07-15 stsp got_author, &ignores, repo, import_progress, NULL);
602 3ce1b845 2019-07-15 stsp if (error)
603 3ce1b845 2019-07-15 stsp goto done;
604 3ce1b845 2019-07-15 stsp
605 3ce1b845 2019-07-15 stsp error = got_ref_alloc(&branch_ref, refname, new_commit_id);
606 3ce1b845 2019-07-15 stsp if (error)
607 3ce1b845 2019-07-15 stsp goto done;
608 3ce1b845 2019-07-15 stsp
609 3ce1b845 2019-07-15 stsp error = got_ref_write(branch_ref, repo);
610 3ce1b845 2019-07-15 stsp if (error)
611 3ce1b845 2019-07-15 stsp goto done;
612 3ce1b845 2019-07-15 stsp
613 3ce1b845 2019-07-15 stsp error = got_object_id_str(&id_str, new_commit_id);
614 3ce1b845 2019-07-15 stsp if (error)
615 3ce1b845 2019-07-15 stsp goto done;
616 3ce1b845 2019-07-15 stsp
617 3ce1b845 2019-07-15 stsp error = got_ref_open(&head_ref, repo, GOT_REF_HEAD, 0);
618 3ce1b845 2019-07-15 stsp if (error) {
619 3ce1b845 2019-07-15 stsp if (error->code != GOT_ERR_NOT_REF)
620 3ce1b845 2019-07-15 stsp goto done;
621 3ce1b845 2019-07-15 stsp
622 3ce1b845 2019-07-15 stsp error = got_ref_alloc_symref(&head_ref, GOT_REF_HEAD,
623 3ce1b845 2019-07-15 stsp branch_ref);
624 3ce1b845 2019-07-15 stsp if (error)
625 3ce1b845 2019-07-15 stsp goto done;
626 3ce1b845 2019-07-15 stsp
627 3ce1b845 2019-07-15 stsp error = got_ref_write(head_ref, repo);
628 3ce1b845 2019-07-15 stsp if (error)
629 3ce1b845 2019-07-15 stsp goto done;
630 3ce1b845 2019-07-15 stsp }
631 3ce1b845 2019-07-15 stsp
632 3ce1b845 2019-07-15 stsp printf("Created branch %s with commit %s\n",
633 3ce1b845 2019-07-15 stsp got_ref_get_name(branch_ref), id_str);
634 2c7829a4 2019-06-17 stsp done:
635 2c7829a4 2019-06-17 stsp free(repo_path);
636 3ce1b845 2019-07-15 stsp free(editor);
637 3ce1b845 2019-07-15 stsp free(refname);
638 3ce1b845 2019-07-15 stsp free(new_commit_id);
639 3ce1b845 2019-07-15 stsp free(id_str);
640 3ce1b845 2019-07-15 stsp if (branch_ref)
641 3ce1b845 2019-07-15 stsp got_ref_close(branch_ref);
642 3ce1b845 2019-07-15 stsp if (head_ref)
643 3ce1b845 2019-07-15 stsp got_ref_close(head_ref);
644 2c7829a4 2019-06-17 stsp return error;
645 0266afb7 2019-01-04 stsp }
646 0266afb7 2019-01-04 stsp
647 4ed7e80c 2018-05-20 stsp __dead static void
648 c09a553d 2018-03-12 stsp usage_checkout(void)
649 c09a553d 2018-03-12 stsp {
650 08573d5b 2019-05-14 stsp fprintf(stderr, "usage: %s checkout [-b branch] [-c commit] "
651 08573d5b 2019-05-14 stsp "[-p prefix] repository-path [worktree-path]\n", getprogname());
652 c09a553d 2018-03-12 stsp exit(1);
653 92a684f4 2018-03-12 stsp }
654 92a684f4 2018-03-12 stsp
655 1ee397ad 2019-07-12 stsp static const struct got_error *
656 a0eb853d 2018-12-29 stsp checkout_progress(void *arg, unsigned char status, const char *path)
657 92a684f4 2018-03-12 stsp {
658 92a684f4 2018-03-12 stsp char *worktree_path = arg;
659 a484d721 2019-06-10 stsp
660 a484d721 2019-06-10 stsp /* Base commit bump happens silently. */
661 a484d721 2019-06-10 stsp if (status == GOT_STATUS_BUMP_BASE)
662 1ee397ad 2019-07-12 stsp return NULL;
663 92a684f4 2018-03-12 stsp
664 92a684f4 2018-03-12 stsp while (path[0] == '/')
665 92a684f4 2018-03-12 stsp path++;
666 92a684f4 2018-03-12 stsp
667 d7b62c98 2018-12-27 stsp printf("%c %s/%s\n", status, worktree_path, path);
668 1ee397ad 2019-07-12 stsp return NULL;
669 99437157 2018-11-11 stsp }
670 99437157 2018-11-11 stsp
671 99437157 2018-11-11 stsp static const struct got_error *
672 6bad629b 2019-02-04 stsp check_cancelled(void *arg)
673 99437157 2018-11-11 stsp {
674 99437157 2018-11-11 stsp if (sigint_received || sigpipe_received)
675 99437157 2018-11-11 stsp return got_error(GOT_ERR_CANCELLED);
676 99437157 2018-11-11 stsp return NULL;
677 8069f636 2019-01-12 stsp }
678 8069f636 2019-01-12 stsp
679 8069f636 2019-01-12 stsp static const struct got_error *
680 024e9686 2019-05-14 stsp check_linear_ancestry(struct got_object_id *commit_id,
681 024e9686 2019-05-14 stsp struct got_object_id *base_commit_id, struct got_repository *repo)
682 8069f636 2019-01-12 stsp {
683 d5bea539 2019-05-13 stsp const struct got_error *err = NULL;
684 024e9686 2019-05-14 stsp struct got_object_id *yca_id;
685 8069f636 2019-01-12 stsp
686 d5bea539 2019-05-13 stsp err = got_commit_graph_find_youngest_common_ancestor(&yca_id,
687 d5bea539 2019-05-13 stsp commit_id, base_commit_id, repo);
688 36a38700 2019-05-10 stsp if (err)
689 36a38700 2019-05-10 stsp return err;
690 8069f636 2019-01-12 stsp
691 d5bea539 2019-05-13 stsp if (yca_id == NULL)
692 d5bea539 2019-05-13 stsp return got_error(GOT_ERR_ANCESTRY);
693 8069f636 2019-01-12 stsp
694 d5bea539 2019-05-13 stsp /*
695 d5bea539 2019-05-13 stsp * Require a straight line of history between the target commit
696 d5bea539 2019-05-13 stsp * and the work tree's base commit.
697 d5bea539 2019-05-13 stsp *
698 d5751d49 2019-05-14 stsp * Non-linear situations such as this require a rebase:
699 d5bea539 2019-05-13 stsp *
700 d5bea539 2019-05-13 stsp * (commit) D F (base_commit)
701 d5bea539 2019-05-13 stsp * \ /
702 d5bea539 2019-05-13 stsp * C E
703 d5bea539 2019-05-13 stsp * \ /
704 d5bea539 2019-05-13 stsp * B (yca)
705 d5bea539 2019-05-13 stsp * |
706 d5bea539 2019-05-13 stsp * A
707 d5bea539 2019-05-13 stsp *
708 d5bea539 2019-05-13 stsp * 'got update' only handles linear cases:
709 d5bea539 2019-05-13 stsp * Update forwards in time: A (base/yca) - B - C - D (commit)
710 efa2b6f7 2019-05-14 stsp * Update backwards in time: D (base) - C - B - A (commit/yca)
711 d5bea539 2019-05-13 stsp */
712 d5bea539 2019-05-13 stsp if (got_object_id_cmp(commit_id, yca_id) != 0 &&
713 d5bea539 2019-05-13 stsp got_object_id_cmp(base_commit_id, yca_id) != 0)
714 d5bea539 2019-05-13 stsp return got_error(GOT_ERR_ANCESTRY);
715 8069f636 2019-01-12 stsp
716 d5bea539 2019-05-13 stsp free(yca_id);
717 d5bea539 2019-05-13 stsp return NULL;
718 c09a553d 2018-03-12 stsp }
719 a367ff0f 2019-05-14 stsp
720 a367ff0f 2019-05-14 stsp static const struct got_error *
721 a367ff0f 2019-05-14 stsp check_same_branch(struct got_object_id *commit_id,
722 a367ff0f 2019-05-14 stsp struct got_reference *head_ref, struct got_repository *repo)
723 a367ff0f 2019-05-14 stsp {
724 a367ff0f 2019-05-14 stsp const struct got_error *err = NULL;
725 a367ff0f 2019-05-14 stsp struct got_commit_graph *graph = NULL;
726 a367ff0f 2019-05-14 stsp struct got_object_id *head_commit_id = NULL;
727 a367ff0f 2019-05-14 stsp int is_same_branch = 0;
728 a367ff0f 2019-05-14 stsp
729 a367ff0f 2019-05-14 stsp err = got_ref_resolve(&head_commit_id, repo, head_ref);
730 a367ff0f 2019-05-14 stsp if (err)
731 a367ff0f 2019-05-14 stsp goto done;
732 a367ff0f 2019-05-14 stsp
733 a367ff0f 2019-05-14 stsp err = got_commit_graph_open(&graph, head_commit_id, "/", 1, repo);
734 a367ff0f 2019-05-14 stsp if (err)
735 a367ff0f 2019-05-14 stsp goto done;
736 a367ff0f 2019-05-14 stsp
737 a367ff0f 2019-05-14 stsp err = got_commit_graph_iter_start(graph, head_commit_id, repo);
738 a367ff0f 2019-05-14 stsp if (err)
739 a367ff0f 2019-05-14 stsp goto done;
740 a367ff0f 2019-05-14 stsp
741 a367ff0f 2019-05-14 stsp for (;;) {
742 a367ff0f 2019-05-14 stsp struct got_object_id *id;
743 a367ff0f 2019-05-14 stsp err = got_commit_graph_iter_next(&id, graph);
744 a367ff0f 2019-05-14 stsp if (err) {
745 a367ff0f 2019-05-14 stsp if (err->code == GOT_ERR_ITER_COMPLETED) {
746 a367ff0f 2019-05-14 stsp err = NULL;
747 a367ff0f 2019-05-14 stsp break;
748 a367ff0f 2019-05-14 stsp }
749 a367ff0f 2019-05-14 stsp else if (err->code != GOT_ERR_ITER_NEED_MORE)
750 a367ff0f 2019-05-14 stsp break;
751 a367ff0f 2019-05-14 stsp err = got_commit_graph_fetch_commits(graph, 1,
752 a367ff0f 2019-05-14 stsp repo);
753 a367ff0f 2019-05-14 stsp if (err)
754 a367ff0f 2019-05-14 stsp break;
755 a367ff0f 2019-05-14 stsp }
756 c09a553d 2018-03-12 stsp
757 a367ff0f 2019-05-14 stsp if (id) {
758 a367ff0f 2019-05-14 stsp if (got_object_id_cmp(id, commit_id) == 0) {
759 a367ff0f 2019-05-14 stsp is_same_branch = 1;
760 a367ff0f 2019-05-14 stsp break;
761 a367ff0f 2019-05-14 stsp }
762 a367ff0f 2019-05-14 stsp }
763 a367ff0f 2019-05-14 stsp }
764 a367ff0f 2019-05-14 stsp done:
765 a367ff0f 2019-05-14 stsp if (graph)
766 a367ff0f 2019-05-14 stsp got_commit_graph_close(graph);
767 a367ff0f 2019-05-14 stsp free(head_commit_id);
768 a367ff0f 2019-05-14 stsp if (!err && !is_same_branch)
769 a367ff0f 2019-05-14 stsp err = got_error(GOT_ERR_ANCESTRY);
770 a367ff0f 2019-05-14 stsp return err;
771 a367ff0f 2019-05-14 stsp }
772 8069f636 2019-01-12 stsp
773 4ed7e80c 2018-05-20 stsp static const struct got_error *
774 c09a553d 2018-03-12 stsp cmd_checkout(int argc, char *argv[])
775 c09a553d 2018-03-12 stsp {
776 c09a553d 2018-03-12 stsp const struct got_error *error = NULL;
777 c09a553d 2018-03-12 stsp struct got_repository *repo = NULL;
778 c09a553d 2018-03-12 stsp struct got_reference *head_ref = NULL;
779 c09a553d 2018-03-12 stsp struct got_worktree *worktree = NULL;
780 c09a553d 2018-03-12 stsp char *repo_path = NULL;
781 c09a553d 2018-03-12 stsp char *worktree_path = NULL;
782 0bb8a95e 2018-03-12 stsp const char *path_prefix = "";
783 08573d5b 2019-05-14 stsp const char *branch_name = GOT_REF_HEAD;
784 8069f636 2019-01-12 stsp char *commit_id_str = NULL;
785 72151b04 2019-05-11 stsp int ch, same_path_prefix;
786 c09a553d 2018-03-12 stsp
787 08573d5b 2019-05-14 stsp while ((ch = getopt(argc, argv, "b:c:p:")) != -1) {
788 0bb8a95e 2018-03-12 stsp switch (ch) {
789 08573d5b 2019-05-14 stsp case 'b':
790 08573d5b 2019-05-14 stsp branch_name = optarg;
791 08573d5b 2019-05-14 stsp break;
792 8069f636 2019-01-12 stsp case 'c':
793 8069f636 2019-01-12 stsp commit_id_str = strdup(optarg);
794 8069f636 2019-01-12 stsp if (commit_id_str == NULL)
795 638f9024 2019-05-13 stsp return got_error_from_errno("strdup");
796 8069f636 2019-01-12 stsp break;
797 0bb8a95e 2018-03-12 stsp case 'p':
798 0bb8a95e 2018-03-12 stsp path_prefix = optarg;
799 0bb8a95e 2018-03-12 stsp break;
800 0bb8a95e 2018-03-12 stsp default:
801 2deda0b9 2019-03-07 stsp usage_checkout();
802 0bb8a95e 2018-03-12 stsp /* NOTREACHED */
803 0bb8a95e 2018-03-12 stsp }
804 0bb8a95e 2018-03-12 stsp }
805 0bb8a95e 2018-03-12 stsp
806 0bb8a95e 2018-03-12 stsp argc -= optind;
807 0bb8a95e 2018-03-12 stsp argv += optind;
808 0bb8a95e 2018-03-12 stsp
809 6715a751 2018-03-16 stsp #ifndef PROFILE
810 68ed9ba5 2019-02-10 stsp if (pledge("stdio rpath wpath cpath fattr flock proc exec sendfd "
811 68ed9ba5 2019-02-10 stsp "unveil", NULL) == -1)
812 c09a553d 2018-03-12 stsp err(1, "pledge");
813 6715a751 2018-03-16 stsp #endif
814 0bb8a95e 2018-03-12 stsp if (argc == 1) {
815 c09a553d 2018-03-12 stsp char *cwd, *base, *dotgit;
816 76089277 2018-04-01 stsp repo_path = realpath(argv[0], NULL);
817 76089277 2018-04-01 stsp if (repo_path == NULL)
818 638f9024 2019-05-13 stsp return got_error_from_errno2("realpath", argv[0]);
819 c09a553d 2018-03-12 stsp cwd = getcwd(NULL, 0);
820 76089277 2018-04-01 stsp if (cwd == NULL) {
821 638f9024 2019-05-13 stsp error = got_error_from_errno("getcwd");
822 76089277 2018-04-01 stsp goto done;
823 76089277 2018-04-01 stsp }
824 230a42bd 2019-05-11 jcs if (path_prefix[0]) {
825 230a42bd 2019-05-11 jcs base = basename(path_prefix);
826 230a42bd 2019-05-11 jcs if (base == NULL) {
827 638f9024 2019-05-13 stsp error = got_error_from_errno2("basename",
828 230a42bd 2019-05-11 jcs path_prefix);
829 230a42bd 2019-05-11 jcs goto done;
830 230a42bd 2019-05-11 jcs }
831 230a42bd 2019-05-11 jcs } else {
832 5d7c1dab 2018-04-01 stsp base = basename(repo_path);
833 230a42bd 2019-05-11 jcs if (base == NULL) {
834 638f9024 2019-05-13 stsp error = got_error_from_errno2("basename",
835 230a42bd 2019-05-11 jcs repo_path);
836 230a42bd 2019-05-11 jcs goto done;
837 230a42bd 2019-05-11 jcs }
838 76089277 2018-04-01 stsp }
839 c09a553d 2018-03-12 stsp dotgit = strstr(base, ".git");
840 c09a553d 2018-03-12 stsp if (dotgit)
841 c09a553d 2018-03-12 stsp *dotgit = '\0';
842 c09a553d 2018-03-12 stsp if (asprintf(&worktree_path, "%s/%s", cwd, base) == -1) {
843 638f9024 2019-05-13 stsp error = got_error_from_errno("asprintf");
844 c09a553d 2018-03-12 stsp free(cwd);
845 76089277 2018-04-01 stsp goto done;
846 c09a553d 2018-03-12 stsp }
847 c09a553d 2018-03-12 stsp free(cwd);
848 0bb8a95e 2018-03-12 stsp } else if (argc == 2) {
849 76089277 2018-04-01 stsp repo_path = realpath(argv[0], NULL);
850 76089277 2018-04-01 stsp if (repo_path == NULL) {
851 638f9024 2019-05-13 stsp error = got_error_from_errno2("realpath", argv[0]);
852 76089277 2018-04-01 stsp goto done;
853 76089277 2018-04-01 stsp }
854 f7b38925 2018-04-01 stsp worktree_path = realpath(argv[1], NULL);
855 76089277 2018-04-01 stsp if (worktree_path == NULL) {
856 638f9024 2019-05-13 stsp error = got_error_from_errno2("realpath", argv[1]);
857 76089277 2018-04-01 stsp goto done;
858 76089277 2018-04-01 stsp }
859 c09a553d 2018-03-12 stsp } else
860 c09a553d 2018-03-12 stsp usage_checkout();
861 c09a553d 2018-03-12 stsp
862 7fbaa4f3 2019-05-11 stsp got_path_strip_trailing_slashes(repo_path);
863 72151b04 2019-05-11 stsp got_path_strip_trailing_slashes(worktree_path);
864 13bfb272 2019-05-10 jcs
865 c09a553d 2018-03-12 stsp error = got_repo_open(&repo, repo_path);
866 c09a553d 2018-03-12 stsp if (error != NULL)
867 c02c541e 2019-03-29 stsp goto done;
868 c02c541e 2019-03-29 stsp
869 314a6357 2019-05-13 stsp error = apply_unveil(got_repo_get_path(repo), 0, worktree_path, 1);
870 c02c541e 2019-03-29 stsp if (error)
871 c09a553d 2018-03-12 stsp goto done;
872 8069f636 2019-01-12 stsp
873 08573d5b 2019-05-14 stsp error = got_ref_open(&head_ref, repo, branch_name, 0);
874 c09a553d 2018-03-12 stsp if (error != NULL)
875 c09a553d 2018-03-12 stsp goto done;
876 c09a553d 2018-03-12 stsp
877 0bb8a95e 2018-03-12 stsp error = got_worktree_init(worktree_path, head_ref, path_prefix, repo);
878 d70b8e30 2018-12-27 stsp if (error != NULL && !(error->code == GOT_ERR_ERRNO && errno == EEXIST))
879 c09a553d 2018-03-12 stsp goto done;
880 c09a553d 2018-03-12 stsp
881 c09a553d 2018-03-12 stsp error = got_worktree_open(&worktree, worktree_path);
882 c09a553d 2018-03-12 stsp if (error != NULL)
883 c09a553d 2018-03-12 stsp goto done;
884 c09a553d 2018-03-12 stsp
885 e5dc7198 2018-12-29 stsp error = got_worktree_match_path_prefix(&same_path_prefix, worktree,
886 e5dc7198 2018-12-29 stsp path_prefix);
887 e5dc7198 2018-12-29 stsp if (error != NULL)
888 e5dc7198 2018-12-29 stsp goto done;
889 e5dc7198 2018-12-29 stsp if (!same_path_prefix) {
890 49520a32 2018-12-29 stsp error = got_error(GOT_ERR_PATH_PREFIX);
891 49520a32 2018-12-29 stsp goto done;
892 49520a32 2018-12-29 stsp }
893 49520a32 2018-12-29 stsp
894 8069f636 2019-01-12 stsp if (commit_id_str) {
895 8069f636 2019-01-12 stsp struct got_object_id *commit_id;
896 e09a504c 2019-06-28 stsp error = got_repo_match_object_id_prefix(&commit_id,
897 dd88155e 2019-06-29 stsp commit_id_str, GOT_OBJ_TYPE_COMMIT, repo);
898 8069f636 2019-01-12 stsp if (error != NULL)
899 8069f636 2019-01-12 stsp goto done;
900 024e9686 2019-05-14 stsp error = check_linear_ancestry(commit_id,
901 024e9686 2019-05-14 stsp got_worktree_get_base_commit_id(worktree), repo);
902 8069f636 2019-01-12 stsp if (error != NULL) {
903 8069f636 2019-01-12 stsp free(commit_id);
904 8069f636 2019-01-12 stsp goto done;
905 8069f636 2019-01-12 stsp }
906 45d344f6 2019-05-14 stsp error = check_same_branch(commit_id, head_ref, repo);
907 45d344f6 2019-05-14 stsp if (error)
908 45d344f6 2019-05-14 stsp goto done;
909 8069f636 2019-01-12 stsp error = got_worktree_set_base_commit_id(worktree, repo,
910 8069f636 2019-01-12 stsp commit_id);
911 8069f636 2019-01-12 stsp free(commit_id);
912 8069f636 2019-01-12 stsp if (error)
913 8069f636 2019-01-12 stsp goto done;
914 8069f636 2019-01-12 stsp }
915 8069f636 2019-01-12 stsp
916 c4cdcb68 2019-04-03 stsp error = got_worktree_checkout_files(worktree, "", repo,
917 6bad629b 2019-02-04 stsp checkout_progress, worktree_path, check_cancelled, NULL);
918 c09a553d 2018-03-12 stsp if (error != NULL)
919 c09a553d 2018-03-12 stsp goto done;
920 c09a553d 2018-03-12 stsp
921 b65ae19a 2018-04-24 stsp printf("Now shut up and hack\n");
922 c09a553d 2018-03-12 stsp
923 c09a553d 2018-03-12 stsp done:
924 8069f636 2019-01-12 stsp free(commit_id_str);
925 76089277 2018-04-01 stsp free(repo_path);
926 507dc3bb 2018-12-29 stsp free(worktree_path);
927 507dc3bb 2018-12-29 stsp return error;
928 507dc3bb 2018-12-29 stsp }
929 507dc3bb 2018-12-29 stsp
930 507dc3bb 2018-12-29 stsp __dead static void
931 507dc3bb 2018-12-29 stsp usage_update(void)
932 507dc3bb 2018-12-29 stsp {
933 024e9686 2019-05-14 stsp fprintf(stderr, "usage: %s update [-b branch] [-c commit] [path]\n",
934 507dc3bb 2018-12-29 stsp getprogname());
935 507dc3bb 2018-12-29 stsp exit(1);
936 507dc3bb 2018-12-29 stsp }
937 507dc3bb 2018-12-29 stsp
938 1ee397ad 2019-07-12 stsp static const struct got_error *
939 507dc3bb 2018-12-29 stsp update_progress(void *arg, unsigned char status, const char *path)
940 507dc3bb 2018-12-29 stsp {
941 784955db 2019-01-12 stsp int *did_something = arg;
942 784955db 2019-01-12 stsp
943 507dc3bb 2018-12-29 stsp if (status == GOT_STATUS_EXISTS)
944 1ee397ad 2019-07-12 stsp return NULL;
945 507dc3bb 2018-12-29 stsp
946 1545c615 2019-02-10 stsp *did_something = 1;
947 a484d721 2019-06-10 stsp
948 a484d721 2019-06-10 stsp /* Base commit bump happens silently. */
949 a484d721 2019-06-10 stsp if (status == GOT_STATUS_BUMP_BASE)
950 1ee397ad 2019-07-12 stsp return NULL;
951 a484d721 2019-06-10 stsp
952 507dc3bb 2018-12-29 stsp while (path[0] == '/')
953 507dc3bb 2018-12-29 stsp path++;
954 507dc3bb 2018-12-29 stsp printf("%c %s\n", status, path);
955 1ee397ad 2019-07-12 stsp return NULL;
956 be7061eb 2018-12-30 stsp }
957 be7061eb 2018-12-30 stsp
958 be7061eb 2018-12-30 stsp static const struct got_error *
959 a1fb16d8 2019-05-24 stsp switch_head_ref(struct got_reference *head_ref,
960 a1fb16d8 2019-05-24 stsp struct got_object_id *commit_id, struct got_worktree *worktree,
961 a1fb16d8 2019-05-24 stsp struct got_repository *repo)
962 a1fb16d8 2019-05-24 stsp {
963 a1fb16d8 2019-05-24 stsp const struct got_error *err = NULL;
964 a1fb16d8 2019-05-24 stsp char *base_id_str;
965 a1fb16d8 2019-05-24 stsp int ref_has_moved = 0;
966 a1fb16d8 2019-05-24 stsp
967 a1fb16d8 2019-05-24 stsp /* Trivial case: switching between two different references. */
968 a1fb16d8 2019-05-24 stsp if (strcmp(got_ref_get_name(head_ref),
969 a1fb16d8 2019-05-24 stsp got_worktree_get_head_ref_name(worktree)) != 0) {
970 a1fb16d8 2019-05-24 stsp printf("Switching work tree from %s to %s\n",
971 a1fb16d8 2019-05-24 stsp got_worktree_get_head_ref_name(worktree),
972 a1fb16d8 2019-05-24 stsp got_ref_get_name(head_ref));
973 a1fb16d8 2019-05-24 stsp return got_worktree_set_head_ref(worktree, head_ref);
974 a1fb16d8 2019-05-24 stsp }
975 a1fb16d8 2019-05-24 stsp
976 a1fb16d8 2019-05-24 stsp err = check_linear_ancestry(commit_id,
977 a1fb16d8 2019-05-24 stsp got_worktree_get_base_commit_id(worktree), repo);
978 a1fb16d8 2019-05-24 stsp if (err) {
979 a1fb16d8 2019-05-24 stsp if (err->code != GOT_ERR_ANCESTRY)
980 a1fb16d8 2019-05-24 stsp return err;
981 a1fb16d8 2019-05-24 stsp ref_has_moved = 1;
982 a1fb16d8 2019-05-24 stsp }
983 a1fb16d8 2019-05-24 stsp if (!ref_has_moved)
984 a1fb16d8 2019-05-24 stsp return NULL;
985 a1fb16d8 2019-05-24 stsp
986 a1fb16d8 2019-05-24 stsp /* Switching to a rebased branch with the same reference name. */
987 a1fb16d8 2019-05-24 stsp err = got_object_id_str(&base_id_str,
988 a1fb16d8 2019-05-24 stsp got_worktree_get_base_commit_id(worktree));
989 a1fb16d8 2019-05-24 stsp if (err)
990 a1fb16d8 2019-05-24 stsp return err;
991 a1fb16d8 2019-05-24 stsp printf("Reference %s now points at a different branch\n",
992 a1fb16d8 2019-05-24 stsp got_worktree_get_head_ref_name(worktree));
993 a1fb16d8 2019-05-24 stsp printf("Switching work tree from %s to %s\n", base_id_str,
994 a1fb16d8 2019-05-24 stsp got_worktree_get_head_ref_name(worktree));
995 a1fb16d8 2019-05-24 stsp return NULL;
996 a1fb16d8 2019-05-24 stsp }
997 a1fb16d8 2019-05-24 stsp
998 a1fb16d8 2019-05-24 stsp static const struct got_error *
999 507dc3bb 2018-12-29 stsp cmd_update(int argc, char *argv[])
1000 507dc3bb 2018-12-29 stsp {
1001 507dc3bb 2018-12-29 stsp const struct got_error *error = NULL;
1002 507dc3bb 2018-12-29 stsp struct got_repository *repo = NULL;
1003 507dc3bb 2018-12-29 stsp struct got_worktree *worktree = NULL;
1004 c4cdcb68 2019-04-03 stsp char *worktree_path = NULL, *path = NULL;
1005 507dc3bb 2018-12-29 stsp struct got_object_id *commit_id = NULL;
1006 9c4b8182 2019-01-02 stsp char *commit_id_str = NULL;
1007 024e9686 2019-05-14 stsp const char *branch_name = NULL;
1008 024e9686 2019-05-14 stsp struct got_reference *head_ref = NULL;
1009 7d5807f4 2019-07-11 stsp int ch, did_something = 0, rebase_in_progress;
1010 507dc3bb 2018-12-29 stsp
1011 024e9686 2019-05-14 stsp while ((ch = getopt(argc, argv, "b:c:")) != -1) {
1012 507dc3bb 2018-12-29 stsp switch (ch) {
1013 024e9686 2019-05-14 stsp case 'b':
1014 024e9686 2019-05-14 stsp branch_name = optarg;
1015 024e9686 2019-05-14 stsp break;
1016 507dc3bb 2018-12-29 stsp case 'c':
1017 9c4b8182 2019-01-02 stsp commit_id_str = strdup(optarg);
1018 9c4b8182 2019-01-02 stsp if (commit_id_str == NULL)
1019 638f9024 2019-05-13 stsp return got_error_from_errno("strdup");
1020 507dc3bb 2018-12-29 stsp break;
1021 507dc3bb 2018-12-29 stsp default:
1022 2deda0b9 2019-03-07 stsp usage_update();
1023 507dc3bb 2018-12-29 stsp /* NOTREACHED */
1024 507dc3bb 2018-12-29 stsp }
1025 507dc3bb 2018-12-29 stsp }
1026 507dc3bb 2018-12-29 stsp
1027 507dc3bb 2018-12-29 stsp argc -= optind;
1028 507dc3bb 2018-12-29 stsp argv += optind;
1029 507dc3bb 2018-12-29 stsp
1030 507dc3bb 2018-12-29 stsp #ifndef PROFILE
1031 68ed9ba5 2019-02-10 stsp if (pledge("stdio rpath wpath cpath fattr flock proc exec sendfd "
1032 68ed9ba5 2019-02-10 stsp "unveil", NULL) == -1)
1033 507dc3bb 2018-12-29 stsp err(1, "pledge");
1034 507dc3bb 2018-12-29 stsp #endif
1035 c4cdcb68 2019-04-03 stsp worktree_path = getcwd(NULL, 0);
1036 c4cdcb68 2019-04-03 stsp if (worktree_path == NULL) {
1037 638f9024 2019-05-13 stsp error = got_error_from_errno("getcwd");
1038 c4cdcb68 2019-04-03 stsp goto done;
1039 c4cdcb68 2019-04-03 stsp }
1040 c4cdcb68 2019-04-03 stsp error = got_worktree_open(&worktree, worktree_path);
1041 7d5807f4 2019-07-11 stsp if (error)
1042 7d5807f4 2019-07-11 stsp goto done;
1043 7d5807f4 2019-07-11 stsp
1044 7d5807f4 2019-07-11 stsp error = got_worktree_rebase_in_progress(&rebase_in_progress, worktree);
1045 c4cdcb68 2019-04-03 stsp if (error)
1046 c4cdcb68 2019-04-03 stsp goto done;
1047 7d5807f4 2019-07-11 stsp if (rebase_in_progress) {
1048 7d5807f4 2019-07-11 stsp error = got_error(GOT_ERR_REBASING);
1049 7d5807f4 2019-07-11 stsp goto done;
1050 7d5807f4 2019-07-11 stsp }
1051 c4cdcb68 2019-04-03 stsp
1052 507dc3bb 2018-12-29 stsp if (argc == 0) {
1053 c4cdcb68 2019-04-03 stsp path = strdup("");
1054 c4cdcb68 2019-04-03 stsp if (path == NULL) {
1055 638f9024 2019-05-13 stsp error = got_error_from_errno("strdup");
1056 507dc3bb 2018-12-29 stsp goto done;
1057 507dc3bb 2018-12-29 stsp }
1058 507dc3bb 2018-12-29 stsp } else if (argc == 1) {
1059 c4cdcb68 2019-04-03 stsp error = got_worktree_resolve_path(&path, worktree, argv[0]);
1060 c4cdcb68 2019-04-03 stsp if (error)
1061 507dc3bb 2018-12-29 stsp goto done;
1062 507dc3bb 2018-12-29 stsp } else
1063 507dc3bb 2018-12-29 stsp usage_update();
1064 507dc3bb 2018-12-29 stsp
1065 507dc3bb 2018-12-29 stsp error = got_repo_open(&repo, got_worktree_get_repo_path(worktree));
1066 507dc3bb 2018-12-29 stsp if (error != NULL)
1067 507dc3bb 2018-12-29 stsp goto done;
1068 507dc3bb 2018-12-29 stsp
1069 97430839 2019-03-11 stsp error = apply_unveil(got_repo_get_path(repo), 0,
1070 314a6357 2019-05-13 stsp got_worktree_get_root_path(worktree), 0);
1071 0266afb7 2019-01-04 stsp if (error)
1072 0266afb7 2019-01-04 stsp goto done;
1073 0266afb7 2019-01-04 stsp
1074 a1fb16d8 2019-05-24 stsp error = got_ref_open(&head_ref, repo, branch_name ? branch_name :
1075 a1fb16d8 2019-05-24 stsp got_worktree_get_head_ref_name(worktree), 0);
1076 024e9686 2019-05-14 stsp if (error != NULL)
1077 024e9686 2019-05-14 stsp goto done;
1078 507dc3bb 2018-12-29 stsp if (commit_id_str == NULL) {
1079 507dc3bb 2018-12-29 stsp error = got_ref_resolve(&commit_id, repo, head_ref);
1080 9c4b8182 2019-01-02 stsp if (error != NULL)
1081 9c4b8182 2019-01-02 stsp goto done;
1082 9c4b8182 2019-01-02 stsp error = got_object_id_str(&commit_id_str, commit_id);
1083 507dc3bb 2018-12-29 stsp if (error != NULL)
1084 507dc3bb 2018-12-29 stsp goto done;
1085 507dc3bb 2018-12-29 stsp } else {
1086 e09a504c 2019-06-28 stsp error = got_repo_match_object_id_prefix(&commit_id,
1087 dd88155e 2019-06-29 stsp commit_id_str, GOT_OBJ_TYPE_COMMIT, repo);
1088 507dc3bb 2018-12-29 stsp if (error != NULL)
1089 a297e751 2019-07-11 stsp goto done;
1090 a297e751 2019-07-11 stsp free(commit_id_str);
1091 a297e751 2019-07-11 stsp error = got_object_id_str(&commit_id_str, commit_id);
1092 a297e751 2019-07-11 stsp if (error)
1093 507dc3bb 2018-12-29 stsp goto done;
1094 507dc3bb 2018-12-29 stsp }
1095 35c965b2 2018-12-29 stsp
1096 a1fb16d8 2019-05-24 stsp if (branch_name) {
1097 024e9686 2019-05-14 stsp struct got_object_id *head_commit_id;
1098 024e9686 2019-05-14 stsp if (strlen(path) != 0) {
1099 a1fb16d8 2019-05-24 stsp fprintf(stderr, "%s: switching between branches "
1100 a1fb16d8 2019-05-24 stsp "requires that the entire work tree "
1101 024e9686 2019-05-14 stsp "gets updated, not just '%s'\n",
1102 024e9686 2019-05-14 stsp getprogname(), path);
1103 024e9686 2019-05-14 stsp error = got_error(GOT_ERR_BAD_PATH);
1104 024e9686 2019-05-14 stsp goto done;
1105 024e9686 2019-05-14 stsp }
1106 024e9686 2019-05-14 stsp error = got_ref_resolve(&head_commit_id, repo, head_ref);
1107 024e9686 2019-05-14 stsp if (error)
1108 024e9686 2019-05-14 stsp goto done;
1109 024e9686 2019-05-14 stsp error = check_linear_ancestry(commit_id, head_commit_id, repo);
1110 a367ff0f 2019-05-14 stsp free(head_commit_id);
1111 024e9686 2019-05-14 stsp if (error != NULL)
1112 024e9686 2019-05-14 stsp goto done;
1113 a367ff0f 2019-05-14 stsp error = check_same_branch(commit_id, head_ref, repo);
1114 a367ff0f 2019-05-14 stsp if (error)
1115 a367ff0f 2019-05-14 stsp goto done;
1116 a1fb16d8 2019-05-24 stsp error = switch_head_ref(head_ref, commit_id, worktree, repo);
1117 024e9686 2019-05-14 stsp if (error)
1118 024e9686 2019-05-14 stsp goto done;
1119 024e9686 2019-05-14 stsp } else {
1120 024e9686 2019-05-14 stsp error = check_linear_ancestry(commit_id,
1121 024e9686 2019-05-14 stsp got_worktree_get_base_commit_id(worktree), repo);
1122 a367ff0f 2019-05-14 stsp if (error != NULL) {
1123 a367ff0f 2019-05-14 stsp if (error->code == GOT_ERR_ANCESTRY)
1124 a367ff0f 2019-05-14 stsp error = got_error(GOT_ERR_BRANCH_MOVED);
1125 024e9686 2019-05-14 stsp goto done;
1126 a367ff0f 2019-05-14 stsp }
1127 a367ff0f 2019-05-14 stsp error = check_same_branch(commit_id, head_ref, repo);
1128 a367ff0f 2019-05-14 stsp if (error)
1129 a367ff0f 2019-05-14 stsp goto done;
1130 024e9686 2019-05-14 stsp }
1131 507dc3bb 2018-12-29 stsp
1132 507dc3bb 2018-12-29 stsp if (got_object_id_cmp(got_worktree_get_base_commit_id(worktree),
1133 507dc3bb 2018-12-29 stsp commit_id) != 0) {
1134 507dc3bb 2018-12-29 stsp error = got_worktree_set_base_commit_id(worktree, repo,
1135 507dc3bb 2018-12-29 stsp commit_id);
1136 507dc3bb 2018-12-29 stsp if (error)
1137 507dc3bb 2018-12-29 stsp goto done;
1138 507dc3bb 2018-12-29 stsp }
1139 507dc3bb 2018-12-29 stsp
1140 c4cdcb68 2019-04-03 stsp error = got_worktree_checkout_files(worktree, path, repo,
1141 6bad629b 2019-02-04 stsp update_progress, &did_something, check_cancelled, NULL);
1142 507dc3bb 2018-12-29 stsp if (error != NULL)
1143 507dc3bb 2018-12-29 stsp goto done;
1144 9c4b8182 2019-01-02 stsp
1145 784955db 2019-01-12 stsp if (did_something)
1146 784955db 2019-01-12 stsp printf("Updated to commit %s\n", commit_id_str);
1147 784955db 2019-01-12 stsp else
1148 784955db 2019-01-12 stsp printf("Already up-to-date\n");
1149 507dc3bb 2018-12-29 stsp done:
1150 c09a553d 2018-03-12 stsp free(worktree_path);
1151 c4cdcb68 2019-04-03 stsp free(path);
1152 507dc3bb 2018-12-29 stsp free(commit_id);
1153 9c4b8182 2019-01-02 stsp free(commit_id_str);
1154 c09a553d 2018-03-12 stsp return error;
1155 c09a553d 2018-03-12 stsp }
1156 c09a553d 2018-03-12 stsp
1157 f42b1b34 2018-03-12 stsp static const struct got_error *
1158 79109fed 2018-03-27 stsp print_patch(struct got_commit_object *commit, struct got_object_id *id,
1159 c0cc5c62 2018-10-18 stsp int diff_context, struct got_repository *repo)
1160 5c860e29 2018-03-12 stsp {
1161 f42b1b34 2018-03-12 stsp const struct got_error *err = NULL;
1162 79109fed 2018-03-27 stsp struct got_tree_object *tree1 = NULL, *tree2;
1163 79f35eb3 2018-06-11 stsp struct got_object_qid *qid;
1164 0f2b3dca 2018-12-22 stsp char *id_str1 = NULL, *id_str2;
1165 aaa13589 2019-06-01 stsp struct got_diff_blob_output_unidiff_arg arg;
1166 79109fed 2018-03-27 stsp
1167 45d799e2 2018-12-23 stsp err = got_object_open_as_tree(&tree2, repo,
1168 45d799e2 2018-12-23 stsp got_object_commit_get_tree_id(commit));
1169 79109fed 2018-03-27 stsp if (err)
1170 79109fed 2018-03-27 stsp return err;
1171 79109fed 2018-03-27 stsp
1172 45d799e2 2018-12-23 stsp qid = SIMPLEQ_FIRST(got_object_commit_get_parent_ids(commit));
1173 79f35eb3 2018-06-11 stsp if (qid != NULL) {
1174 79109fed 2018-03-27 stsp struct got_commit_object *pcommit;
1175 79109fed 2018-03-27 stsp
1176 117e771c 2018-07-23 stsp err = got_object_open_as_commit(&pcommit, repo, qid->id);
1177 79109fed 2018-03-27 stsp if (err)
1178 79109fed 2018-03-27 stsp return err;
1179 79109fed 2018-03-27 stsp
1180 45d799e2 2018-12-23 stsp err = got_object_open_as_tree(&tree1, repo,
1181 45d799e2 2018-12-23 stsp got_object_commit_get_tree_id(pcommit));
1182 79109fed 2018-03-27 stsp got_object_commit_close(pcommit);
1183 0f2b3dca 2018-12-22 stsp if (err)
1184 0f2b3dca 2018-12-22 stsp return err;
1185 0f2b3dca 2018-12-22 stsp
1186 0f2b3dca 2018-12-22 stsp err = got_object_id_str(&id_str1, qid->id);
1187 79109fed 2018-03-27 stsp if (err)
1188 79109fed 2018-03-27 stsp return err;
1189 79109fed 2018-03-27 stsp }
1190 79109fed 2018-03-27 stsp
1191 0f2b3dca 2018-12-22 stsp err = got_object_id_str(&id_str2, id);
1192 0f2b3dca 2018-12-22 stsp if (err)
1193 0f2b3dca 2018-12-22 stsp goto done;
1194 0f2b3dca 2018-12-22 stsp
1195 56765ebb 2018-12-23 stsp printf("diff %s %s\n", id_str1 ? id_str1 : "/dev/null", id_str2);
1196 aaa13589 2019-06-01 stsp arg.diff_context = diff_context;
1197 aaa13589 2019-06-01 stsp arg.outfile = stdout;
1198 aaa13589 2019-06-01 stsp err = got_diff_tree(tree1, tree2, "", "", repo,
1199 aaa13589 2019-06-01 stsp got_diff_blob_output_unidiff, &arg);
1200 0f2b3dca 2018-12-22 stsp done:
1201 79109fed 2018-03-27 stsp if (tree1)
1202 79109fed 2018-03-27 stsp got_object_tree_close(tree1);
1203 79109fed 2018-03-27 stsp got_object_tree_close(tree2);
1204 0f2b3dca 2018-12-22 stsp free(id_str1);
1205 0f2b3dca 2018-12-22 stsp free(id_str2);
1206 79109fed 2018-03-27 stsp return err;
1207 79109fed 2018-03-27 stsp }
1208 79109fed 2018-03-27 stsp
1209 4bb494d5 2018-06-16 stsp static char *
1210 6c281f94 2018-06-11 stsp get_datestr(time_t *time, char *datebuf)
1211 6c281f94 2018-06-11 stsp {
1212 6c281f94 2018-06-11 stsp char *p, *s = ctime_r(time, datebuf);
1213 6c281f94 2018-06-11 stsp p = strchr(s, '\n');
1214 6c281f94 2018-06-11 stsp if (p)
1215 6c281f94 2018-06-11 stsp *p = '\0';
1216 6c281f94 2018-06-11 stsp return s;
1217 6c281f94 2018-06-11 stsp }
1218 6c281f94 2018-06-11 stsp
1219 79109fed 2018-03-27 stsp static const struct got_error *
1220 79109fed 2018-03-27 stsp print_commit(struct got_commit_object *commit, struct got_object_id *id,
1221 199a4027 2019-02-02 stsp struct got_repository *repo, int show_patch, int diff_context,
1222 199a4027 2019-02-02 stsp struct got_reflist_head *refs)
1223 79109fed 2018-03-27 stsp {
1224 79109fed 2018-03-27 stsp const struct got_error *err = NULL;
1225 621015ac 2018-07-23 stsp char *id_str, *datestr, *logmsg0, *logmsg, *line;
1226 6c281f94 2018-06-11 stsp char datebuf[26];
1227 45d799e2 2018-12-23 stsp time_t committer_time;
1228 45d799e2 2018-12-23 stsp const char *author, *committer;
1229 199a4027 2019-02-02 stsp char *refs_str = NULL;
1230 199a4027 2019-02-02 stsp struct got_reflist_entry *re;
1231 5c860e29 2018-03-12 stsp
1232 199a4027 2019-02-02 stsp SIMPLEQ_FOREACH(re, refs, entry) {
1233 199a4027 2019-02-02 stsp char *s;
1234 199a4027 2019-02-02 stsp const char *name;
1235 199a4027 2019-02-02 stsp if (got_object_id_cmp(re->id, id) != 0)
1236 199a4027 2019-02-02 stsp continue;
1237 199a4027 2019-02-02 stsp name = got_ref_get_name(re->ref);
1238 d9498b20 2019-02-05 stsp if (strcmp(name, GOT_REF_HEAD) == 0)
1239 d9498b20 2019-02-05 stsp continue;
1240 199a4027 2019-02-02 stsp if (strncmp(name, "refs/", 5) == 0)
1241 199a4027 2019-02-02 stsp name += 5;
1242 7143d404 2019-03-12 stsp if (strncmp(name, "got/", 4) == 0)
1243 7143d404 2019-03-12 stsp continue;
1244 e34f9ed6 2019-02-02 stsp if (strncmp(name, "heads/", 6) == 0)
1245 e34f9ed6 2019-02-02 stsp name += 6;
1246 141c2bff 2019-02-04 stsp if (strncmp(name, "remotes/", 8) == 0)
1247 141c2bff 2019-02-04 stsp name += 8;
1248 199a4027 2019-02-02 stsp s = refs_str;
1249 199a4027 2019-02-02 stsp if (asprintf(&refs_str, "%s%s%s", s ? s : "", s ? ", " : "",
1250 199a4027 2019-02-02 stsp name) == -1) {
1251 638f9024 2019-05-13 stsp err = got_error_from_errno("asprintf");
1252 199a4027 2019-02-02 stsp free(s);
1253 199a4027 2019-02-02 stsp break;
1254 199a4027 2019-02-02 stsp }
1255 199a4027 2019-02-02 stsp free(s);
1256 199a4027 2019-02-02 stsp }
1257 832c249c 2018-06-10 stsp err = got_object_id_str(&id_str, id);
1258 8bf5b3c9 2018-03-17 stsp if (err)
1259 8bf5b3c9 2018-03-17 stsp return err;
1260 788c352e 2018-06-16 stsp
1261 33d869be 2018-12-22 stsp printf("-----------------------------------------------\n");
1262 199a4027 2019-02-02 stsp printf("commit %s%s%s%s\n", id_str, refs_str ? " (" : "",
1263 199a4027 2019-02-02 stsp refs_str ? refs_str : "", refs_str ? ")" : "");
1264 832c249c 2018-06-10 stsp free(id_str);
1265 d3d493d7 2019-02-21 stsp id_str = NULL;
1266 d3d493d7 2019-02-21 stsp free(refs_str);
1267 d3d493d7 2019-02-21 stsp refs_str = NULL;
1268 45d799e2 2018-12-23 stsp printf("from: %s\n", got_object_commit_get_author(commit));
1269 45d799e2 2018-12-23 stsp committer_time = got_object_commit_get_committer_time(commit);
1270 45d799e2 2018-12-23 stsp datestr = get_datestr(&committer_time, datebuf);
1271 dab5fe87 2018-09-14 stsp printf("date: %s UTC\n", datestr);
1272 45d799e2 2018-12-23 stsp author = got_object_commit_get_author(commit);
1273 45d799e2 2018-12-23 stsp committer = got_object_commit_get_committer(commit);
1274 45d799e2 2018-12-23 stsp if (strcmp(author, committer) != 0)
1275 45d799e2 2018-12-23 stsp printf("via: %s\n", committer);
1276 45d799e2 2018-12-23 stsp if (got_object_commit_get_nparents(commit) > 1) {
1277 45d799e2 2018-12-23 stsp const struct got_object_id_queue *parent_ids;
1278 79f35eb3 2018-06-11 stsp struct got_object_qid *qid;
1279 3fe1abad 2018-06-10 stsp int n = 1;
1280 45d799e2 2018-12-23 stsp parent_ids = got_object_commit_get_parent_ids(commit);
1281 45d799e2 2018-12-23 stsp SIMPLEQ_FOREACH(qid, parent_ids, entry) {
1282 79f35eb3 2018-06-11 stsp err = got_object_id_str(&id_str, qid->id);
1283 a0603db2 2018-06-10 stsp if (err)
1284 a0603db2 2018-06-10 stsp return err;
1285 3fe1abad 2018-06-10 stsp printf("parent %d: %s\n", n++, id_str);
1286 a0603db2 2018-06-10 stsp free(id_str);
1287 a0603db2 2018-06-10 stsp }
1288 a0603db2 2018-06-10 stsp }
1289 832c249c 2018-06-10 stsp
1290 45d799e2 2018-12-23 stsp logmsg0 = strdup(got_object_commit_get_logmsg(commit));
1291 621015ac 2018-07-23 stsp if (logmsg0 == NULL)
1292 638f9024 2019-05-13 stsp return got_error_from_errno("strdup");
1293 8bf5b3c9 2018-03-17 stsp
1294 621015ac 2018-07-23 stsp logmsg = logmsg0;
1295 832c249c 2018-06-10 stsp do {
1296 832c249c 2018-06-10 stsp line = strsep(&logmsg, "\n");
1297 832c249c 2018-06-10 stsp if (line)
1298 832c249c 2018-06-10 stsp printf(" %s\n", line);
1299 832c249c 2018-06-10 stsp } while (line);
1300 621015ac 2018-07-23 stsp free(logmsg0);
1301 832c249c 2018-06-10 stsp
1302 971751ac 2018-03-27 stsp if (show_patch) {
1303 c0cc5c62 2018-10-18 stsp err = print_patch(commit, id, diff_context, repo);
1304 971751ac 2018-03-27 stsp if (err == 0)
1305 971751ac 2018-03-27 stsp printf("\n");
1306 971751ac 2018-03-27 stsp }
1307 07862c20 2018-09-15 stsp
1308 cbe7f848 2019-02-11 stsp if (fflush(stdout) != 0 && err == NULL)
1309 638f9024 2019-05-13 stsp err = got_error_from_errno("fflush");
1310 07862c20 2018-09-15 stsp return err;
1311 f42b1b34 2018-03-12 stsp }
1312 5c860e29 2018-03-12 stsp
1313 f42b1b34 2018-03-12 stsp static const struct got_error *
1314 15a94983 2018-12-23 stsp print_commits(struct got_object_id *root_id, struct got_repository *repo,
1315 15a94983 2018-12-23 stsp char *path, int show_patch, int diff_context, int limit,
1316 199a4027 2019-02-02 stsp int first_parent_traversal, struct got_reflist_head *refs)
1317 f42b1b34 2018-03-12 stsp {
1318 f42b1b34 2018-03-12 stsp const struct got_error *err;
1319 372ccdbb 2018-06-10 stsp struct got_commit_graph *graph;
1320 372ccdbb 2018-06-10 stsp
1321 31cedeaf 2018-09-15 stsp err = got_commit_graph_open(&graph, root_id, path,
1322 31cedeaf 2018-09-15 stsp first_parent_traversal, repo);
1323 f42b1b34 2018-03-12 stsp if (err)
1324 f42b1b34 2018-03-12 stsp return err;
1325 31cedeaf 2018-09-15 stsp err = got_commit_graph_iter_start(graph, root_id, repo);
1326 372ccdbb 2018-06-10 stsp if (err)
1327 fcc85cad 2018-07-22 stsp goto done;
1328 656b1f76 2019-05-11 jcs for (;;) {
1329 372ccdbb 2018-06-10 stsp struct got_commit_object *commit;
1330 372ccdbb 2018-06-10 stsp struct got_object_id *id;
1331 8bf5b3c9 2018-03-17 stsp
1332 84453469 2018-11-11 stsp if (sigint_received || sigpipe_received)
1333 84453469 2018-11-11 stsp break;
1334 84453469 2018-11-11 stsp
1335 b43fbaa0 2018-06-11 stsp err = got_commit_graph_iter_next(&id, graph);
1336 372ccdbb 2018-06-10 stsp if (err) {
1337 9ba79e04 2018-06-11 stsp if (err->code == GOT_ERR_ITER_COMPLETED) {
1338 9ba79e04 2018-06-11 stsp err = NULL;
1339 9ba79e04 2018-06-11 stsp break;
1340 9ba79e04 2018-06-11 stsp }
1341 372ccdbb 2018-06-10 stsp if (err->code != GOT_ERR_ITER_NEED_MORE)
1342 372ccdbb 2018-06-10 stsp break;
1343 31cedeaf 2018-09-15 stsp err = got_commit_graph_fetch_commits(graph, 1, repo);
1344 372ccdbb 2018-06-10 stsp if (err)
1345 372ccdbb 2018-06-10 stsp break;
1346 372ccdbb 2018-06-10 stsp else
1347 372ccdbb 2018-06-10 stsp continue;
1348 7e665116 2018-04-02 stsp }
1349 b43fbaa0 2018-06-11 stsp if (id == NULL)
1350 7e665116 2018-04-02 stsp break;
1351 b43fbaa0 2018-06-11 stsp
1352 f8e900f3 2018-06-11 stsp err = got_object_open_as_commit(&commit, repo, id);
1353 b43fbaa0 2018-06-11 stsp if (err)
1354 fcc85cad 2018-07-22 stsp break;
1355 199a4027 2019-02-02 stsp err = print_commit(commit, id, repo, show_patch, diff_context,
1356 199a4027 2019-02-02 stsp refs);
1357 b43fbaa0 2018-06-11 stsp got_object_commit_close(commit);
1358 372ccdbb 2018-06-10 stsp if (err || (limit && --limit == 0))
1359 7e665116 2018-04-02 stsp break;
1360 31cedeaf 2018-09-15 stsp }
1361 fcc85cad 2018-07-22 stsp done:
1362 372ccdbb 2018-06-10 stsp got_commit_graph_close(graph);
1363 f42b1b34 2018-03-12 stsp return err;
1364 f42b1b34 2018-03-12 stsp }
1365 5c860e29 2018-03-12 stsp
1366 4ed7e80c 2018-05-20 stsp __dead static void
1367 6f3d1eb0 2018-03-12 stsp usage_log(void)
1368 6f3d1eb0 2018-03-12 stsp {
1369 499d7ecc 2019-05-22 stsp fprintf(stderr, "usage: %s log [-b] [-c commit] [-C number] [ -l N ] [-p] "
1370 04ca23f4 2018-07-16 stsp "[-r repository-path] [path]\n", getprogname());
1371 6f3d1eb0 2018-03-12 stsp exit(1);
1372 6f3d1eb0 2018-03-12 stsp }
1373 6f3d1eb0 2018-03-12 stsp
1374 4ed7e80c 2018-05-20 stsp static const struct got_error *
1375 f42b1b34 2018-03-12 stsp cmd_log(int argc, char *argv[])
1376 f42b1b34 2018-03-12 stsp {
1377 f42b1b34 2018-03-12 stsp const struct got_error *error;
1378 04ca23f4 2018-07-16 stsp struct got_repository *repo = NULL;
1379 cffc0aa4 2019-02-05 stsp struct got_worktree *worktree = NULL;
1380 15a94983 2018-12-23 stsp struct got_commit_object *commit = NULL;
1381 3235492e 2018-04-01 stsp struct got_object_id *id = NULL;
1382 04ca23f4 2018-07-16 stsp char *repo_path = NULL, *path = NULL, *cwd = NULL, *in_repo_path = NULL;
1383 d142fc45 2018-04-01 stsp char *start_commit = NULL;
1384 c0cc5c62 2018-10-18 stsp int diff_context = 3, ch;
1385 1fd6d7ea 2018-06-13 stsp int show_patch = 0, limit = 0, first_parent_traversal = 0;
1386 64a96a6d 2018-04-01 stsp const char *errstr;
1387 199a4027 2019-02-02 stsp struct got_reflist_head refs;
1388 1b3893a2 2019-03-18 stsp
1389 1b3893a2 2019-03-18 stsp SIMPLEQ_INIT(&refs);
1390 5c860e29 2018-03-12 stsp
1391 6715a751 2018-03-16 stsp #ifndef PROFILE
1392 6098196c 2019-01-04 stsp if (pledge("stdio rpath wpath cpath flock proc exec sendfd unveil",
1393 6098196c 2019-01-04 stsp NULL)
1394 ad242220 2018-09-08 stsp == -1)
1395 f42b1b34 2018-03-12 stsp err(1, "pledge");
1396 6715a751 2018-03-16 stsp #endif
1397 79109fed 2018-03-27 stsp
1398 499d7ecc 2019-05-22 stsp while ((ch = getopt(argc, argv, "bpc:C:l:r:")) != -1) {
1399 79109fed 2018-03-27 stsp switch (ch) {
1400 499d7ecc 2019-05-22 stsp case 'b':
1401 499d7ecc 2019-05-22 stsp first_parent_traversal = 1;
1402 499d7ecc 2019-05-22 stsp break;
1403 79109fed 2018-03-27 stsp case 'p':
1404 79109fed 2018-03-27 stsp show_patch = 1;
1405 d142fc45 2018-04-01 stsp break;
1406 d142fc45 2018-04-01 stsp case 'c':
1407 d142fc45 2018-04-01 stsp start_commit = optarg;
1408 64a96a6d 2018-04-01 stsp break;
1409 c0cc5c62 2018-10-18 stsp case 'C':
1410 4a8520aa 2018-10-18 stsp diff_context = strtonum(optarg, 0, GOT_DIFF_MAX_CONTEXT,
1411 4a8520aa 2018-10-18 stsp &errstr);
1412 c0cc5c62 2018-10-18 stsp if (errstr != NULL)
1413 c0cc5c62 2018-10-18 stsp err(1, "-C option %s", errstr);
1414 c0cc5c62 2018-10-18 stsp break;
1415 64a96a6d 2018-04-01 stsp case 'l':
1416 64a96a6d 2018-04-01 stsp limit = strtonum(optarg, 1, INT_MAX, &errstr);
1417 64a96a6d 2018-04-01 stsp if (errstr != NULL)
1418 64a96a6d 2018-04-01 stsp err(1, "-l option %s", errstr);
1419 a0603db2 2018-06-10 stsp break;
1420 04ca23f4 2018-07-16 stsp case 'r':
1421 04ca23f4 2018-07-16 stsp repo_path = realpath(optarg, NULL);
1422 04ca23f4 2018-07-16 stsp if (repo_path == NULL)
1423 04ca23f4 2018-07-16 stsp err(1, "-r option");
1424 7fbaa4f3 2019-05-11 stsp got_path_strip_trailing_slashes(repo_path);
1425 04ca23f4 2018-07-16 stsp break;
1426 79109fed 2018-03-27 stsp default:
1427 2deda0b9 2019-03-07 stsp usage_log();
1428 79109fed 2018-03-27 stsp /* NOTREACHED */
1429 79109fed 2018-03-27 stsp }
1430 79109fed 2018-03-27 stsp }
1431 79109fed 2018-03-27 stsp
1432 79109fed 2018-03-27 stsp argc -= optind;
1433 79109fed 2018-03-27 stsp argv += optind;
1434 f42b1b34 2018-03-12 stsp
1435 04ca23f4 2018-07-16 stsp cwd = getcwd(NULL, 0);
1436 04ca23f4 2018-07-16 stsp if (cwd == NULL) {
1437 638f9024 2019-05-13 stsp error = got_error_from_errno("getcwd");
1438 04ca23f4 2018-07-16 stsp goto done;
1439 04ca23f4 2018-07-16 stsp }
1440 cffc0aa4 2019-02-05 stsp
1441 cffc0aa4 2019-02-05 stsp error = got_worktree_open(&worktree, cwd);
1442 cffc0aa4 2019-02-05 stsp if (error && error->code != GOT_ERR_NOT_WORKTREE)
1443 cffc0aa4 2019-02-05 stsp goto done;
1444 cffc0aa4 2019-02-05 stsp error = NULL;
1445 cffc0aa4 2019-02-05 stsp
1446 e7301579 2019-03-18 stsp if (argc == 0) {
1447 cbd1af7a 2019-03-18 stsp path = strdup("");
1448 e7301579 2019-03-18 stsp if (path == NULL) {
1449 638f9024 2019-05-13 stsp error = got_error_from_errno("strdup");
1450 cbd1af7a 2019-03-18 stsp goto done;
1451 e7301579 2019-03-18 stsp }
1452 e7301579 2019-03-18 stsp } else if (argc == 1) {
1453 e7301579 2019-03-18 stsp if (worktree) {
1454 e7301579 2019-03-18 stsp error = got_worktree_resolve_path(&path, worktree,
1455 e7301579 2019-03-18 stsp argv[0]);
1456 e7301579 2019-03-18 stsp if (error)
1457 e7301579 2019-03-18 stsp goto done;
1458 e7301579 2019-03-18 stsp } else {
1459 e7301579 2019-03-18 stsp path = strdup(argv[0]);
1460 e7301579 2019-03-18 stsp if (path == NULL) {
1461 638f9024 2019-05-13 stsp error = got_error_from_errno("strdup");
1462 e7301579 2019-03-18 stsp goto done;
1463 e7301579 2019-03-18 stsp }
1464 e7301579 2019-03-18 stsp }
1465 cbd1af7a 2019-03-18 stsp } else
1466 cbd1af7a 2019-03-18 stsp usage_log();
1467 cbd1af7a 2019-03-18 stsp
1468 5486daa2 2019-05-11 stsp if (repo_path == NULL) {
1469 5486daa2 2019-05-11 stsp repo_path = worktree ?
1470 5486daa2 2019-05-11 stsp strdup(got_worktree_get_repo_path(worktree)) : strdup(cwd);
1471 5486daa2 2019-05-11 stsp }
1472 04ca23f4 2018-07-16 stsp if (repo_path == NULL) {
1473 638f9024 2019-05-13 stsp error = got_error_from_errno("strdup");
1474 cffc0aa4 2019-02-05 stsp goto done;
1475 04ca23f4 2018-07-16 stsp }
1476 6098196c 2019-01-04 stsp
1477 f42b1b34 2018-03-12 stsp error = got_repo_open(&repo, repo_path);
1478 d7d4f210 2018-03-12 stsp if (error != NULL)
1479 04ca23f4 2018-07-16 stsp goto done;
1480 f42b1b34 2018-03-12 stsp
1481 c02c541e 2019-03-29 stsp error = apply_unveil(got_repo_get_path(repo), 1,
1482 314a6357 2019-05-13 stsp worktree ? got_worktree_get_root_path(worktree) : NULL, 0);
1483 c02c541e 2019-03-29 stsp if (error)
1484 c02c541e 2019-03-29 stsp goto done;
1485 c02c541e 2019-03-29 stsp
1486 d142fc45 2018-04-01 stsp if (start_commit == NULL) {
1487 3235492e 2018-04-01 stsp struct got_reference *head_ref;
1488 1cc14b9f 2019-05-14 stsp error = got_ref_open(&head_ref, repo,
1489 1cc14b9f 2019-05-14 stsp worktree ? got_worktree_get_head_ref_name(worktree)
1490 1cc14b9f 2019-05-14 stsp : GOT_REF_HEAD, 0);
1491 3235492e 2018-04-01 stsp if (error != NULL)
1492 3235492e 2018-04-01 stsp return error;
1493 3235492e 2018-04-01 stsp error = got_ref_resolve(&id, repo, head_ref);
1494 3235492e 2018-04-01 stsp got_ref_close(head_ref);
1495 3235492e 2018-04-01 stsp if (error != NULL)
1496 3235492e 2018-04-01 stsp return error;
1497 15a94983 2018-12-23 stsp error = got_object_open_as_commit(&commit, repo, id);
1498 3235492e 2018-04-01 stsp } else {
1499 d1f2edc9 2018-06-13 stsp struct got_reference *ref;
1500 2f17228e 2019-05-12 stsp error = got_ref_open(&ref, repo, start_commit, 0);
1501 3235492e 2018-04-01 stsp if (error == NULL) {
1502 1caad61b 2019-02-01 stsp int obj_type;
1503 d1f2edc9 2018-06-13 stsp error = got_ref_resolve(&id, repo, ref);
1504 d1f2edc9 2018-06-13 stsp got_ref_close(ref);
1505 d1f2edc9 2018-06-13 stsp if (error != NULL)
1506 1caad61b 2019-02-01 stsp goto done;
1507 1caad61b 2019-02-01 stsp error = got_object_get_type(&obj_type, repo, id);
1508 1caad61b 2019-02-01 stsp if (error != NULL)
1509 1caad61b 2019-02-01 stsp goto done;
1510 1caad61b 2019-02-01 stsp if (obj_type == GOT_OBJ_TYPE_TAG) {
1511 1caad61b 2019-02-01 stsp struct got_tag_object *tag;
1512 1caad61b 2019-02-01 stsp error = got_object_open_as_tag(&tag, repo, id);
1513 1caad61b 2019-02-01 stsp if (error != NULL)
1514 1caad61b 2019-02-01 stsp goto done;
1515 1caad61b 2019-02-01 stsp if (got_object_tag_get_object_type(tag) !=
1516 1caad61b 2019-02-01 stsp GOT_OBJ_TYPE_COMMIT) {
1517 1caad61b 2019-02-01 stsp got_object_tag_close(tag);
1518 1caad61b 2019-02-01 stsp error = got_error(GOT_ERR_OBJ_TYPE);
1519 1caad61b 2019-02-01 stsp goto done;
1520 1caad61b 2019-02-01 stsp }
1521 1caad61b 2019-02-01 stsp free(id);
1522 1caad61b 2019-02-01 stsp id = got_object_id_dup(
1523 1caad61b 2019-02-01 stsp got_object_tag_get_object_id(tag));
1524 1caad61b 2019-02-01 stsp if (id == NULL)
1525 638f9024 2019-05-13 stsp error = got_error_from_errno(
1526 230a42bd 2019-05-11 jcs "got_object_id_dup");
1527 1caad61b 2019-02-01 stsp got_object_tag_close(tag);
1528 1caad61b 2019-02-01 stsp if (error)
1529 1caad61b 2019-02-01 stsp goto done;
1530 1caad61b 2019-02-01 stsp } else if (obj_type != GOT_OBJ_TYPE_COMMIT) {
1531 1caad61b 2019-02-01 stsp error = got_error(GOT_ERR_OBJ_TYPE);
1532 1caad61b 2019-02-01 stsp goto done;
1533 1caad61b 2019-02-01 stsp }
1534 15a94983 2018-12-23 stsp error = got_object_open_as_commit(&commit, repo, id);
1535 d1f2edc9 2018-06-13 stsp if (error != NULL)
1536 1caad61b 2019-02-01 stsp goto done;
1537 d1f2edc9 2018-06-13 stsp }
1538 15a94983 2018-12-23 stsp if (commit == NULL) {
1539 e09a504c 2019-06-28 stsp error = got_repo_match_object_id_prefix(&id,
1540 dd88155e 2019-06-29 stsp start_commit, GOT_OBJ_TYPE_COMMIT, repo);
1541 d1f2edc9 2018-06-13 stsp if (error != NULL)
1542 d1f2edc9 2018-06-13 stsp return error;
1543 3235492e 2018-04-01 stsp }
1544 3235492e 2018-04-01 stsp }
1545 d7d4f210 2018-03-12 stsp if (error != NULL)
1546 04ca23f4 2018-07-16 stsp goto done;
1547 04ca23f4 2018-07-16 stsp
1548 23721109 2018-10-22 stsp error = got_repo_map_path(&in_repo_path, repo, path, 1);
1549 04ca23f4 2018-07-16 stsp if (error != NULL)
1550 04ca23f4 2018-07-16 stsp goto done;
1551 04ca23f4 2018-07-16 stsp if (in_repo_path) {
1552 04ca23f4 2018-07-16 stsp free(path);
1553 04ca23f4 2018-07-16 stsp path = in_repo_path;
1554 04ca23f4 2018-07-16 stsp }
1555 199a4027 2019-02-02 stsp
1556 199a4027 2019-02-02 stsp error = got_ref_list(&refs, repo);
1557 199a4027 2019-02-02 stsp if (error)
1558 199a4027 2019-02-02 stsp goto done;
1559 04ca23f4 2018-07-16 stsp
1560 15a94983 2018-12-23 stsp error = print_commits(id, repo, path, show_patch,
1561 199a4027 2019-02-02 stsp diff_context, limit, first_parent_traversal, &refs);
1562 04ca23f4 2018-07-16 stsp done:
1563 04ca23f4 2018-07-16 stsp free(path);
1564 04ca23f4 2018-07-16 stsp free(repo_path);
1565 04ca23f4 2018-07-16 stsp free(cwd);
1566 f42b1b34 2018-03-12 stsp free(id);
1567 cffc0aa4 2019-02-05 stsp if (worktree)
1568 cffc0aa4 2019-02-05 stsp got_worktree_close(worktree);
1569 ad242220 2018-09-08 stsp if (repo) {
1570 ad242220 2018-09-08 stsp const struct got_error *repo_error;
1571 ad242220 2018-09-08 stsp repo_error = got_repo_close(repo);
1572 ad242220 2018-09-08 stsp if (error == NULL)
1573 ad242220 2018-09-08 stsp error = repo_error;
1574 ad242220 2018-09-08 stsp }
1575 e2e879a0 2019-03-11 stsp got_ref_list_free(&refs);
1576 8bf5b3c9 2018-03-17 stsp return error;
1577 5c860e29 2018-03-12 stsp }
1578 5c860e29 2018-03-12 stsp
1579 4ed7e80c 2018-05-20 stsp __dead static void
1580 3f8b7d6a 2018-04-01 stsp usage_diff(void)
1581 3f8b7d6a 2018-04-01 stsp {
1582 b72f483a 2019-02-05 stsp fprintf(stderr, "usage: %s diff [-C number] [-r repository-path] "
1583 927df6b7 2019-02-10 stsp "[object1 object2 | path]\n", getprogname());
1584 3f8b7d6a 2018-04-01 stsp exit(1);
1585 b00d56cd 2018-04-01 stsp }
1586 b00d56cd 2018-04-01 stsp
1587 b72f483a 2019-02-05 stsp struct print_diff_arg {
1588 b72f483a 2019-02-05 stsp struct got_repository *repo;
1589 b72f483a 2019-02-05 stsp struct got_worktree *worktree;
1590 b72f483a 2019-02-05 stsp int diff_context;
1591 3fc0c068 2019-02-10 stsp const char *id_str;
1592 3fc0c068 2019-02-10 stsp int header_shown;
1593 b72f483a 2019-02-05 stsp };
1594 b72f483a 2019-02-05 stsp
1595 4ed7e80c 2018-05-20 stsp static const struct got_error *
1596 b72f483a 2019-02-05 stsp print_diff(void *arg, unsigned char status, const char *path,
1597 016a88dd 2019-05-13 stsp struct got_object_id *blob_id, struct got_object_id *commit_id)
1598 b72f483a 2019-02-05 stsp {
1599 b72f483a 2019-02-05 stsp struct print_diff_arg *a = arg;
1600 b72f483a 2019-02-05 stsp const struct got_error *err = NULL;
1601 b72f483a 2019-02-05 stsp struct got_blob_object *blob1 = NULL;
1602 b72f483a 2019-02-05 stsp FILE *f2 = NULL;
1603 b72f483a 2019-02-05 stsp char *abspath = NULL;
1604 b72f483a 2019-02-05 stsp struct stat sb;
1605 b72f483a 2019-02-05 stsp
1606 2ec1f75b 2019-03-26 stsp if (status != GOT_STATUS_MODIFY && status != GOT_STATUS_ADD &&
1607 7154f6ce 2019-03-27 stsp status != GOT_STATUS_DELETE && status != GOT_STATUS_CONFLICT)
1608 b72f483a 2019-02-05 stsp return NULL;
1609 3fc0c068 2019-02-10 stsp
1610 3fc0c068 2019-02-10 stsp if (!a->header_shown) {
1611 3fc0c068 2019-02-10 stsp printf("diff %s %s\n", a->id_str,
1612 3fc0c068 2019-02-10 stsp got_worktree_get_root_path(a->worktree));
1613 3fc0c068 2019-02-10 stsp a->header_shown = 1;
1614 3fc0c068 2019-02-10 stsp }
1615 b72f483a 2019-02-05 stsp
1616 7154f6ce 2019-03-27 stsp if (status != GOT_STATUS_ADD) {
1617 016a88dd 2019-05-13 stsp err = got_object_open_as_blob(&blob1, a->repo, blob_id, 8192);
1618 d00136be 2019-03-26 stsp if (err)
1619 d00136be 2019-03-26 stsp goto done;
1620 b72f483a 2019-02-05 stsp
1621 d00136be 2019-03-26 stsp }
1622 d00136be 2019-03-26 stsp
1623 7154f6ce 2019-03-27 stsp if (status != GOT_STATUS_DELETE) {
1624 2ec1f75b 2019-03-26 stsp if (asprintf(&abspath, "%s/%s",
1625 2ec1f75b 2019-03-26 stsp got_worktree_get_root_path(a->worktree), path) == -1) {
1626 638f9024 2019-05-13 stsp err = got_error_from_errno("asprintf");
1627 2ec1f75b 2019-03-26 stsp goto done;
1628 2ec1f75b 2019-03-26 stsp }
1629 b72f483a 2019-02-05 stsp
1630 2ec1f75b 2019-03-26 stsp f2 = fopen(abspath, "r");
1631 2ec1f75b 2019-03-26 stsp if (f2 == NULL) {
1632 638f9024 2019-05-13 stsp err = got_error_from_errno2("fopen", abspath);
1633 2ec1f75b 2019-03-26 stsp goto done;
1634 2ec1f75b 2019-03-26 stsp }
1635 2ec1f75b 2019-03-26 stsp if (lstat(abspath, &sb) == -1) {
1636 638f9024 2019-05-13 stsp err = got_error_from_errno2("lstat", abspath);
1637 2ec1f75b 2019-03-26 stsp goto done;
1638 2ec1f75b 2019-03-26 stsp }
1639 2ec1f75b 2019-03-26 stsp } else
1640 2ec1f75b 2019-03-26 stsp sb.st_size = 0;
1641 b72f483a 2019-02-05 stsp
1642 b72f483a 2019-02-05 stsp err = got_diff_blob_file(blob1, f2, sb.st_size, path, a->diff_context,
1643 b72f483a 2019-02-05 stsp stdout);
1644 b72f483a 2019-02-05 stsp done:
1645 b72f483a 2019-02-05 stsp if (blob1)
1646 b72f483a 2019-02-05 stsp got_object_blob_close(blob1);
1647 fb43ecf1 2019-02-11 stsp if (f2 && fclose(f2) != 0 && err == NULL)
1648 638f9024 2019-05-13 stsp err = got_error_from_errno("fclose");
1649 b72f483a 2019-02-05 stsp free(abspath);
1650 927df6b7 2019-02-10 stsp return err;
1651 927df6b7 2019-02-10 stsp }
1652 927df6b7 2019-02-10 stsp
1653 927df6b7 2019-02-10 stsp static const struct got_error *
1654 b00d56cd 2018-04-01 stsp cmd_diff(int argc, char *argv[])
1655 b00d56cd 2018-04-01 stsp {
1656 b00d56cd 2018-04-01 stsp const struct got_error *error;
1657 b00d56cd 2018-04-01 stsp struct got_repository *repo = NULL;
1658 b72f483a 2019-02-05 stsp struct got_worktree *worktree = NULL;
1659 b72f483a 2019-02-05 stsp char *cwd = NULL, *repo_path = NULL;
1660 15a94983 2018-12-23 stsp struct got_object_id *id1 = NULL, *id2 = NULL;
1661 cd628e99 2019-05-28 stsp const char *id_str1 = NULL, *id_str2 = NULL;
1662 5e70831e 2019-05-28 stsp char *label1 = NULL, *label2 = NULL;
1663 15a94983 2018-12-23 stsp int type1, type2;
1664 c0cc5c62 2018-10-18 stsp int diff_context = 3, ch;
1665 c0cc5c62 2018-10-18 stsp const char *errstr;
1666 927df6b7 2019-02-10 stsp char *path = NULL;
1667 b00d56cd 2018-04-01 stsp
1668 b00d56cd 2018-04-01 stsp #ifndef PROFILE
1669 25eccc22 2019-01-04 stsp if (pledge("stdio rpath wpath cpath flock proc exec sendfd unveil",
1670 25eccc22 2019-01-04 stsp NULL) == -1)
1671 b00d56cd 2018-04-01 stsp err(1, "pledge");
1672 b00d56cd 2018-04-01 stsp #endif
1673 b00d56cd 2018-04-01 stsp
1674 b72f483a 2019-02-05 stsp while ((ch = getopt(argc, argv, "C:r:")) != -1) {
1675 b00d56cd 2018-04-01 stsp switch (ch) {
1676 c0cc5c62 2018-10-18 stsp case 'C':
1677 c0cc5c62 2018-10-18 stsp diff_context = strtonum(optarg, 1, INT_MAX, &errstr);
1678 c0cc5c62 2018-10-18 stsp if (errstr != NULL)
1679 c0cc5c62 2018-10-18 stsp err(1, "-C option %s", errstr);
1680 c0cc5c62 2018-10-18 stsp break;
1681 b72f483a 2019-02-05 stsp case 'r':
1682 b72f483a 2019-02-05 stsp repo_path = realpath(optarg, NULL);
1683 b72f483a 2019-02-05 stsp if (repo_path == NULL)
1684 b72f483a 2019-02-05 stsp err(1, "-r option");
1685 7fbaa4f3 2019-05-11 stsp got_path_strip_trailing_slashes(repo_path);
1686 b72f483a 2019-02-05 stsp break;
1687 b00d56cd 2018-04-01 stsp default:
1688 2deda0b9 2019-03-07 stsp usage_diff();
1689 b00d56cd 2018-04-01 stsp /* NOTREACHED */
1690 b00d56cd 2018-04-01 stsp }
1691 b00d56cd 2018-04-01 stsp }
1692 b00d56cd 2018-04-01 stsp
1693 b00d56cd 2018-04-01 stsp argc -= optind;
1694 b00d56cd 2018-04-01 stsp argv += optind;
1695 b00d56cd 2018-04-01 stsp
1696 b72f483a 2019-02-05 stsp cwd = getcwd(NULL, 0);
1697 b72f483a 2019-02-05 stsp if (cwd == NULL) {
1698 638f9024 2019-05-13 stsp error = got_error_from_errno("getcwd");
1699 b72f483a 2019-02-05 stsp goto done;
1700 b72f483a 2019-02-05 stsp }
1701 927df6b7 2019-02-10 stsp error = got_worktree_open(&worktree, cwd);
1702 927df6b7 2019-02-10 stsp if (error && error->code != GOT_ERR_NOT_WORKTREE)
1703 927df6b7 2019-02-10 stsp goto done;
1704 927df6b7 2019-02-10 stsp if (argc <= 1) {
1705 927df6b7 2019-02-10 stsp if (worktree == NULL) {
1706 927df6b7 2019-02-10 stsp error = got_error(GOT_ERR_NOT_WORKTREE);
1707 927df6b7 2019-02-10 stsp goto done;
1708 927df6b7 2019-02-10 stsp }
1709 b72f483a 2019-02-05 stsp if (repo_path)
1710 b72f483a 2019-02-05 stsp errx(1,
1711 b72f483a 2019-02-05 stsp "-r option can't be used when diffing a work tree");
1712 b72f483a 2019-02-05 stsp repo_path = strdup(got_worktree_get_repo_path(worktree));
1713 927df6b7 2019-02-10 stsp if (repo_path == NULL) {
1714 638f9024 2019-05-13 stsp error = got_error_from_errno("strdup");
1715 927df6b7 2019-02-10 stsp goto done;
1716 927df6b7 2019-02-10 stsp }
1717 927df6b7 2019-02-10 stsp if (argc == 1) {
1718 6c7ab921 2019-03-18 stsp error = got_worktree_resolve_path(&path, worktree,
1719 cbd1af7a 2019-03-18 stsp argv[0]);
1720 927df6b7 2019-02-10 stsp if (error)
1721 927df6b7 2019-02-10 stsp goto done;
1722 927df6b7 2019-02-10 stsp } else {
1723 927df6b7 2019-02-10 stsp path = strdup("");
1724 927df6b7 2019-02-10 stsp if (path == NULL) {
1725 638f9024 2019-05-13 stsp error = got_error_from_errno("strdup");
1726 927df6b7 2019-02-10 stsp goto done;
1727 927df6b7 2019-02-10 stsp }
1728 927df6b7 2019-02-10 stsp }
1729 b72f483a 2019-02-05 stsp } else if (argc == 2) {
1730 15a94983 2018-12-23 stsp id_str1 = argv[0];
1731 15a94983 2018-12-23 stsp id_str2 = argv[1];
1732 30db809c 2019-06-05 stsp if (worktree && repo_path == NULL) {
1733 30db809c 2019-06-05 stsp repo_path =
1734 30db809c 2019-06-05 stsp strdup(got_worktree_get_repo_path(worktree));
1735 30db809c 2019-06-05 stsp if (repo_path == NULL) {
1736 30db809c 2019-06-05 stsp error = got_error_from_errno("strdup");
1737 30db809c 2019-06-05 stsp goto done;
1738 30db809c 2019-06-05 stsp }
1739 30db809c 2019-06-05 stsp }
1740 b00d56cd 2018-04-01 stsp } else
1741 b00d56cd 2018-04-01 stsp usage_diff();
1742 25eccc22 2019-01-04 stsp
1743 b72f483a 2019-02-05 stsp if (repo_path == NULL) {
1744 b72f483a 2019-02-05 stsp repo_path = getcwd(NULL, 0);
1745 b72f483a 2019-02-05 stsp if (repo_path == NULL)
1746 638f9024 2019-05-13 stsp return got_error_from_errno("getcwd");
1747 b72f483a 2019-02-05 stsp }
1748 b00d56cd 2018-04-01 stsp
1749 b00d56cd 2018-04-01 stsp error = got_repo_open(&repo, repo_path);
1750 76089277 2018-04-01 stsp free(repo_path);
1751 b00d56cd 2018-04-01 stsp if (error != NULL)
1752 b00d56cd 2018-04-01 stsp goto done;
1753 b00d56cd 2018-04-01 stsp
1754 c02c541e 2019-03-29 stsp error = apply_unveil(got_repo_get_path(repo), 1,
1755 314a6357 2019-05-13 stsp worktree ? got_worktree_get_root_path(worktree) : NULL, 0);
1756 c02c541e 2019-03-29 stsp if (error)
1757 c02c541e 2019-03-29 stsp goto done;
1758 c02c541e 2019-03-29 stsp
1759 30db809c 2019-06-05 stsp if (argc <= 1) {
1760 b72f483a 2019-02-05 stsp struct print_diff_arg arg;
1761 b72f483a 2019-02-05 stsp char *id_str;
1762 b72f483a 2019-02-05 stsp error = got_object_id_str(&id_str,
1763 b72f483a 2019-02-05 stsp got_worktree_get_base_commit_id(worktree));
1764 b72f483a 2019-02-05 stsp if (error)
1765 b72f483a 2019-02-05 stsp goto done;
1766 b72f483a 2019-02-05 stsp arg.repo = repo;
1767 b72f483a 2019-02-05 stsp arg.worktree = worktree;
1768 b72f483a 2019-02-05 stsp arg.diff_context = diff_context;
1769 3fc0c068 2019-02-10 stsp arg.id_str = id_str;
1770 3fc0c068 2019-02-10 stsp arg.header_shown = 0;
1771 b72f483a 2019-02-05 stsp
1772 927df6b7 2019-02-10 stsp error = got_worktree_status(worktree, path, repo, print_diff,
1773 b72f483a 2019-02-05 stsp &arg, check_cancelled, NULL);
1774 b72f483a 2019-02-05 stsp free(id_str);
1775 b72f483a 2019-02-05 stsp goto done;
1776 b72f483a 2019-02-05 stsp }
1777 b72f483a 2019-02-05 stsp
1778 dd88155e 2019-06-29 stsp error = got_repo_match_object_id_prefix(&id1, id_str1,
1779 dd88155e 2019-06-29 stsp GOT_OBJ_TYPE_ANY, repo);
1780 e02e74af 2019-05-28 stsp if (error) {
1781 e02e74af 2019-05-28 stsp struct got_reference *ref;
1782 e02e74af 2019-05-28 stsp if (error->code != GOT_ERR_BAD_OBJ_ID_STR)
1783 e02e74af 2019-05-28 stsp goto done;
1784 e02e74af 2019-05-28 stsp error = got_ref_open(&ref, repo, id_str1, 0);
1785 e02e74af 2019-05-28 stsp if (error != NULL)
1786 e02e74af 2019-05-28 stsp goto done;
1787 5e70831e 2019-05-28 stsp label1 = strdup(got_ref_get_name(ref));
1788 5e70831e 2019-05-28 stsp if (label1 == NULL) {
1789 5e70831e 2019-05-28 stsp error = got_error_from_errno("strdup");
1790 5e70831e 2019-05-28 stsp goto done;
1791 5e70831e 2019-05-28 stsp }
1792 e02e74af 2019-05-28 stsp error = got_ref_resolve(&id1, repo, ref);
1793 e02e74af 2019-05-28 stsp got_ref_close(ref);
1794 e02e74af 2019-05-28 stsp if (error != NULL)
1795 5e70831e 2019-05-28 stsp goto done;
1796 5e70831e 2019-05-28 stsp } else {
1797 a297e751 2019-07-11 stsp error = got_object_id_str(&label1, id1);
1798 5e70831e 2019-05-28 stsp if (label1 == NULL) {
1799 5e70831e 2019-05-28 stsp error = got_error_from_errno("strdup");
1800 e02e74af 2019-05-28 stsp goto done;
1801 5e70831e 2019-05-28 stsp }
1802 e02e74af 2019-05-28 stsp }
1803 b00d56cd 2018-04-01 stsp
1804 dd88155e 2019-06-29 stsp error = got_repo_match_object_id_prefix(&id2, id_str2,
1805 dd88155e 2019-06-29 stsp GOT_OBJ_TYPE_ANY, repo);
1806 e02e74af 2019-05-28 stsp if (error) {
1807 e02e74af 2019-05-28 stsp struct got_reference *ref;
1808 e02e74af 2019-05-28 stsp if (error->code != GOT_ERR_BAD_OBJ_ID_STR)
1809 e02e74af 2019-05-28 stsp goto done;
1810 e02e74af 2019-05-28 stsp error = got_ref_open(&ref, repo, id_str2, 0);
1811 e02e74af 2019-05-28 stsp if (error != NULL)
1812 5e70831e 2019-05-28 stsp goto done;
1813 5e70831e 2019-05-28 stsp label2 = strdup(got_ref_get_name(ref));
1814 5e70831e 2019-05-28 stsp if (label2 == NULL) {
1815 5e70831e 2019-05-28 stsp error = got_error_from_errno("strdup");
1816 e02e74af 2019-05-28 stsp goto done;
1817 5e70831e 2019-05-28 stsp }
1818 e02e74af 2019-05-28 stsp error = got_ref_resolve(&id2, repo, ref);
1819 e02e74af 2019-05-28 stsp got_ref_close(ref);
1820 e02e74af 2019-05-28 stsp if (error != NULL)
1821 e02e74af 2019-05-28 stsp goto done;
1822 5e70831e 2019-05-28 stsp } else {
1823 a297e751 2019-07-11 stsp error = got_object_id_str(&label2, id2);
1824 5e70831e 2019-05-28 stsp if (label2 == NULL) {
1825 5e70831e 2019-05-28 stsp error = got_error_from_errno("strdup");
1826 5e70831e 2019-05-28 stsp goto done;
1827 5e70831e 2019-05-28 stsp }
1828 e02e74af 2019-05-28 stsp }
1829 b00d56cd 2018-04-01 stsp
1830 15a94983 2018-12-23 stsp error = got_object_get_type(&type1, repo, id1);
1831 15a94983 2018-12-23 stsp if (error)
1832 15a94983 2018-12-23 stsp goto done;
1833 15a94983 2018-12-23 stsp
1834 15a94983 2018-12-23 stsp error = got_object_get_type(&type2, repo, id2);
1835 15a94983 2018-12-23 stsp if (error)
1836 15a94983 2018-12-23 stsp goto done;
1837 15a94983 2018-12-23 stsp
1838 15a94983 2018-12-23 stsp if (type1 != type2) {
1839 b00d56cd 2018-04-01 stsp error = got_error(GOT_ERR_OBJ_TYPE);
1840 b00d56cd 2018-04-01 stsp goto done;
1841 b00d56cd 2018-04-01 stsp }
1842 b00d56cd 2018-04-01 stsp
1843 15a94983 2018-12-23 stsp switch (type1) {
1844 b00d56cd 2018-04-01 stsp case GOT_OBJ_TYPE_BLOB:
1845 15a94983 2018-12-23 stsp error = got_diff_objects_as_blobs(id1, id2, NULL, NULL,
1846 54156555 2018-12-24 stsp diff_context, repo, stdout);
1847 b00d56cd 2018-04-01 stsp break;
1848 b00d56cd 2018-04-01 stsp case GOT_OBJ_TYPE_TREE:
1849 15a94983 2018-12-23 stsp error = got_diff_objects_as_trees(id1, id2, "", "",
1850 54156555 2018-12-24 stsp diff_context, repo, stdout);
1851 b00d56cd 2018-04-01 stsp break;
1852 b00d56cd 2018-04-01 stsp case GOT_OBJ_TYPE_COMMIT:
1853 5e70831e 2019-05-28 stsp printf("diff %s %s\n", label1, label2);
1854 15a94983 2018-12-23 stsp error = got_diff_objects_as_commits(id1, id2, diff_context,
1855 c0cc5c62 2018-10-18 stsp repo, stdout);
1856 b00d56cd 2018-04-01 stsp break;
1857 b00d56cd 2018-04-01 stsp default:
1858 b00d56cd 2018-04-01 stsp error = got_error(GOT_ERR_OBJ_TYPE);
1859 b00d56cd 2018-04-01 stsp }
1860 b00d56cd 2018-04-01 stsp
1861 b00d56cd 2018-04-01 stsp done:
1862 5e70831e 2019-05-28 stsp free(label1);
1863 5e70831e 2019-05-28 stsp free(label2);
1864 15a94983 2018-12-23 stsp free(id1);
1865 15a94983 2018-12-23 stsp free(id2);
1866 927df6b7 2019-02-10 stsp free(path);
1867 b72f483a 2019-02-05 stsp if (worktree)
1868 b72f483a 2019-02-05 stsp got_worktree_close(worktree);
1869 ad242220 2018-09-08 stsp if (repo) {
1870 ad242220 2018-09-08 stsp const struct got_error *repo_error;
1871 ad242220 2018-09-08 stsp repo_error = got_repo_close(repo);
1872 ad242220 2018-09-08 stsp if (error == NULL)
1873 ad242220 2018-09-08 stsp error = repo_error;
1874 ad242220 2018-09-08 stsp }
1875 b00d56cd 2018-04-01 stsp return error;
1876 404c43c4 2018-06-21 stsp }
1877 404c43c4 2018-06-21 stsp
1878 404c43c4 2018-06-21 stsp __dead static void
1879 404c43c4 2018-06-21 stsp usage_blame(void)
1880 404c43c4 2018-06-21 stsp {
1881 9270e621 2019-02-05 stsp fprintf(stderr,
1882 9270e621 2019-02-05 stsp "usage: %s blame [-c commit] [-r repository-path] path\n",
1883 404c43c4 2018-06-21 stsp getprogname());
1884 404c43c4 2018-06-21 stsp exit(1);
1885 b00d56cd 2018-04-01 stsp }
1886 b00d56cd 2018-04-01 stsp
1887 404c43c4 2018-06-21 stsp static const struct got_error *
1888 404c43c4 2018-06-21 stsp cmd_blame(int argc, char *argv[])
1889 404c43c4 2018-06-21 stsp {
1890 404c43c4 2018-06-21 stsp const struct got_error *error;
1891 404c43c4 2018-06-21 stsp struct got_repository *repo = NULL;
1892 0c06baac 2019-02-05 stsp struct got_worktree *worktree = NULL;
1893 66bea077 2018-08-02 stsp char *path, *cwd = NULL, *repo_path = NULL, *in_repo_path = NULL;
1894 404c43c4 2018-06-21 stsp struct got_object_id *commit_id = NULL;
1895 404c43c4 2018-06-21 stsp char *commit_id_str = NULL;
1896 404c43c4 2018-06-21 stsp int ch;
1897 404c43c4 2018-06-21 stsp
1898 404c43c4 2018-06-21 stsp #ifndef PROFILE
1899 36e2fb66 2019-01-04 stsp if (pledge("stdio rpath wpath cpath flock proc exec sendfd unveil",
1900 36e2fb66 2019-01-04 stsp NULL) == -1)
1901 404c43c4 2018-06-21 stsp err(1, "pledge");
1902 404c43c4 2018-06-21 stsp #endif
1903 404c43c4 2018-06-21 stsp
1904 66bea077 2018-08-02 stsp while ((ch = getopt(argc, argv, "c:r:")) != -1) {
1905 404c43c4 2018-06-21 stsp switch (ch) {
1906 404c43c4 2018-06-21 stsp case 'c':
1907 404c43c4 2018-06-21 stsp commit_id_str = optarg;
1908 404c43c4 2018-06-21 stsp break;
1909 66bea077 2018-08-02 stsp case 'r':
1910 66bea077 2018-08-02 stsp repo_path = realpath(optarg, NULL);
1911 66bea077 2018-08-02 stsp if (repo_path == NULL)
1912 66bea077 2018-08-02 stsp err(1, "-r option");
1913 7fbaa4f3 2019-05-11 stsp got_path_strip_trailing_slashes(repo_path);
1914 66bea077 2018-08-02 stsp break;
1915 404c43c4 2018-06-21 stsp default:
1916 2deda0b9 2019-03-07 stsp usage_blame();
1917 404c43c4 2018-06-21 stsp /* NOTREACHED */
1918 404c43c4 2018-06-21 stsp }
1919 404c43c4 2018-06-21 stsp }
1920 404c43c4 2018-06-21 stsp
1921 404c43c4 2018-06-21 stsp argc -= optind;
1922 404c43c4 2018-06-21 stsp argv += optind;
1923 404c43c4 2018-06-21 stsp
1924 a39318fd 2018-08-02 stsp if (argc == 1)
1925 404c43c4 2018-06-21 stsp path = argv[0];
1926 a39318fd 2018-08-02 stsp else
1927 404c43c4 2018-06-21 stsp usage_blame();
1928 404c43c4 2018-06-21 stsp
1929 66bea077 2018-08-02 stsp cwd = getcwd(NULL, 0);
1930 66bea077 2018-08-02 stsp if (cwd == NULL) {
1931 638f9024 2019-05-13 stsp error = got_error_from_errno("getcwd");
1932 66bea077 2018-08-02 stsp goto done;
1933 66bea077 2018-08-02 stsp }
1934 66bea077 2018-08-02 stsp if (repo_path == NULL) {
1935 0c06baac 2019-02-05 stsp error = got_worktree_open(&worktree, cwd);
1936 0c06baac 2019-02-05 stsp if (error && error->code != GOT_ERR_NOT_WORKTREE)
1937 66bea077 2018-08-02 stsp goto done;
1938 0c06baac 2019-02-05 stsp else
1939 0c06baac 2019-02-05 stsp error = NULL;
1940 0c06baac 2019-02-05 stsp if (worktree) {
1941 0c06baac 2019-02-05 stsp repo_path =
1942 0c06baac 2019-02-05 stsp strdup(got_worktree_get_repo_path(worktree));
1943 0c06baac 2019-02-05 stsp if (repo_path == NULL)
1944 638f9024 2019-05-13 stsp error = got_error_from_errno("strdup");
1945 0c06baac 2019-02-05 stsp if (error)
1946 0c06baac 2019-02-05 stsp goto done;
1947 0c06baac 2019-02-05 stsp } else {
1948 0c06baac 2019-02-05 stsp repo_path = strdup(cwd);
1949 0c06baac 2019-02-05 stsp if (repo_path == NULL) {
1950 638f9024 2019-05-13 stsp error = got_error_from_errno("strdup");
1951 0c06baac 2019-02-05 stsp goto done;
1952 0c06baac 2019-02-05 stsp }
1953 66bea077 2018-08-02 stsp }
1954 66bea077 2018-08-02 stsp }
1955 36e2fb66 2019-01-04 stsp
1956 c02c541e 2019-03-29 stsp error = got_repo_open(&repo, repo_path);
1957 c02c541e 2019-03-29 stsp if (error != NULL)
1958 36e2fb66 2019-01-04 stsp goto done;
1959 66bea077 2018-08-02 stsp
1960 314a6357 2019-05-13 stsp error = apply_unveil(got_repo_get_path(repo), 1, NULL, 0);
1961 c02c541e 2019-03-29 stsp if (error)
1962 404c43c4 2018-06-21 stsp goto done;
1963 404c43c4 2018-06-21 stsp
1964 0c06baac 2019-02-05 stsp if (worktree) {
1965 6efaaa2d 2019-02-05 stsp const char *prefix = got_worktree_get_path_prefix(worktree);
1966 0c06baac 2019-02-05 stsp char *p, *worktree_subdir = cwd +
1967 0c06baac 2019-02-05 stsp strlen(got_worktree_get_root_path(worktree));
1968 6efaaa2d 2019-02-05 stsp if (asprintf(&p, "%s%s%s%s%s",
1969 6efaaa2d 2019-02-05 stsp prefix, (strcmp(prefix, "/") != 0) ? "/" : "",
1970 6efaaa2d 2019-02-05 stsp worktree_subdir, worktree_subdir[0] ? "/" : "",
1971 6efaaa2d 2019-02-05 stsp path) == -1) {
1972 638f9024 2019-05-13 stsp error = got_error_from_errno("asprintf");
1973 0c06baac 2019-02-05 stsp goto done;
1974 0c06baac 2019-02-05 stsp }
1975 6efaaa2d 2019-02-05 stsp error = got_repo_map_path(&in_repo_path, repo, p, 0);
1976 0c06baac 2019-02-05 stsp free(p);
1977 0c06baac 2019-02-05 stsp } else {
1978 0c06baac 2019-02-05 stsp error = got_repo_map_path(&in_repo_path, repo, path, 1);
1979 0c06baac 2019-02-05 stsp }
1980 0c06baac 2019-02-05 stsp if (error)
1981 66bea077 2018-08-02 stsp goto done;
1982 66bea077 2018-08-02 stsp
1983 404c43c4 2018-06-21 stsp if (commit_id_str == NULL) {
1984 404c43c4 2018-06-21 stsp struct got_reference *head_ref;
1985 2f17228e 2019-05-12 stsp error = got_ref_open(&head_ref, repo, GOT_REF_HEAD, 0);
1986 404c43c4 2018-06-21 stsp if (error != NULL)
1987 66bea077 2018-08-02 stsp goto done;
1988 404c43c4 2018-06-21 stsp error = got_ref_resolve(&commit_id, repo, head_ref);
1989 404c43c4 2018-06-21 stsp got_ref_close(head_ref);
1990 404c43c4 2018-06-21 stsp if (error != NULL)
1991 66bea077 2018-08-02 stsp goto done;
1992 404c43c4 2018-06-21 stsp } else {
1993 e09a504c 2019-06-28 stsp error = got_repo_match_object_id_prefix(&commit_id,
1994 dd88155e 2019-06-29 stsp commit_id_str, GOT_OBJ_TYPE_COMMIT, repo);
1995 404c43c4 2018-06-21 stsp if (error != NULL)
1996 66bea077 2018-08-02 stsp goto done;
1997 404c43c4 2018-06-21 stsp }
1998 404c43c4 2018-06-21 stsp
1999 66bea077 2018-08-02 stsp error = got_blame(in_repo_path, commit_id, repo, stdout);
2000 404c43c4 2018-06-21 stsp done:
2001 66bea077 2018-08-02 stsp free(in_repo_path);
2002 66bea077 2018-08-02 stsp free(repo_path);
2003 66bea077 2018-08-02 stsp free(cwd);
2004 404c43c4 2018-06-21 stsp free(commit_id);
2005 0c06baac 2019-02-05 stsp if (worktree)
2006 0c06baac 2019-02-05 stsp got_worktree_close(worktree);
2007 ad242220 2018-09-08 stsp if (repo) {
2008 ad242220 2018-09-08 stsp const struct got_error *repo_error;
2009 ad242220 2018-09-08 stsp repo_error = got_repo_close(repo);
2010 ad242220 2018-09-08 stsp if (error == NULL)
2011 ad242220 2018-09-08 stsp error = repo_error;
2012 ad242220 2018-09-08 stsp }
2013 404c43c4 2018-06-21 stsp return error;
2014 5de5890b 2018-10-18 stsp }
2015 5de5890b 2018-10-18 stsp
2016 5de5890b 2018-10-18 stsp __dead static void
2017 5de5890b 2018-10-18 stsp usage_tree(void)
2018 5de5890b 2018-10-18 stsp {
2019 c1669e2e 2019-01-09 stsp fprintf(stderr,
2020 c1669e2e 2019-01-09 stsp "usage: %s tree [-c commit] [-r repository-path] [-iR] path\n",
2021 5de5890b 2018-10-18 stsp getprogname());
2022 5de5890b 2018-10-18 stsp exit(1);
2023 5de5890b 2018-10-18 stsp }
2024 5de5890b 2018-10-18 stsp
2025 c1669e2e 2019-01-09 stsp static void
2026 c1669e2e 2019-01-09 stsp print_entry(struct got_tree_entry *te, const char *id, const char *path,
2027 c1669e2e 2019-01-09 stsp const char *root_path)
2028 c1669e2e 2019-01-09 stsp {
2029 c1669e2e 2019-01-09 stsp int is_root_path = (strcmp(path, root_path) == 0);
2030 5de5890b 2018-10-18 stsp
2031 c1669e2e 2019-01-09 stsp path += strlen(root_path);
2032 c1669e2e 2019-01-09 stsp while (path[0] == '/')
2033 c1669e2e 2019-01-09 stsp path++;
2034 c1669e2e 2019-01-09 stsp
2035 c1669e2e 2019-01-09 stsp printf("%s%s%s%s%s\n", id ? id : "", path,
2036 d6e648b4 2019-02-10 stsp is_root_path ? "" : "/", te->name,
2037 d6e648b4 2019-02-10 stsp S_ISDIR(te->mode) ? "/" : ((te->mode & S_IXUSR) ? "*" : ""));
2038 c1669e2e 2019-01-09 stsp }
2039 c1669e2e 2019-01-09 stsp
2040 5de5890b 2018-10-18 stsp static const struct got_error *
2041 5de5890b 2018-10-18 stsp print_tree(const char *path, struct got_object_id *commit_id,
2042 c1669e2e 2019-01-09 stsp int show_ids, int recurse, const char *root_path,
2043 c1669e2e 2019-01-09 stsp struct got_repository *repo)
2044 5de5890b 2018-10-18 stsp {
2045 5de5890b 2018-10-18 stsp const struct got_error *err = NULL;
2046 5de5890b 2018-10-18 stsp struct got_object_id *tree_id = NULL;
2047 5de5890b 2018-10-18 stsp struct got_tree_object *tree = NULL;
2048 5de5890b 2018-10-18 stsp const struct got_tree_entries *entries;
2049 5de5890b 2018-10-18 stsp struct got_tree_entry *te;
2050 5de5890b 2018-10-18 stsp
2051 5de5890b 2018-10-18 stsp err = got_object_id_by_path(&tree_id, repo, commit_id, path);
2052 5de5890b 2018-10-18 stsp if (err)
2053 5de5890b 2018-10-18 stsp goto done;
2054 5de5890b 2018-10-18 stsp
2055 5de5890b 2018-10-18 stsp err = got_object_open_as_tree(&tree, repo, tree_id);
2056 5de5890b 2018-10-18 stsp if (err)
2057 5de5890b 2018-10-18 stsp goto done;
2058 5de5890b 2018-10-18 stsp entries = got_object_tree_get_entries(tree);
2059 5de5890b 2018-10-18 stsp te = SIMPLEQ_FIRST(&entries->head);
2060 5de5890b 2018-10-18 stsp while (te) {
2061 5de5890b 2018-10-18 stsp char *id = NULL;
2062 84453469 2018-11-11 stsp
2063 84453469 2018-11-11 stsp if (sigint_received || sigpipe_received)
2064 84453469 2018-11-11 stsp break;
2065 84453469 2018-11-11 stsp
2066 5de5890b 2018-10-18 stsp if (show_ids) {
2067 5de5890b 2018-10-18 stsp char *id_str;
2068 5de5890b 2018-10-18 stsp err = got_object_id_str(&id_str, te->id);
2069 5de5890b 2018-10-18 stsp if (err)
2070 5de5890b 2018-10-18 stsp goto done;
2071 5de5890b 2018-10-18 stsp if (asprintf(&id, "%s ", id_str) == -1) {
2072 638f9024 2019-05-13 stsp err = got_error_from_errno("asprintf");
2073 5de5890b 2018-10-18 stsp free(id_str);
2074 5de5890b 2018-10-18 stsp goto done;
2075 5de5890b 2018-10-18 stsp }
2076 5de5890b 2018-10-18 stsp free(id_str);
2077 5de5890b 2018-10-18 stsp }
2078 c1669e2e 2019-01-09 stsp print_entry(te, id, path, root_path);
2079 5de5890b 2018-10-18 stsp free(id);
2080 c1669e2e 2019-01-09 stsp
2081 c1669e2e 2019-01-09 stsp if (recurse && S_ISDIR(te->mode)) {
2082 c1669e2e 2019-01-09 stsp char *child_path;
2083 c1669e2e 2019-01-09 stsp if (asprintf(&child_path, "%s%s%s", path,
2084 c1669e2e 2019-01-09 stsp path[0] == '/' && path[1] == '\0' ? "" : "/",
2085 c1669e2e 2019-01-09 stsp te->name) == -1) {
2086 638f9024 2019-05-13 stsp err = got_error_from_errno("asprintf");
2087 c1669e2e 2019-01-09 stsp goto done;
2088 c1669e2e 2019-01-09 stsp }
2089 c1669e2e 2019-01-09 stsp err = print_tree(child_path, commit_id, show_ids, 1,
2090 c1669e2e 2019-01-09 stsp root_path, repo);
2091 c1669e2e 2019-01-09 stsp free(child_path);
2092 c1669e2e 2019-01-09 stsp if (err)
2093 c1669e2e 2019-01-09 stsp goto done;
2094 c1669e2e 2019-01-09 stsp }
2095 c1669e2e 2019-01-09 stsp
2096 c1669e2e 2019-01-09 stsp te = SIMPLEQ_NEXT(te, entry);
2097 5de5890b 2018-10-18 stsp }
2098 5de5890b 2018-10-18 stsp done:
2099 5de5890b 2018-10-18 stsp if (tree)
2100 5de5890b 2018-10-18 stsp got_object_tree_close(tree);
2101 5de5890b 2018-10-18 stsp free(tree_id);
2102 5de5890b 2018-10-18 stsp return err;
2103 404c43c4 2018-06-21 stsp }
2104 404c43c4 2018-06-21 stsp
2105 5de5890b 2018-10-18 stsp static const struct got_error *
2106 5de5890b 2018-10-18 stsp cmd_tree(int argc, char *argv[])
2107 5de5890b 2018-10-18 stsp {
2108 5de5890b 2018-10-18 stsp const struct got_error *error;
2109 5de5890b 2018-10-18 stsp struct got_repository *repo = NULL;
2110 7a2c19d6 2019-02-05 stsp struct got_worktree *worktree = NULL;
2111 7a2c19d6 2019-02-05 stsp const char *path;
2112 7a2c19d6 2019-02-05 stsp char *cwd = NULL, *repo_path = NULL, *in_repo_path = NULL;
2113 5de5890b 2018-10-18 stsp struct got_object_id *commit_id = NULL;
2114 5de5890b 2018-10-18 stsp char *commit_id_str = NULL;
2115 c1669e2e 2019-01-09 stsp int show_ids = 0, recurse = 0;
2116 5de5890b 2018-10-18 stsp int ch;
2117 5de5890b 2018-10-18 stsp
2118 5de5890b 2018-10-18 stsp #ifndef PROFILE
2119 0f8d269b 2019-01-04 stsp if (pledge("stdio rpath wpath cpath flock proc exec sendfd unveil",
2120 0f8d269b 2019-01-04 stsp NULL) == -1)
2121 5de5890b 2018-10-18 stsp err(1, "pledge");
2122 5de5890b 2018-10-18 stsp #endif
2123 5de5890b 2018-10-18 stsp
2124 c1669e2e 2019-01-09 stsp while ((ch = getopt(argc, argv, "c:r:iR")) != -1) {
2125 5de5890b 2018-10-18 stsp switch (ch) {
2126 5de5890b 2018-10-18 stsp case 'c':
2127 5de5890b 2018-10-18 stsp commit_id_str = optarg;
2128 5de5890b 2018-10-18 stsp break;
2129 5de5890b 2018-10-18 stsp case 'r':
2130 5de5890b 2018-10-18 stsp repo_path = realpath(optarg, NULL);
2131 5de5890b 2018-10-18 stsp if (repo_path == NULL)
2132 5de5890b 2018-10-18 stsp err(1, "-r option");
2133 7fbaa4f3 2019-05-11 stsp got_path_strip_trailing_slashes(repo_path);
2134 5de5890b 2018-10-18 stsp break;
2135 5de5890b 2018-10-18 stsp case 'i':
2136 5de5890b 2018-10-18 stsp show_ids = 1;
2137 5de5890b 2018-10-18 stsp break;
2138 c1669e2e 2019-01-09 stsp case 'R':
2139 c1669e2e 2019-01-09 stsp recurse = 1;
2140 c1669e2e 2019-01-09 stsp break;
2141 5de5890b 2018-10-18 stsp default:
2142 2deda0b9 2019-03-07 stsp usage_tree();
2143 5de5890b 2018-10-18 stsp /* NOTREACHED */
2144 5de5890b 2018-10-18 stsp }
2145 5de5890b 2018-10-18 stsp }
2146 5de5890b 2018-10-18 stsp
2147 5de5890b 2018-10-18 stsp argc -= optind;
2148 5de5890b 2018-10-18 stsp argv += optind;
2149 5de5890b 2018-10-18 stsp
2150 5de5890b 2018-10-18 stsp if (argc == 1)
2151 5de5890b 2018-10-18 stsp path = argv[0];
2152 5de5890b 2018-10-18 stsp else if (argc > 1)
2153 5de5890b 2018-10-18 stsp usage_tree();
2154 5de5890b 2018-10-18 stsp else
2155 7a2c19d6 2019-02-05 stsp path = NULL;
2156 7a2c19d6 2019-02-05 stsp
2157 9bf7a39b 2019-02-05 stsp cwd = getcwd(NULL, 0);
2158 9bf7a39b 2019-02-05 stsp if (cwd == NULL) {
2159 638f9024 2019-05-13 stsp error = got_error_from_errno("getcwd");
2160 9bf7a39b 2019-02-05 stsp goto done;
2161 9bf7a39b 2019-02-05 stsp }
2162 5de5890b 2018-10-18 stsp if (repo_path == NULL) {
2163 7a2c19d6 2019-02-05 stsp error = got_worktree_open(&worktree, cwd);
2164 8994de28 2019-02-05 stsp if (error && error->code != GOT_ERR_NOT_WORKTREE)
2165 7a2c19d6 2019-02-05 stsp goto done;
2166 7a2c19d6 2019-02-05 stsp else
2167 7a2c19d6 2019-02-05 stsp error = NULL;
2168 7a2c19d6 2019-02-05 stsp if (worktree) {
2169 7a2c19d6 2019-02-05 stsp repo_path =
2170 7a2c19d6 2019-02-05 stsp strdup(got_worktree_get_repo_path(worktree));
2171 7a2c19d6 2019-02-05 stsp if (repo_path == NULL)
2172 638f9024 2019-05-13 stsp error = got_error_from_errno("strdup");
2173 7a2c19d6 2019-02-05 stsp if (error)
2174 7a2c19d6 2019-02-05 stsp goto done;
2175 7a2c19d6 2019-02-05 stsp } else {
2176 7a2c19d6 2019-02-05 stsp repo_path = strdup(cwd);
2177 7a2c19d6 2019-02-05 stsp if (repo_path == NULL) {
2178 638f9024 2019-05-13 stsp error = got_error_from_errno("strdup");
2179 7a2c19d6 2019-02-05 stsp goto done;
2180 7a2c19d6 2019-02-05 stsp }
2181 5de5890b 2018-10-18 stsp }
2182 5de5890b 2018-10-18 stsp }
2183 5de5890b 2018-10-18 stsp
2184 5de5890b 2018-10-18 stsp error = got_repo_open(&repo, repo_path);
2185 5de5890b 2018-10-18 stsp if (error != NULL)
2186 c02c541e 2019-03-29 stsp goto done;
2187 c02c541e 2019-03-29 stsp
2188 314a6357 2019-05-13 stsp error = apply_unveil(got_repo_get_path(repo), 1, NULL, 0);
2189 c02c541e 2019-03-29 stsp if (error)
2190 5de5890b 2018-10-18 stsp goto done;
2191 5de5890b 2018-10-18 stsp
2192 9bf7a39b 2019-02-05 stsp if (path == NULL) {
2193 9bf7a39b 2019-02-05 stsp if (worktree) {
2194 9bf7a39b 2019-02-05 stsp char *p, *worktree_subdir = cwd +
2195 9bf7a39b 2019-02-05 stsp strlen(got_worktree_get_root_path(worktree));
2196 9bf7a39b 2019-02-05 stsp if (asprintf(&p, "%s/%s",
2197 9bf7a39b 2019-02-05 stsp got_worktree_get_path_prefix(worktree),
2198 9bf7a39b 2019-02-05 stsp worktree_subdir) == -1) {
2199 638f9024 2019-05-13 stsp error = got_error_from_errno("asprintf");
2200 9bf7a39b 2019-02-05 stsp goto done;
2201 9bf7a39b 2019-02-05 stsp }
2202 9bf7a39b 2019-02-05 stsp error = got_repo_map_path(&in_repo_path, repo, p, 1);
2203 9bf7a39b 2019-02-05 stsp free(p);
2204 9bf7a39b 2019-02-05 stsp if (error)
2205 9bf7a39b 2019-02-05 stsp goto done;
2206 9bf7a39b 2019-02-05 stsp } else
2207 9bf7a39b 2019-02-05 stsp path = "/";
2208 9bf7a39b 2019-02-05 stsp }
2209 9bf7a39b 2019-02-05 stsp if (in_repo_path == NULL) {
2210 9bf7a39b 2019-02-05 stsp error = got_repo_map_path(&in_repo_path, repo, path, 1);
2211 9bf7a39b 2019-02-05 stsp if (error != NULL)
2212 9bf7a39b 2019-02-05 stsp goto done;
2213 9bf7a39b 2019-02-05 stsp }
2214 5de5890b 2018-10-18 stsp
2215 5de5890b 2018-10-18 stsp if (commit_id_str == NULL) {
2216 5de5890b 2018-10-18 stsp struct got_reference *head_ref;
2217 2f17228e 2019-05-12 stsp error = got_ref_open(&head_ref, repo, GOT_REF_HEAD, 0);
2218 5de5890b 2018-10-18 stsp if (error != NULL)
2219 5de5890b 2018-10-18 stsp goto done;
2220 5de5890b 2018-10-18 stsp error = got_ref_resolve(&commit_id, repo, head_ref);
2221 5de5890b 2018-10-18 stsp got_ref_close(head_ref);
2222 5de5890b 2018-10-18 stsp if (error != NULL)
2223 5de5890b 2018-10-18 stsp goto done;
2224 5de5890b 2018-10-18 stsp } else {
2225 e09a504c 2019-06-28 stsp error = got_repo_match_object_id_prefix(&commit_id,
2226 dd88155e 2019-06-29 stsp commit_id_str, GOT_OBJ_TYPE_COMMIT, repo);
2227 5de5890b 2018-10-18 stsp if (error != NULL)
2228 5de5890b 2018-10-18 stsp goto done;
2229 5de5890b 2018-10-18 stsp }
2230 5de5890b 2018-10-18 stsp
2231 c1669e2e 2019-01-09 stsp error = print_tree(in_repo_path, commit_id, show_ids, recurse,
2232 c1669e2e 2019-01-09 stsp in_repo_path, repo);
2233 5de5890b 2018-10-18 stsp done:
2234 5de5890b 2018-10-18 stsp free(in_repo_path);
2235 5de5890b 2018-10-18 stsp free(repo_path);
2236 5de5890b 2018-10-18 stsp free(cwd);
2237 5de5890b 2018-10-18 stsp free(commit_id);
2238 7a2c19d6 2019-02-05 stsp if (worktree)
2239 7a2c19d6 2019-02-05 stsp got_worktree_close(worktree);
2240 5de5890b 2018-10-18 stsp if (repo) {
2241 5de5890b 2018-10-18 stsp const struct got_error *repo_error;
2242 5de5890b 2018-10-18 stsp repo_error = got_repo_close(repo);
2243 5de5890b 2018-10-18 stsp if (error == NULL)
2244 5de5890b 2018-10-18 stsp error = repo_error;
2245 5de5890b 2018-10-18 stsp }
2246 5de5890b 2018-10-18 stsp return error;
2247 5de5890b 2018-10-18 stsp }
2248 5de5890b 2018-10-18 stsp
2249 6bad629b 2019-02-04 stsp __dead static void
2250 6bad629b 2019-02-04 stsp usage_status(void)
2251 6bad629b 2019-02-04 stsp {
2252 927df6b7 2019-02-10 stsp fprintf(stderr, "usage: %s status [path]\n", getprogname());
2253 6bad629b 2019-02-04 stsp exit(1);
2254 6bad629b 2019-02-04 stsp }
2255 5c860e29 2018-03-12 stsp
2256 b72f483a 2019-02-05 stsp static const struct got_error *
2257 b72f483a 2019-02-05 stsp print_status(void *arg, unsigned char status, const char *path,
2258 016a88dd 2019-05-13 stsp struct got_object_id *blob_id, struct got_object_id *commit_id)
2259 6bad629b 2019-02-04 stsp {
2260 6bad629b 2019-02-04 stsp printf("%c %s\n", status, path);
2261 b72f483a 2019-02-05 stsp return NULL;
2262 6bad629b 2019-02-04 stsp }
2263 5c860e29 2018-03-12 stsp
2264 6bad629b 2019-02-04 stsp static const struct got_error *
2265 6bad629b 2019-02-04 stsp cmd_status(int argc, char *argv[])
2266 6bad629b 2019-02-04 stsp {
2267 6bad629b 2019-02-04 stsp const struct got_error *error = NULL;
2268 6bad629b 2019-02-04 stsp struct got_repository *repo = NULL;
2269 6bad629b 2019-02-04 stsp struct got_worktree *worktree = NULL;
2270 927df6b7 2019-02-10 stsp char *cwd = NULL, *path = NULL;
2271 6bad629b 2019-02-04 stsp int ch;
2272 5c860e29 2018-03-12 stsp
2273 6bad629b 2019-02-04 stsp while ((ch = getopt(argc, argv, "")) != -1) {
2274 6bad629b 2019-02-04 stsp switch (ch) {
2275 5c860e29 2018-03-12 stsp default:
2276 2deda0b9 2019-03-07 stsp usage_status();
2277 6bad629b 2019-02-04 stsp /* NOTREACHED */
2278 5c860e29 2018-03-12 stsp }
2279 5c860e29 2018-03-12 stsp }
2280 5c860e29 2018-03-12 stsp
2281 6bad629b 2019-02-04 stsp argc -= optind;
2282 6bad629b 2019-02-04 stsp argv += optind;
2283 5c860e29 2018-03-12 stsp
2284 6bad629b 2019-02-04 stsp #ifndef PROFILE
2285 6bad629b 2019-02-04 stsp if (pledge("stdio rpath wpath cpath flock proc exec sendfd unveil",
2286 6bad629b 2019-02-04 stsp NULL) == -1)
2287 6bad629b 2019-02-04 stsp err(1, "pledge");
2288 f42b1b34 2018-03-12 stsp #endif
2289 927df6b7 2019-02-10 stsp cwd = getcwd(NULL, 0);
2290 927df6b7 2019-02-10 stsp if (cwd == NULL) {
2291 638f9024 2019-05-13 stsp error = got_error_from_errno("getcwd");
2292 927df6b7 2019-02-10 stsp goto done;
2293 927df6b7 2019-02-10 stsp }
2294 927df6b7 2019-02-10 stsp
2295 927df6b7 2019-02-10 stsp error = got_worktree_open(&worktree, cwd);
2296 927df6b7 2019-02-10 stsp if (error != NULL)
2297 927df6b7 2019-02-10 stsp goto done;
2298 927df6b7 2019-02-10 stsp
2299 6bad629b 2019-02-04 stsp if (argc == 0) {
2300 927df6b7 2019-02-10 stsp path = strdup("");
2301 927df6b7 2019-02-10 stsp if (path == NULL) {
2302 638f9024 2019-05-13 stsp error = got_error_from_errno("strdup");
2303 6bad629b 2019-02-04 stsp goto done;
2304 6bad629b 2019-02-04 stsp }
2305 6bad629b 2019-02-04 stsp } else if (argc == 1) {
2306 6c7ab921 2019-03-18 stsp error = got_worktree_resolve_path(&path, worktree, argv[0]);
2307 927df6b7 2019-02-10 stsp if (error)
2308 6bad629b 2019-02-04 stsp goto done;
2309 6bad629b 2019-02-04 stsp } else
2310 6bad629b 2019-02-04 stsp usage_status();
2311 6bad629b 2019-02-04 stsp
2312 6bad629b 2019-02-04 stsp error = got_repo_open(&repo, got_worktree_get_repo_path(worktree));
2313 6bad629b 2019-02-04 stsp if (error != NULL)
2314 6bad629b 2019-02-04 stsp goto done;
2315 6bad629b 2019-02-04 stsp
2316 d0eebce4 2019-03-11 stsp error = apply_unveil(got_repo_get_path(repo), 1,
2317 314a6357 2019-05-13 stsp got_worktree_get_root_path(worktree), 0);
2318 6bad629b 2019-02-04 stsp if (error)
2319 6bad629b 2019-02-04 stsp goto done;
2320 6bad629b 2019-02-04 stsp
2321 927df6b7 2019-02-10 stsp error = got_worktree_status(worktree, path, repo, print_status, NULL,
2322 6bad629b 2019-02-04 stsp check_cancelled, NULL);
2323 6bad629b 2019-02-04 stsp done:
2324 927df6b7 2019-02-10 stsp free(cwd);
2325 927df6b7 2019-02-10 stsp free(path);
2326 d0eebce4 2019-03-11 stsp return error;
2327 d0eebce4 2019-03-11 stsp }
2328 d0eebce4 2019-03-11 stsp
2329 d0eebce4 2019-03-11 stsp __dead static void
2330 d0eebce4 2019-03-11 stsp usage_ref(void)
2331 d0eebce4 2019-03-11 stsp {
2332 d0eebce4 2019-03-11 stsp fprintf(stderr,
2333 d83d9d5c 2019-05-13 stsp "usage: %s ref [-r repository] -l | -d name | name target\n",
2334 d0eebce4 2019-03-11 stsp getprogname());
2335 d0eebce4 2019-03-11 stsp exit(1);
2336 d0eebce4 2019-03-11 stsp }
2337 d0eebce4 2019-03-11 stsp
2338 d0eebce4 2019-03-11 stsp static const struct got_error *
2339 d0eebce4 2019-03-11 stsp list_refs(struct got_repository *repo)
2340 d0eebce4 2019-03-11 stsp {
2341 d0eebce4 2019-03-11 stsp static const struct got_error *err = NULL;
2342 d0eebce4 2019-03-11 stsp struct got_reflist_head refs;
2343 d0eebce4 2019-03-11 stsp struct got_reflist_entry *re;
2344 d0eebce4 2019-03-11 stsp
2345 d0eebce4 2019-03-11 stsp SIMPLEQ_INIT(&refs);
2346 d0eebce4 2019-03-11 stsp err = got_ref_list(&refs, repo);
2347 d0eebce4 2019-03-11 stsp if (err)
2348 d0eebce4 2019-03-11 stsp return err;
2349 d0eebce4 2019-03-11 stsp
2350 d0eebce4 2019-03-11 stsp SIMPLEQ_FOREACH(re, &refs, entry) {
2351 d0eebce4 2019-03-11 stsp char *refstr;
2352 d0eebce4 2019-03-11 stsp refstr = got_ref_to_str(re->ref);
2353 d0eebce4 2019-03-11 stsp if (refstr == NULL)
2354 638f9024 2019-05-13 stsp return got_error_from_errno("got_ref_to_str");
2355 d0eebce4 2019-03-11 stsp printf("%s: %s\n", got_ref_get_name(re->ref), refstr);
2356 d0eebce4 2019-03-11 stsp free(refstr);
2357 d0eebce4 2019-03-11 stsp }
2358 d0eebce4 2019-03-11 stsp
2359 e2e879a0 2019-03-11 stsp got_ref_list_free(&refs);
2360 d0eebce4 2019-03-11 stsp return NULL;
2361 d0eebce4 2019-03-11 stsp }
2362 d0eebce4 2019-03-11 stsp
2363 d0eebce4 2019-03-11 stsp static const struct got_error *
2364 d0eebce4 2019-03-11 stsp delete_ref(struct got_repository *repo, const char *refname)
2365 d0eebce4 2019-03-11 stsp {
2366 d0eebce4 2019-03-11 stsp const struct got_error *err = NULL;
2367 d0eebce4 2019-03-11 stsp struct got_reference *ref;
2368 d0eebce4 2019-03-11 stsp
2369 2f17228e 2019-05-12 stsp err = got_ref_open(&ref, repo, refname, 0);
2370 d0eebce4 2019-03-11 stsp if (err)
2371 d0eebce4 2019-03-11 stsp return err;
2372 d0eebce4 2019-03-11 stsp
2373 d0eebce4 2019-03-11 stsp err = got_ref_delete(ref, repo);
2374 d0eebce4 2019-03-11 stsp got_ref_close(ref);
2375 d0eebce4 2019-03-11 stsp return err;
2376 d0eebce4 2019-03-11 stsp }
2377 d0eebce4 2019-03-11 stsp
2378 d0eebce4 2019-03-11 stsp static const struct got_error *
2379 d83d9d5c 2019-05-13 stsp add_ref(struct got_repository *repo, const char *refname, const char *target)
2380 d0eebce4 2019-03-11 stsp {
2381 d0eebce4 2019-03-11 stsp const struct got_error *err = NULL;
2382 d0eebce4 2019-03-11 stsp struct got_object_id *id;
2383 d0eebce4 2019-03-11 stsp struct got_reference *ref = NULL;
2384 d1644381 2019-07-14 stsp
2385 d1644381 2019-07-14 stsp /*
2386 d1644381 2019-07-14 stsp * Don't let the user create a reference named '-'.
2387 d1644381 2019-07-14 stsp * While technically a valid reference name, this case is usually
2388 d1644381 2019-07-14 stsp * an unintended typo.
2389 d1644381 2019-07-14 stsp */
2390 d1644381 2019-07-14 stsp if (refname[0] == '-' && refname[1] == '\0')
2391 d1644381 2019-07-14 stsp return got_error(GOT_ERR_BAD_REF_NAME);
2392 d0eebce4 2019-03-11 stsp
2393 dd88155e 2019-06-29 stsp err = got_repo_match_object_id_prefix(&id, target, GOT_OBJ_TYPE_ANY,
2394 dd88155e 2019-06-29 stsp repo);
2395 d83d9d5c 2019-05-13 stsp if (err) {
2396 d83d9d5c 2019-05-13 stsp struct got_reference *target_ref;
2397 d0eebce4 2019-03-11 stsp
2398 d83d9d5c 2019-05-13 stsp if (err->code != GOT_ERR_BAD_OBJ_ID_STR)
2399 d83d9d5c 2019-05-13 stsp return err;
2400 d83d9d5c 2019-05-13 stsp err = got_ref_open(&target_ref, repo, target, 0);
2401 d83d9d5c 2019-05-13 stsp if (err)
2402 d83d9d5c 2019-05-13 stsp return err;
2403 d83d9d5c 2019-05-13 stsp err = got_ref_resolve(&id, repo, target_ref);
2404 d83d9d5c 2019-05-13 stsp got_ref_close(target_ref);
2405 d83d9d5c 2019-05-13 stsp if (err)
2406 d83d9d5c 2019-05-13 stsp return err;
2407 d83d9d5c 2019-05-13 stsp }
2408 d83d9d5c 2019-05-13 stsp
2409 d0eebce4 2019-03-11 stsp err = got_ref_alloc(&ref, refname, id);
2410 d0eebce4 2019-03-11 stsp if (err)
2411 d0eebce4 2019-03-11 stsp goto done;
2412 d0eebce4 2019-03-11 stsp
2413 d0eebce4 2019-03-11 stsp err = got_ref_write(ref, repo);
2414 d0eebce4 2019-03-11 stsp done:
2415 d0eebce4 2019-03-11 stsp if (ref)
2416 d0eebce4 2019-03-11 stsp got_ref_close(ref);
2417 d0eebce4 2019-03-11 stsp free(id);
2418 d0eebce4 2019-03-11 stsp return err;
2419 d0eebce4 2019-03-11 stsp }
2420 d0eebce4 2019-03-11 stsp
2421 d0eebce4 2019-03-11 stsp static const struct got_error *
2422 d0eebce4 2019-03-11 stsp cmd_ref(int argc, char *argv[])
2423 d0eebce4 2019-03-11 stsp {
2424 d0eebce4 2019-03-11 stsp const struct got_error *error = NULL;
2425 d0eebce4 2019-03-11 stsp struct got_repository *repo = NULL;
2426 d0eebce4 2019-03-11 stsp struct got_worktree *worktree = NULL;
2427 d0eebce4 2019-03-11 stsp char *cwd = NULL, *repo_path = NULL;
2428 d0eebce4 2019-03-11 stsp int ch, do_list = 0;
2429 d0eebce4 2019-03-11 stsp const char *delref = NULL;
2430 d0eebce4 2019-03-11 stsp
2431 d0eebce4 2019-03-11 stsp /* TODO: Add -s option for adding symbolic references. */
2432 d0eebce4 2019-03-11 stsp while ((ch = getopt(argc, argv, "d:r:l")) != -1) {
2433 d0eebce4 2019-03-11 stsp switch (ch) {
2434 d0eebce4 2019-03-11 stsp case 'd':
2435 d0eebce4 2019-03-11 stsp delref = optarg;
2436 d0eebce4 2019-03-11 stsp break;
2437 d0eebce4 2019-03-11 stsp case 'r':
2438 d0eebce4 2019-03-11 stsp repo_path = realpath(optarg, NULL);
2439 d0eebce4 2019-03-11 stsp if (repo_path == NULL)
2440 d0eebce4 2019-03-11 stsp err(1, "-r option");
2441 7fbaa4f3 2019-05-11 stsp got_path_strip_trailing_slashes(repo_path);
2442 d0eebce4 2019-03-11 stsp break;
2443 d0eebce4 2019-03-11 stsp case 'l':
2444 d0eebce4 2019-03-11 stsp do_list = 1;
2445 d0eebce4 2019-03-11 stsp break;
2446 d0eebce4 2019-03-11 stsp default:
2447 d0eebce4 2019-03-11 stsp usage_ref();
2448 d0eebce4 2019-03-11 stsp /* NOTREACHED */
2449 d0eebce4 2019-03-11 stsp }
2450 d0eebce4 2019-03-11 stsp }
2451 d0eebce4 2019-03-11 stsp
2452 d0eebce4 2019-03-11 stsp if (do_list && delref)
2453 d0eebce4 2019-03-11 stsp errx(1, "-l and -d options are mutually exclusive\n");
2454 d0eebce4 2019-03-11 stsp
2455 d0eebce4 2019-03-11 stsp argc -= optind;
2456 d0eebce4 2019-03-11 stsp argv += optind;
2457 d0eebce4 2019-03-11 stsp
2458 d0eebce4 2019-03-11 stsp if (do_list || delref) {
2459 d0eebce4 2019-03-11 stsp if (argc > 0)
2460 d0eebce4 2019-03-11 stsp usage_ref();
2461 d0eebce4 2019-03-11 stsp } else if (argc != 2)
2462 d0eebce4 2019-03-11 stsp usage_ref();
2463 d0eebce4 2019-03-11 stsp
2464 d0eebce4 2019-03-11 stsp #ifndef PROFILE
2465 e0b57350 2019-03-12 stsp if (do_list) {
2466 e0b57350 2019-03-12 stsp if (pledge("stdio rpath wpath flock proc exec sendfd unveil",
2467 e0b57350 2019-03-12 stsp NULL) == -1)
2468 e0b57350 2019-03-12 stsp err(1, "pledge");
2469 e0b57350 2019-03-12 stsp } else {
2470 e0b57350 2019-03-12 stsp if (pledge("stdio rpath wpath cpath fattr flock proc exec "
2471 e0b57350 2019-03-12 stsp "sendfd unveil", NULL) == -1)
2472 e0b57350 2019-03-12 stsp err(1, "pledge");
2473 e0b57350 2019-03-12 stsp }
2474 d0eebce4 2019-03-11 stsp #endif
2475 d0eebce4 2019-03-11 stsp cwd = getcwd(NULL, 0);
2476 d0eebce4 2019-03-11 stsp if (cwd == NULL) {
2477 638f9024 2019-05-13 stsp error = got_error_from_errno("getcwd");
2478 d0eebce4 2019-03-11 stsp goto done;
2479 d0eebce4 2019-03-11 stsp }
2480 d0eebce4 2019-03-11 stsp
2481 d0eebce4 2019-03-11 stsp if (repo_path == NULL) {
2482 d0eebce4 2019-03-11 stsp error = got_worktree_open(&worktree, cwd);
2483 d0eebce4 2019-03-11 stsp if (error && error->code != GOT_ERR_NOT_WORKTREE)
2484 d0eebce4 2019-03-11 stsp goto done;
2485 d0eebce4 2019-03-11 stsp else
2486 d0eebce4 2019-03-11 stsp error = NULL;
2487 d0eebce4 2019-03-11 stsp if (worktree) {
2488 d0eebce4 2019-03-11 stsp repo_path =
2489 d0eebce4 2019-03-11 stsp strdup(got_worktree_get_repo_path(worktree));
2490 d0eebce4 2019-03-11 stsp if (repo_path == NULL)
2491 638f9024 2019-05-13 stsp error = got_error_from_errno("strdup");
2492 d0eebce4 2019-03-11 stsp if (error)
2493 d0eebce4 2019-03-11 stsp goto done;
2494 d0eebce4 2019-03-11 stsp } else {
2495 d0eebce4 2019-03-11 stsp repo_path = strdup(cwd);
2496 d0eebce4 2019-03-11 stsp if (repo_path == NULL) {
2497 638f9024 2019-05-13 stsp error = got_error_from_errno("strdup");
2498 d0eebce4 2019-03-11 stsp goto done;
2499 d0eebce4 2019-03-11 stsp }
2500 d0eebce4 2019-03-11 stsp }
2501 d0eebce4 2019-03-11 stsp }
2502 d0eebce4 2019-03-11 stsp
2503 d0eebce4 2019-03-11 stsp error = got_repo_open(&repo, repo_path);
2504 d0eebce4 2019-03-11 stsp if (error != NULL)
2505 d0eebce4 2019-03-11 stsp goto done;
2506 d0eebce4 2019-03-11 stsp
2507 c02c541e 2019-03-29 stsp error = apply_unveil(got_repo_get_path(repo), do_list,
2508 314a6357 2019-05-13 stsp worktree ? got_worktree_get_root_path(worktree) : NULL, 0);
2509 c02c541e 2019-03-29 stsp if (error)
2510 c02c541e 2019-03-29 stsp goto done;
2511 c02c541e 2019-03-29 stsp
2512 d0eebce4 2019-03-11 stsp if (do_list)
2513 d0eebce4 2019-03-11 stsp error = list_refs(repo);
2514 d0eebce4 2019-03-11 stsp else if (delref)
2515 d0eebce4 2019-03-11 stsp error = delete_ref(repo, delref);
2516 d0eebce4 2019-03-11 stsp else
2517 d0eebce4 2019-03-11 stsp error = add_ref(repo, argv[0], argv[1]);
2518 4e759de4 2019-06-26 stsp done:
2519 4e759de4 2019-06-26 stsp if (repo)
2520 4e759de4 2019-06-26 stsp got_repo_close(repo);
2521 4e759de4 2019-06-26 stsp if (worktree)
2522 4e759de4 2019-06-26 stsp got_worktree_close(worktree);
2523 4e759de4 2019-06-26 stsp free(cwd);
2524 4e759de4 2019-06-26 stsp free(repo_path);
2525 4e759de4 2019-06-26 stsp return error;
2526 4e759de4 2019-06-26 stsp }
2527 4e759de4 2019-06-26 stsp
2528 4e759de4 2019-06-26 stsp __dead static void
2529 4e759de4 2019-06-26 stsp usage_branch(void)
2530 4e759de4 2019-06-26 stsp {
2531 4e759de4 2019-06-26 stsp fprintf(stderr,
2532 4e759de4 2019-06-26 stsp "usage: %s branch [-r repository] -l | -d name | "
2533 4e759de4 2019-06-26 stsp "name [base-branch]\n", getprogname());
2534 4e759de4 2019-06-26 stsp exit(1);
2535 4e759de4 2019-06-26 stsp }
2536 4e759de4 2019-06-26 stsp
2537 4e759de4 2019-06-26 stsp static const struct got_error *
2538 ba882ee3 2019-07-11 stsp list_branches(struct got_repository *repo, struct got_worktree *worktree)
2539 4e759de4 2019-06-26 stsp {
2540 4e759de4 2019-06-26 stsp static const struct got_error *err = NULL;
2541 4e759de4 2019-06-26 stsp struct got_reflist_head refs;
2542 4e759de4 2019-06-26 stsp struct got_reflist_entry *re;
2543 4e759de4 2019-06-26 stsp
2544 4e759de4 2019-06-26 stsp SIMPLEQ_INIT(&refs);
2545 ba882ee3 2019-07-11 stsp
2546 4e759de4 2019-06-26 stsp err = got_ref_list(&refs, repo);
2547 4e759de4 2019-06-26 stsp if (err)
2548 4e759de4 2019-06-26 stsp return err;
2549 4e759de4 2019-06-26 stsp
2550 4e759de4 2019-06-26 stsp SIMPLEQ_FOREACH(re, &refs, entry) {
2551 ba882ee3 2019-07-11 stsp const char *refname, *marker = " ";
2552 4e759de4 2019-06-26 stsp char *refstr;
2553 4e759de4 2019-06-26 stsp refname = got_ref_get_name(re->ref);
2554 4e759de4 2019-06-26 stsp if (strncmp(refname, "refs/heads/", 11) != 0)
2555 4e759de4 2019-06-26 stsp continue;
2556 ba882ee3 2019-07-11 stsp if (worktree && strcmp(refname,
2557 ba882ee3 2019-07-11 stsp got_worktree_get_head_ref_name(worktree)) == 0) {
2558 ba882ee3 2019-07-11 stsp struct got_object_id *id = NULL;
2559 ba882ee3 2019-07-11 stsp err = got_ref_resolve(&id, repo, re->ref);
2560 ba882ee3 2019-07-11 stsp if (err)
2561 ba882ee3 2019-07-11 stsp return err;
2562 ba882ee3 2019-07-11 stsp if (got_object_id_cmp(id,
2563 ba882ee3 2019-07-11 stsp got_worktree_get_base_commit_id(worktree)) == 0)
2564 ba882ee3 2019-07-11 stsp marker = "* ";
2565 ba882ee3 2019-07-11 stsp else
2566 ba882ee3 2019-07-11 stsp marker = "~ ";
2567 ba882ee3 2019-07-11 stsp free(id);
2568 ba882ee3 2019-07-11 stsp }
2569 4e759de4 2019-06-26 stsp refname += 11;
2570 4e759de4 2019-06-26 stsp refstr = got_ref_to_str(re->ref);
2571 4e759de4 2019-06-26 stsp if (refstr == NULL)
2572 4e759de4 2019-06-26 stsp return got_error_from_errno("got_ref_to_str");
2573 ba882ee3 2019-07-11 stsp printf("%s%s: %s\n", marker, refname, refstr);
2574 4e759de4 2019-06-26 stsp free(refstr);
2575 4e759de4 2019-06-26 stsp }
2576 4e759de4 2019-06-26 stsp
2577 4e759de4 2019-06-26 stsp got_ref_list_free(&refs);
2578 4e759de4 2019-06-26 stsp return NULL;
2579 4e759de4 2019-06-26 stsp }
2580 4e759de4 2019-06-26 stsp
2581 4e759de4 2019-06-26 stsp static const struct got_error *
2582 4e759de4 2019-06-26 stsp delete_branch(struct got_repository *repo, const char *branch_name)
2583 4e759de4 2019-06-26 stsp {
2584 4e759de4 2019-06-26 stsp const struct got_error *err = NULL;
2585 4e759de4 2019-06-26 stsp struct got_reference *ref;
2586 4e759de4 2019-06-26 stsp char *refname;
2587 4e759de4 2019-06-26 stsp
2588 4e759de4 2019-06-26 stsp if (asprintf(&refname, "refs/heads/%s", branch_name) == -1)
2589 4e759de4 2019-06-26 stsp return got_error_from_errno("asprintf");
2590 4e759de4 2019-06-26 stsp
2591 4e759de4 2019-06-26 stsp err = got_ref_open(&ref, repo, refname, 0);
2592 4e759de4 2019-06-26 stsp if (err)
2593 4e759de4 2019-06-26 stsp goto done;
2594 4e759de4 2019-06-26 stsp
2595 4e759de4 2019-06-26 stsp err = got_ref_delete(ref, repo);
2596 4e759de4 2019-06-26 stsp got_ref_close(ref);
2597 4e759de4 2019-06-26 stsp done:
2598 4e759de4 2019-06-26 stsp free(refname);
2599 4e759de4 2019-06-26 stsp return err;
2600 4e759de4 2019-06-26 stsp }
2601 4e759de4 2019-06-26 stsp
2602 4e759de4 2019-06-26 stsp static const struct got_error *
2603 4e759de4 2019-06-26 stsp add_branch(struct got_repository *repo, const char *branch_name,
2604 4e759de4 2019-06-26 stsp const char *base_branch)
2605 4e759de4 2019-06-26 stsp {
2606 4e759de4 2019-06-26 stsp const struct got_error *err = NULL;
2607 4e759de4 2019-06-26 stsp struct got_object_id *id = NULL;
2608 4e759de4 2019-06-26 stsp struct got_reference *ref = NULL;
2609 4e759de4 2019-06-26 stsp char *base_refname = NULL, *refname = NULL;
2610 4e759de4 2019-06-26 stsp struct got_reference *base_ref;
2611 d3f84d51 2019-07-11 stsp
2612 d3f84d51 2019-07-11 stsp /*
2613 d3f84d51 2019-07-11 stsp * Don't let the user create a branch named '-'.
2614 d3f84d51 2019-07-11 stsp * While technically a valid reference name, this case is usually
2615 d3f84d51 2019-07-11 stsp * an unintended typo.
2616 d3f84d51 2019-07-11 stsp */
2617 d3f84d51 2019-07-11 stsp if (branch_name[0] == '-' && branch_name[1] == '\0')
2618 d3f84d51 2019-07-11 stsp return got_error(GOT_ERR_BAD_REF_NAME);
2619 4e759de4 2019-06-26 stsp
2620 4e759de4 2019-06-26 stsp if (strcmp(GOT_REF_HEAD, base_branch) == 0) {
2621 4e759de4 2019-06-26 stsp base_refname = strdup(GOT_REF_HEAD);
2622 4e759de4 2019-06-26 stsp if (base_refname == NULL)
2623 4e759de4 2019-06-26 stsp return got_error_from_errno("strdup");
2624 4e759de4 2019-06-26 stsp } else if (asprintf(&base_refname, "refs/heads/%s", base_branch) == -1)
2625 4e759de4 2019-06-26 stsp return got_error_from_errno("asprintf");
2626 4e759de4 2019-06-26 stsp
2627 4e759de4 2019-06-26 stsp err = got_ref_open(&base_ref, repo, base_refname, 0);
2628 4e759de4 2019-06-26 stsp if (err)
2629 4e759de4 2019-06-26 stsp goto done;
2630 4e759de4 2019-06-26 stsp err = got_ref_resolve(&id, repo, base_ref);
2631 4e759de4 2019-06-26 stsp got_ref_close(base_ref);
2632 4e759de4 2019-06-26 stsp if (err)
2633 4e759de4 2019-06-26 stsp goto done;
2634 4e759de4 2019-06-26 stsp
2635 4e759de4 2019-06-26 stsp if (asprintf(&refname, "refs/heads/%s", branch_name) == -1) {
2636 4e759de4 2019-06-26 stsp err = got_error_from_errno("asprintf");
2637 4e759de4 2019-06-26 stsp goto done;
2638 4e759de4 2019-06-26 stsp }
2639 4e759de4 2019-06-26 stsp
2640 4e759de4 2019-06-26 stsp err = got_ref_open(&ref, repo, refname, 0);
2641 4e759de4 2019-06-26 stsp if (err == NULL) {
2642 4e759de4 2019-06-26 stsp err = got_error(GOT_ERR_BRANCH_EXISTS);
2643 4e759de4 2019-06-26 stsp goto done;
2644 4e759de4 2019-06-26 stsp } else if (err->code != GOT_ERR_NOT_REF)
2645 4e759de4 2019-06-26 stsp goto done;
2646 4e759de4 2019-06-26 stsp
2647 4e759de4 2019-06-26 stsp err = got_ref_alloc(&ref, refname, id);
2648 4e759de4 2019-06-26 stsp if (err)
2649 4e759de4 2019-06-26 stsp goto done;
2650 4e759de4 2019-06-26 stsp
2651 4e759de4 2019-06-26 stsp err = got_ref_write(ref, repo);
2652 d0eebce4 2019-03-11 stsp done:
2653 4e759de4 2019-06-26 stsp if (ref)
2654 4e759de4 2019-06-26 stsp got_ref_close(ref);
2655 4e759de4 2019-06-26 stsp free(id);
2656 4e759de4 2019-06-26 stsp free(base_refname);
2657 4e759de4 2019-06-26 stsp free(refname);
2658 4e759de4 2019-06-26 stsp return err;
2659 4e759de4 2019-06-26 stsp }
2660 4e759de4 2019-06-26 stsp
2661 4e759de4 2019-06-26 stsp static const struct got_error *
2662 4e759de4 2019-06-26 stsp cmd_branch(int argc, char *argv[])
2663 4e759de4 2019-06-26 stsp {
2664 4e759de4 2019-06-26 stsp const struct got_error *error = NULL;
2665 4e759de4 2019-06-26 stsp struct got_repository *repo = NULL;
2666 4e759de4 2019-06-26 stsp struct got_worktree *worktree = NULL;
2667 4e759de4 2019-06-26 stsp char *cwd = NULL, *repo_path = NULL;
2668 4e759de4 2019-06-26 stsp int ch, do_list = 0;
2669 4e759de4 2019-06-26 stsp const char *delref = NULL;
2670 4e759de4 2019-06-26 stsp
2671 4e759de4 2019-06-26 stsp while ((ch = getopt(argc, argv, "d:r:l")) != -1) {
2672 4e759de4 2019-06-26 stsp switch (ch) {
2673 4e759de4 2019-06-26 stsp case 'd':
2674 4e759de4 2019-06-26 stsp delref = optarg;
2675 4e759de4 2019-06-26 stsp break;
2676 4e759de4 2019-06-26 stsp case 'r':
2677 4e759de4 2019-06-26 stsp repo_path = realpath(optarg, NULL);
2678 4e759de4 2019-06-26 stsp if (repo_path == NULL)
2679 4e759de4 2019-06-26 stsp err(1, "-r option");
2680 4e759de4 2019-06-26 stsp got_path_strip_trailing_slashes(repo_path);
2681 4e759de4 2019-06-26 stsp break;
2682 4e759de4 2019-06-26 stsp case 'l':
2683 4e759de4 2019-06-26 stsp do_list = 1;
2684 4e759de4 2019-06-26 stsp break;
2685 4e759de4 2019-06-26 stsp default:
2686 4e759de4 2019-06-26 stsp usage_branch();
2687 4e759de4 2019-06-26 stsp /* NOTREACHED */
2688 4e759de4 2019-06-26 stsp }
2689 4e759de4 2019-06-26 stsp }
2690 4e759de4 2019-06-26 stsp
2691 4e759de4 2019-06-26 stsp if (do_list && delref)
2692 4e759de4 2019-06-26 stsp errx(1, "-l and -d options are mutually exclusive\n");
2693 4e759de4 2019-06-26 stsp
2694 4e759de4 2019-06-26 stsp argc -= optind;
2695 4e759de4 2019-06-26 stsp argv += optind;
2696 4e759de4 2019-06-26 stsp
2697 4e759de4 2019-06-26 stsp if (do_list || delref) {
2698 4e759de4 2019-06-26 stsp if (argc > 0)
2699 4e759de4 2019-06-26 stsp usage_branch();
2700 4e759de4 2019-06-26 stsp } else if (argc < 1 || argc > 2)
2701 4e759de4 2019-06-26 stsp usage_branch();
2702 4e759de4 2019-06-26 stsp
2703 4e759de4 2019-06-26 stsp #ifndef PROFILE
2704 4e759de4 2019-06-26 stsp if (do_list) {
2705 4e759de4 2019-06-26 stsp if (pledge("stdio rpath wpath flock proc exec sendfd unveil",
2706 4e759de4 2019-06-26 stsp NULL) == -1)
2707 4e759de4 2019-06-26 stsp err(1, "pledge");
2708 4e759de4 2019-06-26 stsp } else {
2709 4e759de4 2019-06-26 stsp if (pledge("stdio rpath wpath cpath fattr flock proc exec "
2710 4e759de4 2019-06-26 stsp "sendfd unveil", NULL) == -1)
2711 4e759de4 2019-06-26 stsp err(1, "pledge");
2712 4e759de4 2019-06-26 stsp }
2713 4e759de4 2019-06-26 stsp #endif
2714 4e759de4 2019-06-26 stsp cwd = getcwd(NULL, 0);
2715 4e759de4 2019-06-26 stsp if (cwd == NULL) {
2716 4e759de4 2019-06-26 stsp error = got_error_from_errno("getcwd");
2717 4e759de4 2019-06-26 stsp goto done;
2718 4e759de4 2019-06-26 stsp }
2719 4e759de4 2019-06-26 stsp
2720 4e759de4 2019-06-26 stsp if (repo_path == NULL) {
2721 4e759de4 2019-06-26 stsp error = got_worktree_open(&worktree, cwd);
2722 4e759de4 2019-06-26 stsp if (error && error->code != GOT_ERR_NOT_WORKTREE)
2723 4e759de4 2019-06-26 stsp goto done;
2724 4e759de4 2019-06-26 stsp else
2725 4e759de4 2019-06-26 stsp error = NULL;
2726 4e759de4 2019-06-26 stsp if (worktree) {
2727 4e759de4 2019-06-26 stsp repo_path =
2728 4e759de4 2019-06-26 stsp strdup(got_worktree_get_repo_path(worktree));
2729 4e759de4 2019-06-26 stsp if (repo_path == NULL)
2730 4e759de4 2019-06-26 stsp error = got_error_from_errno("strdup");
2731 4e759de4 2019-06-26 stsp if (error)
2732 4e759de4 2019-06-26 stsp goto done;
2733 4e759de4 2019-06-26 stsp } else {
2734 4e759de4 2019-06-26 stsp repo_path = strdup(cwd);
2735 4e759de4 2019-06-26 stsp if (repo_path == NULL) {
2736 4e759de4 2019-06-26 stsp error = got_error_from_errno("strdup");
2737 4e759de4 2019-06-26 stsp goto done;
2738 4e759de4 2019-06-26 stsp }
2739 4e759de4 2019-06-26 stsp }
2740 4e759de4 2019-06-26 stsp }
2741 4e759de4 2019-06-26 stsp
2742 4e759de4 2019-06-26 stsp error = got_repo_open(&repo, repo_path);
2743 4e759de4 2019-06-26 stsp if (error != NULL)
2744 4e759de4 2019-06-26 stsp goto done;
2745 4e759de4 2019-06-26 stsp
2746 4e759de4 2019-06-26 stsp error = apply_unveil(got_repo_get_path(repo), do_list,
2747 4e759de4 2019-06-26 stsp worktree ? got_worktree_get_root_path(worktree) : NULL, 0);
2748 4e759de4 2019-06-26 stsp if (error)
2749 4e759de4 2019-06-26 stsp goto done;
2750 4e759de4 2019-06-26 stsp
2751 4e759de4 2019-06-26 stsp if (do_list)
2752 ba882ee3 2019-07-11 stsp error = list_branches(repo, worktree);
2753 4e759de4 2019-06-26 stsp else if (delref)
2754 4e759de4 2019-06-26 stsp error = delete_branch(repo, delref);
2755 4e759de4 2019-06-26 stsp else {
2756 4e759de4 2019-06-26 stsp const char *base_branch;
2757 4e759de4 2019-06-26 stsp if (argc == 1) {
2758 4e759de4 2019-06-26 stsp base_branch = worktree ?
2759 4e759de4 2019-06-26 stsp got_worktree_get_head_ref_name(worktree) :
2760 4e759de4 2019-06-26 stsp GOT_REF_HEAD;
2761 4e759de4 2019-06-26 stsp } else
2762 4e759de4 2019-06-26 stsp base_branch = argv[1];
2763 4e759de4 2019-06-26 stsp error = add_branch(repo, argv[0], base_branch);
2764 4e759de4 2019-06-26 stsp }
2765 4e759de4 2019-06-26 stsp done:
2766 d0eebce4 2019-03-11 stsp if (repo)
2767 d0eebce4 2019-03-11 stsp got_repo_close(repo);
2768 d0eebce4 2019-03-11 stsp if (worktree)
2769 d0eebce4 2019-03-11 stsp got_worktree_close(worktree);
2770 d0eebce4 2019-03-11 stsp free(cwd);
2771 d0eebce4 2019-03-11 stsp free(repo_path);
2772 d00136be 2019-03-26 stsp return error;
2773 d00136be 2019-03-26 stsp }
2774 d00136be 2019-03-26 stsp
2775 d00136be 2019-03-26 stsp __dead static void
2776 d00136be 2019-03-26 stsp usage_add(void)
2777 d00136be 2019-03-26 stsp {
2778 fbb7e5c7 2019-05-11 stsp fprintf(stderr, "usage: %s add file-path ...\n", getprogname());
2779 d00136be 2019-03-26 stsp exit(1);
2780 d00136be 2019-03-26 stsp }
2781 d00136be 2019-03-26 stsp
2782 d00136be 2019-03-26 stsp static const struct got_error *
2783 d00136be 2019-03-26 stsp cmd_add(int argc, char *argv[])
2784 d00136be 2019-03-26 stsp {
2785 d00136be 2019-03-26 stsp const struct got_error *error = NULL;
2786 031a5338 2019-03-26 stsp struct got_repository *repo = NULL;
2787 d00136be 2019-03-26 stsp struct got_worktree *worktree = NULL;
2788 1dd54920 2019-05-11 stsp char *cwd = NULL;
2789 1dd54920 2019-05-11 stsp struct got_pathlist_head paths;
2790 1dd54920 2019-05-11 stsp struct got_pathlist_entry *pe;
2791 723c305c 2019-05-11 jcs int ch, x;
2792 1dd54920 2019-05-11 stsp
2793 1dd54920 2019-05-11 stsp TAILQ_INIT(&paths);
2794 d00136be 2019-03-26 stsp
2795 d00136be 2019-03-26 stsp while ((ch = getopt(argc, argv, "")) != -1) {
2796 d00136be 2019-03-26 stsp switch (ch) {
2797 d00136be 2019-03-26 stsp default:
2798 d00136be 2019-03-26 stsp usage_add();
2799 d00136be 2019-03-26 stsp /* NOTREACHED */
2800 d00136be 2019-03-26 stsp }
2801 d00136be 2019-03-26 stsp }
2802 d00136be 2019-03-26 stsp
2803 d00136be 2019-03-26 stsp argc -= optind;
2804 d00136be 2019-03-26 stsp argv += optind;
2805 d00136be 2019-03-26 stsp
2806 43012d58 2019-07-14 stsp #ifndef PROFILE
2807 43012d58 2019-07-14 stsp if (pledge("stdio rpath wpath cpath flock proc exec sendfd unveil",
2808 43012d58 2019-07-14 stsp NULL) == -1)
2809 43012d58 2019-07-14 stsp err(1, "pledge");
2810 43012d58 2019-07-14 stsp #endif
2811 723c305c 2019-05-11 jcs if (argc < 1)
2812 d00136be 2019-03-26 stsp usage_add();
2813 d00136be 2019-03-26 stsp
2814 723c305c 2019-05-11 jcs /* make sure each file exists before doing anything halfway */
2815 723c305c 2019-05-11 jcs for (x = 0; x < argc; x++) {
2816 1dd54920 2019-05-11 stsp char *path = realpath(argv[x], NULL);
2817 723c305c 2019-05-11 jcs if (path == NULL) {
2818 638f9024 2019-05-13 stsp error = got_error_from_errno2("realpath", argv[x]);
2819 723c305c 2019-05-11 jcs goto done;
2820 723c305c 2019-05-11 jcs }
2821 1dd54920 2019-05-11 stsp free(path);
2822 d00136be 2019-03-26 stsp }
2823 d00136be 2019-03-26 stsp
2824 d00136be 2019-03-26 stsp cwd = getcwd(NULL, 0);
2825 d00136be 2019-03-26 stsp if (cwd == NULL) {
2826 638f9024 2019-05-13 stsp error = got_error_from_errno("getcwd");
2827 d00136be 2019-03-26 stsp goto done;
2828 d00136be 2019-03-26 stsp }
2829 723c305c 2019-05-11 jcs
2830 d00136be 2019-03-26 stsp error = got_worktree_open(&worktree, cwd);
2831 d00136be 2019-03-26 stsp if (error)
2832 d00136be 2019-03-26 stsp goto done;
2833 d00136be 2019-03-26 stsp
2834 031a5338 2019-03-26 stsp error = got_repo_open(&repo, got_worktree_get_repo_path(worktree));
2835 031a5338 2019-03-26 stsp if (error != NULL)
2836 031a5338 2019-03-26 stsp goto done;
2837 031a5338 2019-03-26 stsp
2838 031a5338 2019-03-26 stsp error = apply_unveil(got_repo_get_path(repo), 1,
2839 314a6357 2019-05-13 stsp got_worktree_get_root_path(worktree), 0);
2840 d00136be 2019-03-26 stsp if (error)
2841 d00136be 2019-03-26 stsp goto done;
2842 d00136be 2019-03-26 stsp
2843 723c305c 2019-05-11 jcs for (x = 0; x < argc; x++) {
2844 1dd54920 2019-05-11 stsp char *path = realpath(argv[x], NULL);
2845 723c305c 2019-05-11 jcs if (path == NULL) {
2846 638f9024 2019-05-13 stsp error = got_error_from_errno2("realpath", argv[x]);
2847 723c305c 2019-05-11 jcs goto done;
2848 723c305c 2019-05-11 jcs }
2849 723c305c 2019-05-11 jcs
2850 723c305c 2019-05-11 jcs got_path_strip_trailing_slashes(path);
2851 1dd54920 2019-05-11 stsp error = got_pathlist_insert(&pe, &paths, path, NULL);
2852 1dd54920 2019-05-11 stsp if (error) {
2853 1dd54920 2019-05-11 stsp free(path);
2854 723c305c 2019-05-11 jcs goto done;
2855 1dd54920 2019-05-11 stsp }
2856 723c305c 2019-05-11 jcs }
2857 1dd54920 2019-05-11 stsp error = got_worktree_schedule_add(worktree, &paths, print_status,
2858 1dd54920 2019-05-11 stsp NULL, repo);
2859 d00136be 2019-03-26 stsp done:
2860 031a5338 2019-03-26 stsp if (repo)
2861 031a5338 2019-03-26 stsp got_repo_close(repo);
2862 d00136be 2019-03-26 stsp if (worktree)
2863 d00136be 2019-03-26 stsp got_worktree_close(worktree);
2864 1dd54920 2019-05-11 stsp TAILQ_FOREACH(pe, &paths, entry)
2865 1dd54920 2019-05-11 stsp free((char *)pe->path);
2866 1dd54920 2019-05-11 stsp got_pathlist_free(&paths);
2867 2ec1f75b 2019-03-26 stsp free(cwd);
2868 2ec1f75b 2019-03-26 stsp return error;
2869 2ec1f75b 2019-03-26 stsp }
2870 2ec1f75b 2019-03-26 stsp
2871 2ec1f75b 2019-03-26 stsp __dead static void
2872 648e4ef7 2019-07-09 stsp usage_remove(void)
2873 2ec1f75b 2019-03-26 stsp {
2874 648e4ef7 2019-07-09 stsp fprintf(stderr, "usage: %s remove [-f] file-path ...\n", getprogname());
2875 2ec1f75b 2019-03-26 stsp exit(1);
2876 2ec1f75b 2019-03-26 stsp }
2877 2ec1f75b 2019-03-26 stsp
2878 2ec1f75b 2019-03-26 stsp static const struct got_error *
2879 648e4ef7 2019-07-09 stsp cmd_remove(int argc, char *argv[])
2880 2ec1f75b 2019-03-26 stsp {
2881 2ec1f75b 2019-03-26 stsp const struct got_error *error = NULL;
2882 2ec1f75b 2019-03-26 stsp struct got_worktree *worktree = NULL;
2883 2ec1f75b 2019-03-26 stsp struct got_repository *repo = NULL;
2884 17ed4618 2019-06-02 stsp char *cwd = NULL;
2885 17ed4618 2019-06-02 stsp struct got_pathlist_head paths;
2886 17ed4618 2019-06-02 stsp struct got_pathlist_entry *pe;
2887 17ed4618 2019-06-02 stsp int ch, i, delete_local_mods = 0;
2888 2ec1f75b 2019-03-26 stsp
2889 17ed4618 2019-06-02 stsp TAILQ_INIT(&paths);
2890 17ed4618 2019-06-02 stsp
2891 2ec1f75b 2019-03-26 stsp while ((ch = getopt(argc, argv, "f")) != -1) {
2892 2ec1f75b 2019-03-26 stsp switch (ch) {
2893 2ec1f75b 2019-03-26 stsp case 'f':
2894 2ec1f75b 2019-03-26 stsp delete_local_mods = 1;
2895 2ec1f75b 2019-03-26 stsp break;
2896 2ec1f75b 2019-03-26 stsp default:
2897 2ec1f75b 2019-03-26 stsp usage_add();
2898 2ec1f75b 2019-03-26 stsp /* NOTREACHED */
2899 2ec1f75b 2019-03-26 stsp }
2900 2ec1f75b 2019-03-26 stsp }
2901 2ec1f75b 2019-03-26 stsp
2902 2ec1f75b 2019-03-26 stsp argc -= optind;
2903 2ec1f75b 2019-03-26 stsp argv += optind;
2904 2ec1f75b 2019-03-26 stsp
2905 43012d58 2019-07-14 stsp #ifndef PROFILE
2906 43012d58 2019-07-14 stsp if (pledge("stdio rpath wpath cpath flock proc exec sendfd unveil",
2907 43012d58 2019-07-14 stsp NULL) == -1)
2908 43012d58 2019-07-14 stsp err(1, "pledge");
2909 43012d58 2019-07-14 stsp #endif
2910 17ed4618 2019-06-02 stsp if (argc < 1)
2911 648e4ef7 2019-07-09 stsp usage_remove();
2912 2ec1f75b 2019-03-26 stsp
2913 17ed4618 2019-06-02 stsp /* make sure each file exists before doing anything halfway */
2914 17ed4618 2019-06-02 stsp for (i = 0; i < argc; i++) {
2915 17ed4618 2019-06-02 stsp char *path = realpath(argv[i], NULL);
2916 17ed4618 2019-06-02 stsp if (path == NULL) {
2917 17ed4618 2019-06-02 stsp error = got_error_from_errno2("realpath", argv[i]);
2918 17ed4618 2019-06-02 stsp goto done;
2919 17ed4618 2019-06-02 stsp }
2920 17ed4618 2019-06-02 stsp free(path);
2921 2ec1f75b 2019-03-26 stsp }
2922 2ec1f75b 2019-03-26 stsp
2923 2ec1f75b 2019-03-26 stsp cwd = getcwd(NULL, 0);
2924 2ec1f75b 2019-03-26 stsp if (cwd == NULL) {
2925 638f9024 2019-05-13 stsp error = got_error_from_errno("getcwd");
2926 2ec1f75b 2019-03-26 stsp goto done;
2927 2ec1f75b 2019-03-26 stsp }
2928 2ec1f75b 2019-03-26 stsp error = got_worktree_open(&worktree, cwd);
2929 2ec1f75b 2019-03-26 stsp if (error)
2930 2ec1f75b 2019-03-26 stsp goto done;
2931 2ec1f75b 2019-03-26 stsp
2932 2ec1f75b 2019-03-26 stsp error = got_repo_open(&repo, got_worktree_get_repo_path(worktree));
2933 2af4a041 2019-05-11 jcs if (error)
2934 2ec1f75b 2019-03-26 stsp goto done;
2935 2ec1f75b 2019-03-26 stsp
2936 c2253644 2019-03-26 stsp error = apply_unveil(got_repo_get_path(repo), 1,
2937 314a6357 2019-05-13 stsp got_worktree_get_root_path(worktree), 0);
2938 2ec1f75b 2019-03-26 stsp if (error)
2939 2ec1f75b 2019-03-26 stsp goto done;
2940 2ec1f75b 2019-03-26 stsp
2941 17ed4618 2019-06-02 stsp for (i = 0; i < argc; i++) {
2942 17ed4618 2019-06-02 stsp char *path = realpath(argv[i], NULL);
2943 17ed4618 2019-06-02 stsp if (path == NULL) {
2944 17ed4618 2019-06-02 stsp error = got_error_from_errno2("realpath", argv[i]);
2945 17ed4618 2019-06-02 stsp goto done;
2946 17ed4618 2019-06-02 stsp }
2947 17ed4618 2019-06-02 stsp
2948 17ed4618 2019-06-02 stsp got_path_strip_trailing_slashes(path);
2949 17ed4618 2019-06-02 stsp error = got_pathlist_insert(&pe, &paths, path, NULL);
2950 17ed4618 2019-06-02 stsp if (error) {
2951 17ed4618 2019-06-02 stsp free(path);
2952 17ed4618 2019-06-02 stsp goto done;
2953 17ed4618 2019-06-02 stsp }
2954 17ed4618 2019-06-02 stsp }
2955 17ed4618 2019-06-02 stsp error = got_worktree_schedule_delete(worktree, &paths,
2956 17ed4618 2019-06-02 stsp delete_local_mods, print_status, NULL, repo);
2957 a129376b 2019-03-28 stsp if (error)
2958 a129376b 2019-03-28 stsp goto done;
2959 a129376b 2019-03-28 stsp done:
2960 a129376b 2019-03-28 stsp if (repo)
2961 a129376b 2019-03-28 stsp got_repo_close(repo);
2962 a129376b 2019-03-28 stsp if (worktree)
2963 a129376b 2019-03-28 stsp got_worktree_close(worktree);
2964 17ed4618 2019-06-02 stsp TAILQ_FOREACH(pe, &paths, entry)
2965 17ed4618 2019-06-02 stsp free((char *)pe->path);
2966 17ed4618 2019-06-02 stsp got_pathlist_free(&paths);
2967 a129376b 2019-03-28 stsp free(cwd);
2968 a129376b 2019-03-28 stsp return error;
2969 a129376b 2019-03-28 stsp }
2970 a129376b 2019-03-28 stsp
2971 a129376b 2019-03-28 stsp __dead static void
2972 a129376b 2019-03-28 stsp usage_revert(void)
2973 a129376b 2019-03-28 stsp {
2974 e20a8b6f 2019-06-04 stsp fprintf(stderr, "usage: %s revert file-path ...\n", getprogname());
2975 a129376b 2019-03-28 stsp exit(1);
2976 a129376b 2019-03-28 stsp }
2977 a129376b 2019-03-28 stsp
2978 1ee397ad 2019-07-12 stsp static const struct got_error *
2979 a129376b 2019-03-28 stsp revert_progress(void *arg, unsigned char status, const char *path)
2980 a129376b 2019-03-28 stsp {
2981 a129376b 2019-03-28 stsp while (path[0] == '/')
2982 a129376b 2019-03-28 stsp path++;
2983 a129376b 2019-03-28 stsp printf("%c %s\n", status, path);
2984 1ee397ad 2019-07-12 stsp return NULL;
2985 a129376b 2019-03-28 stsp }
2986 a129376b 2019-03-28 stsp
2987 a129376b 2019-03-28 stsp static const struct got_error *
2988 a129376b 2019-03-28 stsp cmd_revert(int argc, char *argv[])
2989 a129376b 2019-03-28 stsp {
2990 a129376b 2019-03-28 stsp const struct got_error *error = NULL;
2991 a129376b 2019-03-28 stsp struct got_worktree *worktree = NULL;
2992 a129376b 2019-03-28 stsp struct got_repository *repo = NULL;
2993 a129376b 2019-03-28 stsp char *cwd = NULL, *path = NULL;
2994 e20a8b6f 2019-06-04 stsp struct got_pathlist_head paths;
2995 e20a8b6f 2019-06-04 stsp struct got_pathlist_entry *pe;
2996 e20a8b6f 2019-06-04 stsp int ch, i;
2997 a129376b 2019-03-28 stsp
2998 e20a8b6f 2019-06-04 stsp TAILQ_INIT(&paths);
2999 e20a8b6f 2019-06-04 stsp
3000 a129376b 2019-03-28 stsp while ((ch = getopt(argc, argv, "")) != -1) {
3001 a129376b 2019-03-28 stsp switch (ch) {
3002 a129376b 2019-03-28 stsp default:
3003 a129376b 2019-03-28 stsp usage_revert();
3004 a129376b 2019-03-28 stsp /* NOTREACHED */
3005 a129376b 2019-03-28 stsp }
3006 a129376b 2019-03-28 stsp }
3007 a129376b 2019-03-28 stsp
3008 a129376b 2019-03-28 stsp argc -= optind;
3009 a129376b 2019-03-28 stsp argv += optind;
3010 a129376b 2019-03-28 stsp
3011 43012d58 2019-07-14 stsp #ifndef PROFILE
3012 43012d58 2019-07-14 stsp if (pledge("stdio rpath wpath cpath fattr flock proc exec sendfd "
3013 43012d58 2019-07-14 stsp "unveil", NULL) == -1)
3014 43012d58 2019-07-14 stsp err(1, "pledge");
3015 43012d58 2019-07-14 stsp #endif
3016 e20a8b6f 2019-06-04 stsp if (argc < 1)
3017 a129376b 2019-03-28 stsp usage_revert();
3018 a129376b 2019-03-28 stsp
3019 e20a8b6f 2019-06-04 stsp for (i = 0; i < argc; i++) {
3020 e20a8b6f 2019-06-04 stsp char *path = realpath(argv[i], NULL);
3021 e20a8b6f 2019-06-04 stsp if (path == NULL) {
3022 e20a8b6f 2019-06-04 stsp error = got_error_from_errno2("realpath", argv[i]);
3023 e20a8b6f 2019-06-04 stsp goto done;
3024 e20a8b6f 2019-06-04 stsp }
3025 e20a8b6f 2019-06-04 stsp
3026 e20a8b6f 2019-06-04 stsp got_path_strip_trailing_slashes(path);
3027 e20a8b6f 2019-06-04 stsp error = got_pathlist_insert(&pe, &paths, path, NULL);
3028 e20a8b6f 2019-06-04 stsp if (error) {
3029 e20a8b6f 2019-06-04 stsp free(path);
3030 e20a8b6f 2019-06-04 stsp goto done;
3031 e20a8b6f 2019-06-04 stsp }
3032 a129376b 2019-03-28 stsp }
3033 a129376b 2019-03-28 stsp
3034 a129376b 2019-03-28 stsp cwd = getcwd(NULL, 0);
3035 a129376b 2019-03-28 stsp if (cwd == NULL) {
3036 638f9024 2019-05-13 stsp error = got_error_from_errno("getcwd");
3037 a129376b 2019-03-28 stsp goto done;
3038 a129376b 2019-03-28 stsp }
3039 a129376b 2019-03-28 stsp error = got_worktree_open(&worktree, cwd);
3040 2ec1f75b 2019-03-26 stsp if (error)
3041 2ec1f75b 2019-03-26 stsp goto done;
3042 a129376b 2019-03-28 stsp
3043 a129376b 2019-03-28 stsp error = got_repo_open(&repo, got_worktree_get_repo_path(worktree));
3044 a129376b 2019-03-28 stsp if (error != NULL)
3045 a129376b 2019-03-28 stsp goto done;
3046 a129376b 2019-03-28 stsp
3047 a129376b 2019-03-28 stsp error = apply_unveil(got_repo_get_path(repo), 1,
3048 314a6357 2019-05-13 stsp got_worktree_get_root_path(worktree), 0);
3049 a129376b 2019-03-28 stsp if (error)
3050 a129376b 2019-03-28 stsp goto done;
3051 a129376b 2019-03-28 stsp
3052 e20a8b6f 2019-06-04 stsp error = got_worktree_revert(worktree, &paths,
3053 a129376b 2019-03-28 stsp revert_progress, NULL, repo);
3054 a129376b 2019-03-28 stsp if (error)
3055 a129376b 2019-03-28 stsp goto done;
3056 2ec1f75b 2019-03-26 stsp done:
3057 2ec1f75b 2019-03-26 stsp if (repo)
3058 2ec1f75b 2019-03-26 stsp got_repo_close(repo);
3059 2ec1f75b 2019-03-26 stsp if (worktree)
3060 2ec1f75b 2019-03-26 stsp got_worktree_close(worktree);
3061 2ec1f75b 2019-03-26 stsp free(path);
3062 d00136be 2019-03-26 stsp free(cwd);
3063 6bad629b 2019-02-04 stsp return error;
3064 c4296144 2019-05-09 stsp }
3065 c4296144 2019-05-09 stsp
3066 c4296144 2019-05-09 stsp __dead static void
3067 c4296144 2019-05-09 stsp usage_commit(void)
3068 c4296144 2019-05-09 stsp {
3069 c6fc0acd 2019-05-09 stsp fprintf(stderr, "usage: %s commit [-m msg] file-path\n", getprogname());
3070 c4296144 2019-05-09 stsp exit(1);
3071 33ad4cbe 2019-05-12 jcs }
3072 33ad4cbe 2019-05-12 jcs
3073 e2ba3d07 2019-05-13 stsp struct collect_commit_logmsg_arg {
3074 e2ba3d07 2019-05-13 stsp const char *cmdline_log;
3075 e2ba3d07 2019-05-13 stsp const char *editor;
3076 e0870e44 2019-05-13 stsp const char *worktree_path;
3077 76d98825 2019-06-03 stsp const char *branch_name;
3078 314a6357 2019-05-13 stsp const char *repo_path;
3079 e0870e44 2019-05-13 stsp char *logmsg_path;
3080 e2ba3d07 2019-05-13 stsp
3081 e2ba3d07 2019-05-13 stsp };
3082 e0870e44 2019-05-13 stsp
3083 33ad4cbe 2019-05-12 jcs static const struct got_error *
3084 33ad4cbe 2019-05-12 jcs collect_commit_logmsg(struct got_pathlist_head *commitable_paths, char **logmsg,
3085 33ad4cbe 2019-05-12 jcs void *arg)
3086 33ad4cbe 2019-05-12 jcs {
3087 76d98825 2019-06-03 stsp char *initial_content = NULL;
3088 33ad4cbe 2019-05-12 jcs struct got_pathlist_entry *pe;
3089 33ad4cbe 2019-05-12 jcs const struct got_error *err = NULL;
3090 e0870e44 2019-05-13 stsp char *template = NULL;
3091 e2ba3d07 2019-05-13 stsp struct collect_commit_logmsg_arg *a = arg;
3092 3ce1b845 2019-07-15 stsp int fd;
3093 33ad4cbe 2019-05-12 jcs size_t len;
3094 33ad4cbe 2019-05-12 jcs
3095 33ad4cbe 2019-05-12 jcs /* if a message was specified on the command line, just use it */
3096 e2ba3d07 2019-05-13 stsp if (a->cmdline_log != NULL && strlen(a->cmdline_log) != 0) {
3097 e2ba3d07 2019-05-13 stsp len = strlen(a->cmdline_log) + 1;
3098 33ad4cbe 2019-05-12 jcs *logmsg = malloc(len + 1);
3099 9f42ff69 2019-05-13 stsp if (*logmsg == NULL)
3100 638f9024 2019-05-13 stsp return got_error_from_errno("malloc");
3101 e2ba3d07 2019-05-13 stsp strlcpy(*logmsg, a->cmdline_log, len);
3102 33ad4cbe 2019-05-12 jcs return NULL;
3103 33ad4cbe 2019-05-12 jcs }
3104 33ad4cbe 2019-05-12 jcs
3105 e0870e44 2019-05-13 stsp if (asprintf(&template, "%s/logmsg", a->worktree_path) == -1)
3106 76d98825 2019-06-03 stsp return got_error_from_errno("asprintf");
3107 76d98825 2019-06-03 stsp
3108 76d98825 2019-06-03 stsp if (asprintf(&initial_content,
3109 76d98825 2019-06-03 stsp "\n# changes to be committed on branch %s:\n",
3110 76d98825 2019-06-03 stsp a->branch_name) == -1)
3111 638f9024 2019-05-13 stsp return got_error_from_errno("asprintf");
3112 e0870e44 2019-05-13 stsp
3113 e0870e44 2019-05-13 stsp err = got_opentemp_named_fd(&a->logmsg_path, &fd, template);
3114 793c30b5 2019-05-13 stsp if (err)
3115 e0870e44 2019-05-13 stsp goto done;
3116 33ad4cbe 2019-05-12 jcs
3117 e0870e44 2019-05-13 stsp dprintf(fd, initial_content);
3118 33ad4cbe 2019-05-12 jcs
3119 33ad4cbe 2019-05-12 jcs TAILQ_FOREACH(pe, commitable_paths, entry) {
3120 33ad4cbe 2019-05-12 jcs struct got_commitable *ct = pe->data;
3121 8656d6c4 2019-05-20 stsp dprintf(fd, "# %c %s\n",
3122 8656d6c4 2019-05-20 stsp got_commitable_get_status(ct),
3123 8656d6c4 2019-05-20 stsp got_commitable_get_path(ct));
3124 33ad4cbe 2019-05-12 jcs }
3125 33ad4cbe 2019-05-12 jcs close(fd);
3126 33ad4cbe 2019-05-12 jcs
3127 3ce1b845 2019-07-15 stsp err = edit_logmsg(logmsg, a->editor, a->logmsg_path, initial_content);
3128 33ad4cbe 2019-05-12 jcs done:
3129 3ce1b845 2019-07-15 stsp unlink(a->logmsg_path);
3130 3ce1b845 2019-07-15 stsp free(a->logmsg_path);
3131 76d98825 2019-06-03 stsp free(initial_content);
3132 e0870e44 2019-05-13 stsp free(template);
3133 314a6357 2019-05-13 stsp
3134 314a6357 2019-05-13 stsp /* Editor is done; we can now apply unveil(2) */
3135 3ce1b845 2019-07-15 stsp if (err == NULL) {
3136 314a6357 2019-05-13 stsp err = apply_unveil(a->repo_path, 0, a->worktree_path, 0);
3137 3ce1b845 2019-07-15 stsp if (err) {
3138 3ce1b845 2019-07-15 stsp free(*logmsg);
3139 3ce1b845 2019-07-15 stsp *logmsg = NULL;
3140 3ce1b845 2019-07-15 stsp }
3141 3ce1b845 2019-07-15 stsp }
3142 33ad4cbe 2019-05-12 jcs return err;
3143 6bad629b 2019-02-04 stsp }
3144 c4296144 2019-05-09 stsp
3145 c4296144 2019-05-09 stsp static const struct got_error *
3146 c4296144 2019-05-09 stsp cmd_commit(int argc, char *argv[])
3147 c4296144 2019-05-09 stsp {
3148 c4296144 2019-05-09 stsp const struct got_error *error = NULL;
3149 c4296144 2019-05-09 stsp struct got_worktree *worktree = NULL;
3150 c4296144 2019-05-09 stsp struct got_repository *repo = NULL;
3151 c4296144 2019-05-09 stsp char *cwd = NULL, *path = NULL, *id_str = NULL;
3152 c4296144 2019-05-09 stsp struct got_object_id *id = NULL;
3153 33ad4cbe 2019-05-12 jcs const char *logmsg = NULL;
3154 35bd8fed 2019-05-09 stsp const char *got_author = getenv("GOT_AUTHOR");
3155 e2ba3d07 2019-05-13 stsp struct collect_commit_logmsg_arg cl_arg;
3156 e2ba3d07 2019-05-13 stsp char *editor = NULL;
3157 7d5807f4 2019-07-11 stsp int ch, rebase_in_progress;
3158 c4296144 2019-05-09 stsp
3159 c4296144 2019-05-09 stsp while ((ch = getopt(argc, argv, "m:")) != -1) {
3160 c4296144 2019-05-09 stsp switch (ch) {
3161 c4296144 2019-05-09 stsp case 'm':
3162 c4296144 2019-05-09 stsp logmsg = optarg;
3163 c4296144 2019-05-09 stsp break;
3164 c4296144 2019-05-09 stsp default:
3165 c4296144 2019-05-09 stsp usage_commit();
3166 c4296144 2019-05-09 stsp /* NOTREACHED */
3167 c4296144 2019-05-09 stsp }
3168 c4296144 2019-05-09 stsp }
3169 c4296144 2019-05-09 stsp
3170 c4296144 2019-05-09 stsp argc -= optind;
3171 c4296144 2019-05-09 stsp argv += optind;
3172 c4296144 2019-05-09 stsp
3173 43012d58 2019-07-14 stsp #ifndef PROFILE
3174 43012d58 2019-07-14 stsp if (pledge("stdio rpath wpath cpath fattr flock proc exec sendfd "
3175 43012d58 2019-07-14 stsp "unveil", NULL) == -1)
3176 43012d58 2019-07-14 stsp err(1, "pledge");
3177 43012d58 2019-07-14 stsp #endif
3178 c4296144 2019-05-09 stsp if (argc == 1) {
3179 c4296144 2019-05-09 stsp path = realpath(argv[0], NULL);
3180 c4296144 2019-05-09 stsp if (path == NULL) {
3181 638f9024 2019-05-13 stsp error = got_error_from_errno2("realpath", argv[0]);
3182 c4296144 2019-05-09 stsp goto done;
3183 c4296144 2019-05-09 stsp }
3184 7fbaa4f3 2019-05-11 stsp got_path_strip_trailing_slashes(path);
3185 c4296144 2019-05-09 stsp } else if (argc != 0)
3186 c4296144 2019-05-09 stsp usage_commit();
3187 c4296144 2019-05-09 stsp
3188 35bd8fed 2019-05-09 stsp if (got_author == NULL) {
3189 35bd8fed 2019-05-09 stsp /* TODO: Look current user up in password database */
3190 35bd8fed 2019-05-09 stsp error = got_error(GOT_ERR_COMMIT_NO_AUTHOR);
3191 35bd8fed 2019-05-09 stsp goto done;
3192 35bd8fed 2019-05-09 stsp }
3193 c4296144 2019-05-09 stsp
3194 c4296144 2019-05-09 stsp cwd = getcwd(NULL, 0);
3195 c4296144 2019-05-09 stsp if (cwd == NULL) {
3196 638f9024 2019-05-13 stsp error = got_error_from_errno("getcwd");
3197 c4296144 2019-05-09 stsp goto done;
3198 c4296144 2019-05-09 stsp }
3199 c4296144 2019-05-09 stsp error = got_worktree_open(&worktree, cwd);
3200 7d5807f4 2019-07-11 stsp if (error)
3201 7d5807f4 2019-07-11 stsp goto done;
3202 7d5807f4 2019-07-11 stsp
3203 7d5807f4 2019-07-11 stsp error = got_worktree_rebase_in_progress(&rebase_in_progress, worktree);
3204 c4296144 2019-05-09 stsp if (error)
3205 7d5807f4 2019-07-11 stsp goto done;
3206 7d5807f4 2019-07-11 stsp if (rebase_in_progress) {
3207 7d5807f4 2019-07-11 stsp error = got_error(GOT_ERR_REBASING);
3208 c4296144 2019-05-09 stsp goto done;
3209 7d5807f4 2019-07-11 stsp }
3210 c4296144 2019-05-09 stsp
3211 c4296144 2019-05-09 stsp error = got_repo_open(&repo, got_worktree_get_repo_path(worktree));
3212 c4296144 2019-05-09 stsp if (error != NULL)
3213 c4296144 2019-05-09 stsp goto done;
3214 c4296144 2019-05-09 stsp
3215 314a6357 2019-05-13 stsp /*
3216 314a6357 2019-05-13 stsp * unveil(2) traverses exec(2); if an editor is used we have
3217 314a6357 2019-05-13 stsp * to apply unveil after the log message has been written.
3218 314a6357 2019-05-13 stsp */
3219 314a6357 2019-05-13 stsp if (logmsg == NULL || strlen(logmsg) == 0)
3220 314a6357 2019-05-13 stsp error = get_editor(&editor);
3221 314a6357 2019-05-13 stsp else
3222 314a6357 2019-05-13 stsp error = apply_unveil(got_repo_get_path(repo), 0,
3223 314a6357 2019-05-13 stsp got_worktree_get_root_path(worktree), 0);
3224 c4296144 2019-05-09 stsp if (error)
3225 c4296144 2019-05-09 stsp goto done;
3226 c4296144 2019-05-09 stsp
3227 e2ba3d07 2019-05-13 stsp cl_arg.editor = editor;
3228 e2ba3d07 2019-05-13 stsp cl_arg.cmdline_log = logmsg;
3229 e0870e44 2019-05-13 stsp cl_arg.worktree_path = got_worktree_get_root_path(worktree);
3230 76d98825 2019-06-03 stsp cl_arg.branch_name = got_worktree_get_head_ref_name(worktree);
3231 76d98825 2019-06-03 stsp if (strncmp(cl_arg.branch_name, "refs/", 5) == 0)
3232 76d98825 2019-06-03 stsp cl_arg.branch_name += 5;
3233 76d98825 2019-06-03 stsp if (strncmp(cl_arg.branch_name, "heads/", 6) == 0)
3234 76d98825 2019-06-03 stsp cl_arg.branch_name += 6;
3235 314a6357 2019-05-13 stsp cl_arg.repo_path = got_repo_get_path(repo);
3236 e0870e44 2019-05-13 stsp cl_arg.logmsg_path = NULL;
3237 35bd8fed 2019-05-09 stsp error = got_worktree_commit(&id, worktree, path, got_author, NULL,
3238 e2ba3d07 2019-05-13 stsp collect_commit_logmsg, &cl_arg, print_status, NULL, repo);
3239 e0870e44 2019-05-13 stsp if (error) {
3240 e0870e44 2019-05-13 stsp if (cl_arg.logmsg_path)
3241 e0870e44 2019-05-13 stsp fprintf(stderr, "%s: log message preserved in %s\n",
3242 e0870e44 2019-05-13 stsp getprogname(), cl_arg.logmsg_path);
3243 c4296144 2019-05-09 stsp goto done;
3244 e0870e44 2019-05-13 stsp }
3245 c4296144 2019-05-09 stsp
3246 e0870e44 2019-05-13 stsp if (cl_arg.logmsg_path)
3247 e0870e44 2019-05-13 stsp unlink(cl_arg.logmsg_path);
3248 e0870e44 2019-05-13 stsp
3249 c4296144 2019-05-09 stsp error = got_object_id_str(&id_str, id);
3250 c4296144 2019-05-09 stsp if (error)
3251 c4296144 2019-05-09 stsp goto done;
3252 a7648d7a 2019-06-02 stsp printf("Created commit %s\n", id_str);
3253 c4296144 2019-05-09 stsp done:
3254 c4296144 2019-05-09 stsp if (repo)
3255 c4296144 2019-05-09 stsp got_repo_close(repo);
3256 c4296144 2019-05-09 stsp if (worktree)
3257 c4296144 2019-05-09 stsp got_worktree_close(worktree);
3258 c4296144 2019-05-09 stsp free(path);
3259 c4296144 2019-05-09 stsp free(cwd);
3260 c4296144 2019-05-09 stsp free(id_str);
3261 e2ba3d07 2019-05-13 stsp free(editor);
3262 234035bc 2019-06-01 stsp return error;
3263 234035bc 2019-06-01 stsp }
3264 234035bc 2019-06-01 stsp
3265 234035bc 2019-06-01 stsp __dead static void
3266 234035bc 2019-06-01 stsp usage_cherrypick(void)
3267 234035bc 2019-06-01 stsp {
3268 234035bc 2019-06-01 stsp fprintf(stderr, "usage: %s cherrypick commit-id\n", getprogname());
3269 234035bc 2019-06-01 stsp exit(1);
3270 234035bc 2019-06-01 stsp }
3271 234035bc 2019-06-01 stsp
3272 234035bc 2019-06-01 stsp static const struct got_error *
3273 234035bc 2019-06-01 stsp cmd_cherrypick(int argc, char *argv[])
3274 234035bc 2019-06-01 stsp {
3275 234035bc 2019-06-01 stsp const struct got_error *error = NULL;
3276 234035bc 2019-06-01 stsp struct got_worktree *worktree = NULL;
3277 234035bc 2019-06-01 stsp struct got_repository *repo = NULL;
3278 234035bc 2019-06-01 stsp char *cwd = NULL, *commit_id_str = NULL;
3279 234035bc 2019-06-01 stsp struct got_object_id *commit_id = NULL;
3280 234035bc 2019-06-01 stsp struct got_commit_object *commit = NULL;
3281 234035bc 2019-06-01 stsp struct got_object_qid *pid;
3282 234035bc 2019-06-01 stsp struct got_reference *head_ref = NULL;
3283 234035bc 2019-06-01 stsp int ch, did_something = 0;
3284 234035bc 2019-06-01 stsp
3285 234035bc 2019-06-01 stsp while ((ch = getopt(argc, argv, "")) != -1) {
3286 234035bc 2019-06-01 stsp switch (ch) {
3287 234035bc 2019-06-01 stsp default:
3288 234035bc 2019-06-01 stsp usage_cherrypick();
3289 234035bc 2019-06-01 stsp /* NOTREACHED */
3290 234035bc 2019-06-01 stsp }
3291 234035bc 2019-06-01 stsp }
3292 234035bc 2019-06-01 stsp
3293 234035bc 2019-06-01 stsp argc -= optind;
3294 234035bc 2019-06-01 stsp argv += optind;
3295 234035bc 2019-06-01 stsp
3296 43012d58 2019-07-14 stsp #ifndef PROFILE
3297 43012d58 2019-07-14 stsp if (pledge("stdio rpath wpath cpath fattr flock proc exec sendfd "
3298 43012d58 2019-07-14 stsp "unveil", NULL) == -1)
3299 43012d58 2019-07-14 stsp err(1, "pledge");
3300 43012d58 2019-07-14 stsp #endif
3301 234035bc 2019-06-01 stsp if (argc != 1)
3302 234035bc 2019-06-01 stsp usage_cherrypick();
3303 234035bc 2019-06-01 stsp
3304 234035bc 2019-06-01 stsp cwd = getcwd(NULL, 0);
3305 234035bc 2019-06-01 stsp if (cwd == NULL) {
3306 234035bc 2019-06-01 stsp error = got_error_from_errno("getcwd");
3307 234035bc 2019-06-01 stsp goto done;
3308 234035bc 2019-06-01 stsp }
3309 234035bc 2019-06-01 stsp error = got_worktree_open(&worktree, cwd);
3310 234035bc 2019-06-01 stsp if (error)
3311 234035bc 2019-06-01 stsp goto done;
3312 234035bc 2019-06-01 stsp
3313 234035bc 2019-06-01 stsp error = got_repo_open(&repo, got_worktree_get_repo_path(worktree));
3314 234035bc 2019-06-01 stsp if (error != NULL)
3315 234035bc 2019-06-01 stsp goto done;
3316 234035bc 2019-06-01 stsp
3317 234035bc 2019-06-01 stsp error = apply_unveil(got_repo_get_path(repo), 0,
3318 234035bc 2019-06-01 stsp got_worktree_get_root_path(worktree), 0);
3319 234035bc 2019-06-01 stsp if (error)
3320 234035bc 2019-06-01 stsp goto done;
3321 234035bc 2019-06-01 stsp
3322 dd88155e 2019-06-29 stsp error = got_repo_match_object_id_prefix(&commit_id, argv[0],
3323 dd88155e 2019-06-29 stsp GOT_OBJ_TYPE_COMMIT, repo);
3324 234035bc 2019-06-01 stsp if (error != NULL) {
3325 234035bc 2019-06-01 stsp struct got_reference *ref;
3326 234035bc 2019-06-01 stsp if (error->code != GOT_ERR_BAD_OBJ_ID_STR)
3327 234035bc 2019-06-01 stsp goto done;
3328 234035bc 2019-06-01 stsp error = got_ref_open(&ref, repo, argv[0], 0);
3329 234035bc 2019-06-01 stsp if (error != NULL)
3330 234035bc 2019-06-01 stsp goto done;
3331 234035bc 2019-06-01 stsp error = got_ref_resolve(&commit_id, repo, ref);
3332 234035bc 2019-06-01 stsp got_ref_close(ref);
3333 234035bc 2019-06-01 stsp if (error != NULL)
3334 234035bc 2019-06-01 stsp goto done;
3335 234035bc 2019-06-01 stsp }
3336 234035bc 2019-06-01 stsp error = got_object_id_str(&commit_id_str, commit_id);
3337 234035bc 2019-06-01 stsp if (error)
3338 234035bc 2019-06-01 stsp goto done;
3339 234035bc 2019-06-01 stsp
3340 234035bc 2019-06-01 stsp error = got_ref_open(&head_ref, repo,
3341 234035bc 2019-06-01 stsp got_worktree_get_head_ref_name(worktree), 0);
3342 234035bc 2019-06-01 stsp if (error != NULL)
3343 234035bc 2019-06-01 stsp goto done;
3344 234035bc 2019-06-01 stsp
3345 234035bc 2019-06-01 stsp error = check_same_branch(commit_id, head_ref, repo);
3346 234035bc 2019-06-01 stsp if (error) {
3347 234035bc 2019-06-01 stsp if (error->code != GOT_ERR_ANCESTRY)
3348 234035bc 2019-06-01 stsp goto done;
3349 234035bc 2019-06-01 stsp error = NULL;
3350 234035bc 2019-06-01 stsp } else {
3351 234035bc 2019-06-01 stsp error = got_error(GOT_ERR_SAME_BRANCH);
3352 234035bc 2019-06-01 stsp goto done;
3353 234035bc 2019-06-01 stsp }
3354 234035bc 2019-06-01 stsp
3355 234035bc 2019-06-01 stsp error = got_object_open_as_commit(&commit, repo, commit_id);
3356 234035bc 2019-06-01 stsp if (error)
3357 234035bc 2019-06-01 stsp goto done;
3358 234035bc 2019-06-01 stsp pid = SIMPLEQ_FIRST(got_object_commit_get_parent_ids(commit));
3359 03415a1a 2019-06-02 stsp error = got_worktree_merge_files(worktree, pid ? pid->id : NULL,
3360 03415a1a 2019-06-02 stsp commit_id, repo, update_progress, &did_something, check_cancelled,
3361 03415a1a 2019-06-02 stsp NULL);
3362 234035bc 2019-06-01 stsp if (error != NULL)
3363 234035bc 2019-06-01 stsp goto done;
3364 234035bc 2019-06-01 stsp
3365 234035bc 2019-06-01 stsp if (did_something)
3366 a7648d7a 2019-06-02 stsp printf("Merged commit %s\n", commit_id_str);
3367 234035bc 2019-06-01 stsp done:
3368 234035bc 2019-06-01 stsp if (commit)
3369 234035bc 2019-06-01 stsp got_object_commit_close(commit);
3370 234035bc 2019-06-01 stsp free(commit_id_str);
3371 234035bc 2019-06-01 stsp if (head_ref)
3372 234035bc 2019-06-01 stsp got_ref_close(head_ref);
3373 234035bc 2019-06-01 stsp if (worktree)
3374 234035bc 2019-06-01 stsp got_worktree_close(worktree);
3375 234035bc 2019-06-01 stsp if (repo)
3376 234035bc 2019-06-01 stsp got_repo_close(repo);
3377 c4296144 2019-05-09 stsp return error;
3378 c4296144 2019-05-09 stsp }
3379 5ef14e63 2019-06-02 stsp
3380 5ef14e63 2019-06-02 stsp __dead static void
3381 5ef14e63 2019-06-02 stsp usage_backout(void)
3382 5ef14e63 2019-06-02 stsp {
3383 5ef14e63 2019-06-02 stsp fprintf(stderr, "usage: %s backout commit-id\n", getprogname());
3384 5ef14e63 2019-06-02 stsp exit(1);
3385 5ef14e63 2019-06-02 stsp }
3386 5ef14e63 2019-06-02 stsp
3387 5ef14e63 2019-06-02 stsp static const struct got_error *
3388 5ef14e63 2019-06-02 stsp cmd_backout(int argc, char *argv[])
3389 5ef14e63 2019-06-02 stsp {
3390 5ef14e63 2019-06-02 stsp const struct got_error *error = NULL;
3391 5ef14e63 2019-06-02 stsp struct got_worktree *worktree = NULL;
3392 5ef14e63 2019-06-02 stsp struct got_repository *repo = NULL;
3393 5ef14e63 2019-06-02 stsp char *cwd = NULL, *commit_id_str = NULL;
3394 5ef14e63 2019-06-02 stsp struct got_object_id *commit_id = NULL;
3395 5ef14e63 2019-06-02 stsp struct got_commit_object *commit = NULL;
3396 5ef14e63 2019-06-02 stsp struct got_object_qid *pid;
3397 5ef14e63 2019-06-02 stsp struct got_reference *head_ref = NULL;
3398 5ef14e63 2019-06-02 stsp int ch, did_something = 0;
3399 5ef14e63 2019-06-02 stsp
3400 5ef14e63 2019-06-02 stsp while ((ch = getopt(argc, argv, "")) != -1) {
3401 5ef14e63 2019-06-02 stsp switch (ch) {
3402 5ef14e63 2019-06-02 stsp default:
3403 5ef14e63 2019-06-02 stsp usage_backout();
3404 5ef14e63 2019-06-02 stsp /* NOTREACHED */
3405 5ef14e63 2019-06-02 stsp }
3406 5ef14e63 2019-06-02 stsp }
3407 5ef14e63 2019-06-02 stsp
3408 5ef14e63 2019-06-02 stsp argc -= optind;
3409 5ef14e63 2019-06-02 stsp argv += optind;
3410 5ef14e63 2019-06-02 stsp
3411 43012d58 2019-07-14 stsp #ifndef PROFILE
3412 43012d58 2019-07-14 stsp if (pledge("stdio rpath wpath cpath fattr flock proc exec sendfd "
3413 43012d58 2019-07-14 stsp "unveil", NULL) == -1)
3414 43012d58 2019-07-14 stsp err(1, "pledge");
3415 43012d58 2019-07-14 stsp #endif
3416 5ef14e63 2019-06-02 stsp if (argc != 1)
3417 5ef14e63 2019-06-02 stsp usage_backout();
3418 5ef14e63 2019-06-02 stsp
3419 5ef14e63 2019-06-02 stsp cwd = getcwd(NULL, 0);
3420 5ef14e63 2019-06-02 stsp if (cwd == NULL) {
3421 5ef14e63 2019-06-02 stsp error = got_error_from_errno("getcwd");
3422 5ef14e63 2019-06-02 stsp goto done;
3423 5ef14e63 2019-06-02 stsp }
3424 5ef14e63 2019-06-02 stsp error = got_worktree_open(&worktree, cwd);
3425 5ef14e63 2019-06-02 stsp if (error)
3426 5ef14e63 2019-06-02 stsp goto done;
3427 5ef14e63 2019-06-02 stsp
3428 5ef14e63 2019-06-02 stsp error = got_repo_open(&repo, got_worktree_get_repo_path(worktree));
3429 5ef14e63 2019-06-02 stsp if (error != NULL)
3430 5ef14e63 2019-06-02 stsp goto done;
3431 5ef14e63 2019-06-02 stsp
3432 5ef14e63 2019-06-02 stsp error = apply_unveil(got_repo_get_path(repo), 0,
3433 5ef14e63 2019-06-02 stsp got_worktree_get_root_path(worktree), 0);
3434 5ef14e63 2019-06-02 stsp if (error)
3435 5ef14e63 2019-06-02 stsp goto done;
3436 5ef14e63 2019-06-02 stsp
3437 dd88155e 2019-06-29 stsp error = got_repo_match_object_id_prefix(&commit_id, argv[0],
3438 dd88155e 2019-06-29 stsp GOT_OBJ_TYPE_COMMIT, repo);
3439 5ef14e63 2019-06-02 stsp if (error != NULL) {
3440 5ef14e63 2019-06-02 stsp struct got_reference *ref;
3441 5ef14e63 2019-06-02 stsp if (error->code != GOT_ERR_BAD_OBJ_ID_STR)
3442 5ef14e63 2019-06-02 stsp goto done;
3443 5ef14e63 2019-06-02 stsp error = got_ref_open(&ref, repo, argv[0], 0);
3444 5ef14e63 2019-06-02 stsp if (error != NULL)
3445 5ef14e63 2019-06-02 stsp goto done;
3446 5ef14e63 2019-06-02 stsp error = got_ref_resolve(&commit_id, repo, ref);
3447 5ef14e63 2019-06-02 stsp got_ref_close(ref);
3448 5ef14e63 2019-06-02 stsp if (error != NULL)
3449 5ef14e63 2019-06-02 stsp goto done;
3450 5ef14e63 2019-06-02 stsp }
3451 5ef14e63 2019-06-02 stsp error = got_object_id_str(&commit_id_str, commit_id);
3452 5ef14e63 2019-06-02 stsp if (error)
3453 5ef14e63 2019-06-02 stsp goto done;
3454 5ef14e63 2019-06-02 stsp
3455 5ef14e63 2019-06-02 stsp error = got_ref_open(&head_ref, repo,
3456 5ef14e63 2019-06-02 stsp got_worktree_get_head_ref_name(worktree), 0);
3457 5ef14e63 2019-06-02 stsp if (error != NULL)
3458 5ef14e63 2019-06-02 stsp goto done;
3459 5ef14e63 2019-06-02 stsp
3460 5ef14e63 2019-06-02 stsp error = check_same_branch(commit_id, head_ref, repo);
3461 5ef14e63 2019-06-02 stsp if (error)
3462 5ef14e63 2019-06-02 stsp goto done;
3463 5ef14e63 2019-06-02 stsp
3464 5ef14e63 2019-06-02 stsp error = got_object_open_as_commit(&commit, repo, commit_id);
3465 5ef14e63 2019-06-02 stsp if (error)
3466 5ef14e63 2019-06-02 stsp goto done;
3467 5ef14e63 2019-06-02 stsp pid = SIMPLEQ_FIRST(got_object_commit_get_parent_ids(commit));
3468 5ef14e63 2019-06-02 stsp if (pid == NULL) {
3469 5ef14e63 2019-06-02 stsp error = got_error(GOT_ERR_ROOT_COMMIT);
3470 5ef14e63 2019-06-02 stsp goto done;
3471 5ef14e63 2019-06-02 stsp }
3472 5ef14e63 2019-06-02 stsp
3473 5ef14e63 2019-06-02 stsp error = got_worktree_merge_files(worktree, commit_id, pid->id, repo,
3474 5ef14e63 2019-06-02 stsp update_progress, &did_something, check_cancelled, NULL);
3475 5ef14e63 2019-06-02 stsp if (error != NULL)
3476 5ef14e63 2019-06-02 stsp goto done;
3477 5ef14e63 2019-06-02 stsp
3478 5ef14e63 2019-06-02 stsp if (did_something)
3479 a7648d7a 2019-06-02 stsp printf("Backed out commit %s\n", commit_id_str);
3480 5ef14e63 2019-06-02 stsp done:
3481 5ef14e63 2019-06-02 stsp if (commit)
3482 5ef14e63 2019-06-02 stsp got_object_commit_close(commit);
3483 5ef14e63 2019-06-02 stsp free(commit_id_str);
3484 5ef14e63 2019-06-02 stsp if (head_ref)
3485 5ef14e63 2019-06-02 stsp got_ref_close(head_ref);
3486 818c7501 2019-07-11 stsp if (worktree)
3487 818c7501 2019-07-11 stsp got_worktree_close(worktree);
3488 818c7501 2019-07-11 stsp if (repo)
3489 818c7501 2019-07-11 stsp got_repo_close(repo);
3490 818c7501 2019-07-11 stsp return error;
3491 818c7501 2019-07-11 stsp }
3492 818c7501 2019-07-11 stsp
3493 818c7501 2019-07-11 stsp __dead static void
3494 818c7501 2019-07-11 stsp usage_rebase(void)
3495 818c7501 2019-07-11 stsp {
3496 af54c8f8 2019-07-11 stsp fprintf(stderr, "usage: %s rebase [-a] | [-c] | branch\n",
3497 af54c8f8 2019-07-11 stsp getprogname());
3498 818c7501 2019-07-11 stsp exit(1);
3499 818c7501 2019-07-11 stsp }
3500 818c7501 2019-07-11 stsp
3501 818c7501 2019-07-11 stsp static const struct got_error *
3502 818c7501 2019-07-11 stsp show_rebase_progress(struct got_commit_object *commit,
3503 818c7501 2019-07-11 stsp struct got_object_id *old_id, struct got_object_id *new_id)
3504 818c7501 2019-07-11 stsp {
3505 818c7501 2019-07-11 stsp const struct got_error *err;
3506 818c7501 2019-07-11 stsp char *old_id_str = NULL, *new_id_str = NULL;
3507 818c7501 2019-07-11 stsp char *logmsg0 = NULL, *logmsg, *nl;
3508 818c7501 2019-07-11 stsp size_t len;
3509 818c7501 2019-07-11 stsp
3510 818c7501 2019-07-11 stsp err = got_object_id_str(&old_id_str, old_id);
3511 818c7501 2019-07-11 stsp if (err)
3512 818c7501 2019-07-11 stsp goto done;
3513 818c7501 2019-07-11 stsp
3514 ff0d2220 2019-07-11 stsp if (new_id) {
3515 ff0d2220 2019-07-11 stsp err = got_object_id_str(&new_id_str, new_id);
3516 ff0d2220 2019-07-11 stsp if (err)
3517 ff0d2220 2019-07-11 stsp goto done;
3518 ff0d2220 2019-07-11 stsp }
3519 818c7501 2019-07-11 stsp
3520 818c7501 2019-07-11 stsp logmsg0 = strdup(got_object_commit_get_logmsg(commit));
3521 818c7501 2019-07-11 stsp if (logmsg0 == NULL) {
3522 818c7501 2019-07-11 stsp err = got_error_from_errno("strdup");
3523 818c7501 2019-07-11 stsp goto done;
3524 818c7501 2019-07-11 stsp }
3525 818c7501 2019-07-11 stsp logmsg = logmsg0;
3526 818c7501 2019-07-11 stsp
3527 10796104 2019-07-11 stsp while (isspace((unsigned char)logmsg[0]))
3528 818c7501 2019-07-11 stsp logmsg++;
3529 818c7501 2019-07-11 stsp
3530 818c7501 2019-07-11 stsp old_id_str[12] = '\0';
3531 ff0d2220 2019-07-11 stsp if (new_id_str)
3532 ff0d2220 2019-07-11 stsp new_id_str[12] = '\0';
3533 818c7501 2019-07-11 stsp len = strlen(logmsg);
3534 818c7501 2019-07-11 stsp if (len > 42)
3535 818c7501 2019-07-11 stsp len = 42;
3536 818c7501 2019-07-11 stsp logmsg[len] = '\0';
3537 818c7501 2019-07-11 stsp nl = strchr(logmsg, '\n');
3538 818c7501 2019-07-11 stsp if (nl)
3539 818c7501 2019-07-11 stsp *nl = '\0';
3540 ff0d2220 2019-07-11 stsp printf("%s -> %s: %s\n", old_id_str,
3541 ff0d2220 2019-07-11 stsp new_id_str ? new_id_str : "no-op change", logmsg);
3542 818c7501 2019-07-11 stsp done:
3543 818c7501 2019-07-11 stsp free(old_id_str);
3544 818c7501 2019-07-11 stsp free(new_id_str);
3545 818c7501 2019-07-11 stsp free(logmsg0);
3546 818c7501 2019-07-11 stsp return err;
3547 818c7501 2019-07-11 stsp }
3548 818c7501 2019-07-11 stsp
3549 1ee397ad 2019-07-12 stsp static const struct got_error *
3550 818c7501 2019-07-11 stsp rebase_progress(void *arg, unsigned char status, const char *path)
3551 818c7501 2019-07-11 stsp {
3552 818c7501 2019-07-11 stsp unsigned char *rebase_status = arg;
3553 818c7501 2019-07-11 stsp
3554 818c7501 2019-07-11 stsp while (path[0] == '/')
3555 818c7501 2019-07-11 stsp path++;
3556 818c7501 2019-07-11 stsp printf("%c %s\n", status, path);
3557 818c7501 2019-07-11 stsp
3558 818c7501 2019-07-11 stsp if (*rebase_status == GOT_STATUS_CONFLICT)
3559 1ee397ad 2019-07-12 stsp return NULL;
3560 818c7501 2019-07-11 stsp if (status == GOT_STATUS_CONFLICT || status == GOT_STATUS_MERGE)
3561 818c7501 2019-07-11 stsp *rebase_status = status;
3562 1ee397ad 2019-07-12 stsp return NULL;
3563 818c7501 2019-07-11 stsp }
3564 818c7501 2019-07-11 stsp
3565 818c7501 2019-07-11 stsp static const struct got_error *
3566 818c7501 2019-07-11 stsp rebase_complete(struct got_worktree *worktree, struct got_reference *branch,
3567 818c7501 2019-07-11 stsp struct got_reference *new_base_branch, struct got_reference *tmp_branch,
3568 818c7501 2019-07-11 stsp struct got_repository *repo)
3569 818c7501 2019-07-11 stsp {
3570 818c7501 2019-07-11 stsp printf("Switching work tree to %s\n", got_ref_get_name(branch));
3571 818c7501 2019-07-11 stsp return got_worktree_rebase_complete(worktree,
3572 818c7501 2019-07-11 stsp new_base_branch, tmp_branch, branch, repo);
3573 818c7501 2019-07-11 stsp }
3574 818c7501 2019-07-11 stsp
3575 818c7501 2019-07-11 stsp static const struct got_error *
3576 01757395 2019-07-12 stsp rebase_commit(struct got_pathlist_head *merged_paths,
3577 01757395 2019-07-12 stsp struct got_worktree *worktree, struct got_reference *tmp_branch,
3578 ff0d2220 2019-07-11 stsp struct got_object_id *commit_id, struct got_repository *repo)
3579 ff0d2220 2019-07-11 stsp {
3580 ff0d2220 2019-07-11 stsp const struct got_error *error;
3581 ff0d2220 2019-07-11 stsp struct got_commit_object *commit;
3582 ff0d2220 2019-07-11 stsp struct got_object_id *new_commit_id;
3583 ff0d2220 2019-07-11 stsp
3584 ff0d2220 2019-07-11 stsp error = got_object_open_as_commit(&commit, repo, commit_id);
3585 ff0d2220 2019-07-11 stsp if (error)
3586 ff0d2220 2019-07-11 stsp return error;
3587 ff0d2220 2019-07-11 stsp
3588 01757395 2019-07-12 stsp error = got_worktree_rebase_commit(&new_commit_id, merged_paths,
3589 01757395 2019-07-12 stsp worktree, tmp_branch, commit, commit_id, repo);
3590 ff0d2220 2019-07-11 stsp if (error) {
3591 ff0d2220 2019-07-11 stsp if (error->code != GOT_ERR_COMMIT_NO_CHANGES)
3592 ff0d2220 2019-07-11 stsp goto done;
3593 ff0d2220 2019-07-11 stsp error = show_rebase_progress(commit, commit_id, NULL);
3594 ff0d2220 2019-07-11 stsp } else {
3595 ff0d2220 2019-07-11 stsp error = show_rebase_progress(commit, commit_id, new_commit_id);
3596 ff0d2220 2019-07-11 stsp free(new_commit_id);
3597 ff0d2220 2019-07-11 stsp }
3598 ff0d2220 2019-07-11 stsp done:
3599 ff0d2220 2019-07-11 stsp got_object_commit_close(commit);
3600 ff0d2220 2019-07-11 stsp return error;
3601 64c6d990 2019-07-11 stsp }
3602 64c6d990 2019-07-11 stsp
3603 64c6d990 2019-07-11 stsp struct check_path_prefix_arg {
3604 64c6d990 2019-07-11 stsp const char *path_prefix;
3605 64c6d990 2019-07-11 stsp size_t len;
3606 64c6d990 2019-07-11 stsp };
3607 64c6d990 2019-07-11 stsp
3608 64c6d990 2019-07-11 stsp static const struct got_error *
3609 64c6d990 2019-07-11 stsp check_path_prefix(void *arg, struct got_blob_object *blob1,
3610 64c6d990 2019-07-11 stsp struct got_blob_object *blob2, struct got_object_id *id1,
3611 64c6d990 2019-07-11 stsp struct got_object_id *id2, const char *path1, const char *path2,
3612 64c6d990 2019-07-11 stsp struct got_repository *repo)
3613 64c6d990 2019-07-11 stsp {
3614 64c6d990 2019-07-11 stsp struct check_path_prefix_arg *a = arg;
3615 64c6d990 2019-07-11 stsp
3616 64c6d990 2019-07-11 stsp if ((path1 && !got_path_is_child(path1, a->path_prefix, a->len)) ||
3617 64c6d990 2019-07-11 stsp (path2 && !got_path_is_child(path2, a->path_prefix, a->len)))
3618 64c6d990 2019-07-11 stsp return got_error(GOT_ERR_REBASE_PATH);
3619 64c6d990 2019-07-11 stsp
3620 64c6d990 2019-07-11 stsp return NULL;
3621 64c6d990 2019-07-11 stsp }
3622 64c6d990 2019-07-11 stsp
3623 64c6d990 2019-07-11 stsp static const struct got_error *
3624 64c6d990 2019-07-11 stsp rebase_check_path_prefix(struct got_object_id *parent_id,
3625 64c6d990 2019-07-11 stsp struct got_object_id *commit_id, const char *path_prefix,
3626 64c6d990 2019-07-11 stsp struct got_repository *repo)
3627 64c6d990 2019-07-11 stsp {
3628 64c6d990 2019-07-11 stsp const struct got_error *err;
3629 64c6d990 2019-07-11 stsp struct got_tree_object *tree1 = NULL, *tree2 = NULL;
3630 64c6d990 2019-07-11 stsp struct got_commit_object *commit = NULL, *parent_commit = NULL;
3631 64c6d990 2019-07-11 stsp struct check_path_prefix_arg cpp_arg;
3632 64c6d990 2019-07-11 stsp
3633 64c6d990 2019-07-11 stsp if (got_path_is_root_dir(path_prefix))
3634 64c6d990 2019-07-11 stsp return NULL;
3635 64c6d990 2019-07-11 stsp
3636 64c6d990 2019-07-11 stsp err = got_object_open_as_commit(&commit, repo, commit_id);
3637 64c6d990 2019-07-11 stsp if (err)
3638 64c6d990 2019-07-11 stsp goto done;
3639 64c6d990 2019-07-11 stsp
3640 64c6d990 2019-07-11 stsp err = got_object_open_as_commit(&parent_commit, repo, parent_id);
3641 64c6d990 2019-07-11 stsp if (err)
3642 64c6d990 2019-07-11 stsp goto done;
3643 64c6d990 2019-07-11 stsp
3644 64c6d990 2019-07-11 stsp err = got_object_open_as_tree(&tree1, repo,
3645 64c6d990 2019-07-11 stsp got_object_commit_get_tree_id(parent_commit));
3646 64c6d990 2019-07-11 stsp if (err)
3647 64c6d990 2019-07-11 stsp goto done;
3648 64c6d990 2019-07-11 stsp
3649 64c6d990 2019-07-11 stsp err = got_object_open_as_tree(&tree2, repo,
3650 64c6d990 2019-07-11 stsp got_object_commit_get_tree_id(commit));
3651 64c6d990 2019-07-11 stsp if (err)
3652 64c6d990 2019-07-11 stsp goto done;
3653 64c6d990 2019-07-11 stsp
3654 64c6d990 2019-07-11 stsp cpp_arg.path_prefix = path_prefix;
3655 64c6d990 2019-07-11 stsp cpp_arg.len = strlen(path_prefix);
3656 64c6d990 2019-07-11 stsp err = got_diff_tree(tree1, tree2, "", "", repo, check_path_prefix,
3657 64c6d990 2019-07-11 stsp &cpp_arg);
3658 64c6d990 2019-07-11 stsp done:
3659 64c6d990 2019-07-11 stsp if (tree1)
3660 64c6d990 2019-07-11 stsp got_object_tree_close(tree1);
3661 64c6d990 2019-07-11 stsp if (tree2)
3662 64c6d990 2019-07-11 stsp got_object_tree_close(tree2);
3663 64c6d990 2019-07-11 stsp if (commit)
3664 64c6d990 2019-07-11 stsp got_object_commit_close(commit);
3665 64c6d990 2019-07-11 stsp if (parent_commit)
3666 64c6d990 2019-07-11 stsp got_object_commit_close(parent_commit);
3667 64c6d990 2019-07-11 stsp return err;
3668 ff0d2220 2019-07-11 stsp }
3669 ff0d2220 2019-07-11 stsp
3670 ff0d2220 2019-07-11 stsp static const struct got_error *
3671 818c7501 2019-07-11 stsp cmd_rebase(int argc, char *argv[])
3672 818c7501 2019-07-11 stsp {
3673 818c7501 2019-07-11 stsp const struct got_error *error = NULL;
3674 818c7501 2019-07-11 stsp struct got_worktree *worktree = NULL;
3675 818c7501 2019-07-11 stsp struct got_repository *repo = NULL;
3676 818c7501 2019-07-11 stsp char *cwd = NULL;
3677 818c7501 2019-07-11 stsp struct got_reference *branch = NULL;
3678 818c7501 2019-07-11 stsp struct got_reference *new_base_branch = NULL, *tmp_branch = NULL;
3679 818c7501 2019-07-11 stsp struct got_object_id *commit_id = NULL, *parent_id = NULL;
3680 818c7501 2019-07-11 stsp struct got_object_id *resume_commit_id = NULL;
3681 818c7501 2019-07-11 stsp struct got_object_id *branch_head_commit_id = NULL, *yca_id = NULL;
3682 818c7501 2019-07-11 stsp struct got_commit_graph *graph = NULL;
3683 818c7501 2019-07-11 stsp struct got_commit_object *commit = NULL;
3684 818c7501 2019-07-11 stsp int ch, rebase_in_progress = 0, abort_rebase = 0, continue_rebase = 0;
3685 818c7501 2019-07-11 stsp unsigned char rebase_status = GOT_STATUS_NO_CHANGE;
3686 818c7501 2019-07-11 stsp struct got_object_id_queue commits;
3687 01757395 2019-07-12 stsp struct got_pathlist_head merged_paths;
3688 818c7501 2019-07-11 stsp const struct got_object_id_queue *parent_ids;
3689 818c7501 2019-07-11 stsp struct got_object_qid *qid, *pid;
3690 818c7501 2019-07-11 stsp
3691 818c7501 2019-07-11 stsp SIMPLEQ_INIT(&commits);
3692 01757395 2019-07-12 stsp TAILQ_INIT(&merged_paths);
3693 818c7501 2019-07-11 stsp
3694 818c7501 2019-07-11 stsp while ((ch = getopt(argc, argv, "ac")) != -1) {
3695 818c7501 2019-07-11 stsp switch (ch) {
3696 818c7501 2019-07-11 stsp case 'a':
3697 818c7501 2019-07-11 stsp abort_rebase = 1;
3698 818c7501 2019-07-11 stsp break;
3699 818c7501 2019-07-11 stsp case 'c':
3700 818c7501 2019-07-11 stsp continue_rebase = 1;
3701 818c7501 2019-07-11 stsp break;
3702 818c7501 2019-07-11 stsp default:
3703 818c7501 2019-07-11 stsp usage_rebase();
3704 818c7501 2019-07-11 stsp /* NOTREACHED */
3705 818c7501 2019-07-11 stsp }
3706 818c7501 2019-07-11 stsp }
3707 818c7501 2019-07-11 stsp
3708 818c7501 2019-07-11 stsp argc -= optind;
3709 818c7501 2019-07-11 stsp argv += optind;
3710 818c7501 2019-07-11 stsp
3711 43012d58 2019-07-14 stsp #ifndef PROFILE
3712 43012d58 2019-07-14 stsp if (pledge("stdio rpath wpath cpath fattr flock proc exec sendfd "
3713 43012d58 2019-07-14 stsp "unveil", NULL) == -1)
3714 43012d58 2019-07-14 stsp err(1, "pledge");
3715 43012d58 2019-07-14 stsp #endif
3716 818c7501 2019-07-11 stsp if (abort_rebase && continue_rebase)
3717 818c7501 2019-07-11 stsp usage_rebase();
3718 818c7501 2019-07-11 stsp else if (abort_rebase || continue_rebase) {
3719 818c7501 2019-07-11 stsp if (argc != 0)
3720 818c7501 2019-07-11 stsp usage_rebase();
3721 818c7501 2019-07-11 stsp } else if (argc != 1)
3722 818c7501 2019-07-11 stsp usage_rebase();
3723 818c7501 2019-07-11 stsp
3724 818c7501 2019-07-11 stsp cwd = getcwd(NULL, 0);
3725 818c7501 2019-07-11 stsp if (cwd == NULL) {
3726 818c7501 2019-07-11 stsp error = got_error_from_errno("getcwd");
3727 818c7501 2019-07-11 stsp goto done;
3728 818c7501 2019-07-11 stsp }
3729 818c7501 2019-07-11 stsp error = got_worktree_open(&worktree, cwd);
3730 818c7501 2019-07-11 stsp if (error)
3731 818c7501 2019-07-11 stsp goto done;
3732 818c7501 2019-07-11 stsp
3733 818c7501 2019-07-11 stsp error = got_repo_open(&repo, got_worktree_get_repo_path(worktree));
3734 818c7501 2019-07-11 stsp if (error != NULL)
3735 818c7501 2019-07-11 stsp goto done;
3736 818c7501 2019-07-11 stsp
3737 818c7501 2019-07-11 stsp error = apply_unveil(got_repo_get_path(repo), 0,
3738 818c7501 2019-07-11 stsp got_worktree_get_root_path(worktree), 0);
3739 818c7501 2019-07-11 stsp if (error)
3740 818c7501 2019-07-11 stsp goto done;
3741 818c7501 2019-07-11 stsp
3742 818c7501 2019-07-11 stsp error = got_worktree_rebase_in_progress(&rebase_in_progress, worktree);
3743 818c7501 2019-07-11 stsp if (error)
3744 818c7501 2019-07-11 stsp goto done;
3745 818c7501 2019-07-11 stsp
3746 818c7501 2019-07-11 stsp if (rebase_in_progress && abort_rebase) {
3747 818c7501 2019-07-11 stsp int did_something;
3748 818c7501 2019-07-11 stsp error = got_worktree_rebase_continue(&resume_commit_id,
3749 818c7501 2019-07-11 stsp &new_base_branch, &tmp_branch, &branch, worktree, repo);
3750 818c7501 2019-07-11 stsp if (error)
3751 818c7501 2019-07-11 stsp goto done;
3752 818c7501 2019-07-11 stsp printf("Switching work tree to %s\n",
3753 818c7501 2019-07-11 stsp got_ref_get_symref_target(new_base_branch));
3754 818c7501 2019-07-11 stsp error = got_worktree_rebase_abort(worktree, repo,
3755 818c7501 2019-07-11 stsp new_base_branch, update_progress, &did_something);
3756 818c7501 2019-07-11 stsp if (error)
3757 818c7501 2019-07-11 stsp goto done;
3758 818c7501 2019-07-11 stsp printf("Rebase of %s aborted\n", got_ref_get_name(branch));
3759 818c7501 2019-07-11 stsp goto done; /* nothing else to do */
3760 818c7501 2019-07-11 stsp } else if (abort_rebase) {
3761 818c7501 2019-07-11 stsp error = got_error(GOT_ERR_NOT_REBASING);
3762 818c7501 2019-07-11 stsp goto done;
3763 818c7501 2019-07-11 stsp }
3764 818c7501 2019-07-11 stsp
3765 818c7501 2019-07-11 stsp if (continue_rebase) {
3766 818c7501 2019-07-11 stsp error = got_worktree_rebase_continue(&resume_commit_id,
3767 818c7501 2019-07-11 stsp &new_base_branch, &tmp_branch, &branch, worktree, repo);
3768 818c7501 2019-07-11 stsp if (error)
3769 818c7501 2019-07-11 stsp goto done;
3770 818c7501 2019-07-11 stsp
3771 01757395 2019-07-12 stsp error = rebase_commit(NULL, worktree, tmp_branch,
3772 01757395 2019-07-12 stsp resume_commit_id, repo);
3773 818c7501 2019-07-11 stsp if (error)
3774 818c7501 2019-07-11 stsp goto done;
3775 818c7501 2019-07-11 stsp
3776 ff0d2220 2019-07-11 stsp yca_id = got_object_id_dup(resume_commit_id);
3777 ff0d2220 2019-07-11 stsp if (yca_id == NULL) {
3778 818c7501 2019-07-11 stsp error = got_error_from_errno("got_object_id_dup");
3779 818c7501 2019-07-11 stsp goto done;
3780 818c7501 2019-07-11 stsp }
3781 818c7501 2019-07-11 stsp } else {
3782 818c7501 2019-07-11 stsp error = got_ref_open(&branch, repo, argv[0], 0);
3783 818c7501 2019-07-11 stsp if (error != NULL)
3784 818c7501 2019-07-11 stsp goto done;
3785 818c7501 2019-07-11 stsp
3786 818c7501 2019-07-11 stsp error = check_same_branch(
3787 818c7501 2019-07-11 stsp got_worktree_get_base_commit_id(worktree), branch, repo);
3788 818c7501 2019-07-11 stsp if (error) {
3789 818c7501 2019-07-11 stsp if (error->code != GOT_ERR_ANCESTRY)
3790 818c7501 2019-07-11 stsp goto done;
3791 818c7501 2019-07-11 stsp error = NULL;
3792 818c7501 2019-07-11 stsp } else {
3793 818c7501 2019-07-11 stsp error = got_error_msg(GOT_ERR_SAME_BRANCH,
3794 818c7501 2019-07-11 stsp "specified branch resolves to a commit which "
3795 818c7501 2019-07-11 stsp "is already contained in work tree's branch");
3796 818c7501 2019-07-11 stsp goto done;
3797 818c7501 2019-07-11 stsp }
3798 ff0d2220 2019-07-11 stsp }
3799 818c7501 2019-07-11 stsp
3800 ff0d2220 2019-07-11 stsp error = got_ref_resolve(&branch_head_commit_id, repo, branch);
3801 ff0d2220 2019-07-11 stsp if (error)
3802 ff0d2220 2019-07-11 stsp goto done;
3803 ff0d2220 2019-07-11 stsp
3804 ff0d2220 2019-07-11 stsp if (!continue_rebase) {
3805 818c7501 2019-07-11 stsp error = got_commit_graph_find_youngest_common_ancestor(&yca_id,
3806 818c7501 2019-07-11 stsp got_worktree_get_base_commit_id(worktree),
3807 818c7501 2019-07-11 stsp branch_head_commit_id, repo);
3808 818c7501 2019-07-11 stsp if (error)
3809 818c7501 2019-07-11 stsp goto done;
3810 818c7501 2019-07-11 stsp if (yca_id == NULL) {
3811 818c7501 2019-07-11 stsp error = got_error_msg(GOT_ERR_ANCESTRY,
3812 818c7501 2019-07-11 stsp "specified branch shares no common ancestry "
3813 818c7501 2019-07-11 stsp "with work tree's branch");
3814 818c7501 2019-07-11 stsp goto done;
3815 818c7501 2019-07-11 stsp }
3816 818c7501 2019-07-11 stsp
3817 818c7501 2019-07-11 stsp error = got_worktree_rebase_prepare(&new_base_branch,
3818 818c7501 2019-07-11 stsp &tmp_branch, worktree, branch, repo);
3819 818c7501 2019-07-11 stsp if (error)
3820 818c7501 2019-07-11 stsp goto done;
3821 818c7501 2019-07-11 stsp }
3822 818c7501 2019-07-11 stsp
3823 ff0d2220 2019-07-11 stsp commit_id = branch_head_commit_id;
3824 818c7501 2019-07-11 stsp error = got_object_open_as_commit(&commit, repo, commit_id);
3825 818c7501 2019-07-11 stsp if (error)
3826 818c7501 2019-07-11 stsp goto done;
3827 818c7501 2019-07-11 stsp
3828 818c7501 2019-07-11 stsp error = got_commit_graph_open(&graph, commit_id, "/", 1, repo);
3829 818c7501 2019-07-11 stsp if (error)
3830 818c7501 2019-07-11 stsp goto done;
3831 818c7501 2019-07-11 stsp parent_ids = got_object_commit_get_parent_ids(commit);
3832 818c7501 2019-07-11 stsp pid = SIMPLEQ_FIRST(parent_ids);
3833 818c7501 2019-07-11 stsp error = got_commit_graph_iter_start(graph, pid->id, repo);
3834 818c7501 2019-07-11 stsp got_object_commit_close(commit);
3835 818c7501 2019-07-11 stsp commit = NULL;
3836 818c7501 2019-07-11 stsp if (error)
3837 818c7501 2019-07-11 stsp goto done;
3838 818c7501 2019-07-11 stsp while (got_object_id_cmp(commit_id, yca_id) != 0) {
3839 818c7501 2019-07-11 stsp error = got_commit_graph_iter_next(&parent_id, graph);
3840 818c7501 2019-07-11 stsp if (error) {
3841 818c7501 2019-07-11 stsp if (error->code == GOT_ERR_ITER_COMPLETED) {
3842 818c7501 2019-07-11 stsp error = got_error_msg(GOT_ERR_ANCESTRY,
3843 818c7501 2019-07-11 stsp "ran out of commits to rebase before "
3844 818c7501 2019-07-11 stsp "youngest common ancestor commit has "
3845 818c7501 2019-07-11 stsp "been reached?!?");
3846 818c7501 2019-07-11 stsp goto done;
3847 818c7501 2019-07-11 stsp } else if (error->code != GOT_ERR_ITER_NEED_MORE)
3848 818c7501 2019-07-11 stsp goto done;
3849 818c7501 2019-07-11 stsp error = got_commit_graph_fetch_commits(graph, 1, repo);
3850 818c7501 2019-07-11 stsp if (error)
3851 818c7501 2019-07-11 stsp goto done;
3852 818c7501 2019-07-11 stsp } else {
3853 64c6d990 2019-07-11 stsp error = rebase_check_path_prefix(parent_id, commit_id,
3854 64c6d990 2019-07-11 stsp got_worktree_get_path_prefix(worktree), repo);
3855 64c6d990 2019-07-11 stsp if (error)
3856 64c6d990 2019-07-11 stsp goto done;
3857 64c6d990 2019-07-11 stsp
3858 818c7501 2019-07-11 stsp error = got_object_qid_alloc(&qid, commit_id);
3859 818c7501 2019-07-11 stsp if (error)
3860 818c7501 2019-07-11 stsp goto done;
3861 818c7501 2019-07-11 stsp SIMPLEQ_INSERT_HEAD(&commits, qid, entry);
3862 818c7501 2019-07-11 stsp commit_id = parent_id;
3863 818c7501 2019-07-11 stsp }
3864 818c7501 2019-07-11 stsp }
3865 818c7501 2019-07-11 stsp
3866 818c7501 2019-07-11 stsp if (SIMPLEQ_EMPTY(&commits)) {
3867 ff0d2220 2019-07-11 stsp if (continue_rebase)
3868 ff0d2220 2019-07-11 stsp error = rebase_complete(worktree, branch,
3869 ff0d2220 2019-07-11 stsp new_base_branch, tmp_branch, repo);
3870 ff0d2220 2019-07-11 stsp else
3871 ff0d2220 2019-07-11 stsp error = got_error(GOT_ERR_EMPTY_REBASE);
3872 818c7501 2019-07-11 stsp goto done;
3873 818c7501 2019-07-11 stsp }
3874 818c7501 2019-07-11 stsp
3875 818c7501 2019-07-11 stsp pid = NULL;
3876 818c7501 2019-07-11 stsp SIMPLEQ_FOREACH(qid, &commits, entry) {
3877 818c7501 2019-07-11 stsp commit_id = qid->id;
3878 818c7501 2019-07-11 stsp parent_id = pid ? pid->id : yca_id;
3879 818c7501 2019-07-11 stsp pid = qid;
3880 818c7501 2019-07-11 stsp
3881 01757395 2019-07-12 stsp error = got_worktree_rebase_merge_files(&merged_paths,
3882 01757395 2019-07-12 stsp worktree, parent_id, commit_id, repo, rebase_progress,
3883 01757395 2019-07-12 stsp &rebase_status, check_cancelled, NULL);
3884 818c7501 2019-07-11 stsp if (error)
3885 818c7501 2019-07-11 stsp goto done;
3886 818c7501 2019-07-11 stsp
3887 01757395 2019-07-12 stsp if (rebase_status == GOT_STATUS_CONFLICT) {
3888 01757395 2019-07-12 stsp got_worktree_rebase_pathlist_free(&merged_paths);
3889 818c7501 2019-07-11 stsp break;
3890 01757395 2019-07-12 stsp }
3891 818c7501 2019-07-11 stsp
3892 01757395 2019-07-12 stsp error = rebase_commit(&merged_paths, worktree, tmp_branch,
3893 01757395 2019-07-12 stsp commit_id, repo);
3894 01757395 2019-07-12 stsp got_worktree_rebase_pathlist_free(&merged_paths);
3895 818c7501 2019-07-11 stsp if (error)
3896 818c7501 2019-07-11 stsp goto done;
3897 818c7501 2019-07-11 stsp }
3898 818c7501 2019-07-11 stsp
3899 818c7501 2019-07-11 stsp if (rebase_status == GOT_STATUS_CONFLICT) {
3900 818c7501 2019-07-11 stsp error = got_worktree_rebase_postpone(worktree);
3901 818c7501 2019-07-11 stsp if (error)
3902 818c7501 2019-07-11 stsp goto done;
3903 818c7501 2019-07-11 stsp error = got_error_msg(GOT_ERR_CONFLICTS,
3904 11495e04 2019-07-12 stsp "conflicts must be resolved before rebasing can continue");
3905 818c7501 2019-07-11 stsp } else
3906 818c7501 2019-07-11 stsp error = rebase_complete(worktree, branch, new_base_branch,
3907 818c7501 2019-07-11 stsp tmp_branch, repo);
3908 818c7501 2019-07-11 stsp done:
3909 818c7501 2019-07-11 stsp got_object_id_queue_free(&commits);
3910 818c7501 2019-07-11 stsp free(branch_head_commit_id);
3911 818c7501 2019-07-11 stsp free(resume_commit_id);
3912 818c7501 2019-07-11 stsp free(yca_id);
3913 818c7501 2019-07-11 stsp if (graph)
3914 818c7501 2019-07-11 stsp got_commit_graph_close(graph);
3915 818c7501 2019-07-11 stsp if (commit)
3916 818c7501 2019-07-11 stsp got_object_commit_close(commit);
3917 818c7501 2019-07-11 stsp if (branch)
3918 818c7501 2019-07-11 stsp got_ref_close(branch);
3919 818c7501 2019-07-11 stsp if (new_base_branch)
3920 818c7501 2019-07-11 stsp got_ref_close(new_base_branch);
3921 818c7501 2019-07-11 stsp if (tmp_branch)
3922 818c7501 2019-07-11 stsp got_ref_close(tmp_branch);
3923 5ef14e63 2019-06-02 stsp if (worktree)
3924 5ef14e63 2019-06-02 stsp got_worktree_close(worktree);
3925 5ef14e63 2019-06-02 stsp if (repo)
3926 5ef14e63 2019-06-02 stsp got_repo_close(repo);
3927 5ef14e63 2019-06-02 stsp return error;
3928 5ef14e63 2019-06-02 stsp }