]>
Commit | Line | Data |
---|---|---|
0a43fb22 JS |
1 | /* |
2 | * The Scalar command-line interface. | |
3 | */ | |
4 | ||
5579f44d | 5 | #include "git-compat-util.h" |
0b027f6c | 6 | #include "abspath.h" |
0a43fb22 JS |
7 | #include "gettext.h" |
8 | #include "parse-options.h" | |
d0feac4e DS |
9 | #include "config.h" |
10 | #include "run-command.h" | |
3f1917dc MJC |
11 | #include "simple-ipc.h" |
12 | #include "fsmonitor-ipc.h" | |
13 | #include "fsmonitor-settings.h" | |
546f822d | 14 | #include "refs.h" |
d85ada7c MJC |
15 | #include "dir.h" |
16 | #include "packfile.h" | |
ddc35d83 | 17 | #include "help.h" |
e38da487 | 18 | #include "setup.h" |
74ea5c95 | 19 | #include "trace2.h" |
d0feac4e | 20 | |
d0feac4e DS |
21 | static void setup_enlistment_directory(int argc, const char **argv, |
22 | const char * const *usagestr, | |
23 | const struct option *options, | |
24 | struct strbuf *enlistment_root) | |
25 | { | |
26 | struct strbuf path = STRBUF_INIT; | |
65f6a9eb VD |
27 | int enlistment_is_repo_parent = 0; |
28 | size_t len; | |
d0feac4e DS |
29 | |
30 | if (startup_info->have_repository) | |
31 | BUG("gitdir already set up?!?"); | |
32 | ||
33 | if (argc > 1) | |
34 | usage_with_options(usagestr, options); | |
35 | ||
36 | /* find the worktree, determine its corresponding root */ | |
b4485574 | 37 | if (argc == 1) { |
d0feac4e | 38 | strbuf_add_absolute_path(&path, argv[0]); |
b4485574 JS |
39 | if (!is_directory(path.buf)) |
40 | die(_("'%s' does not exist"), path.buf); | |
65f6a9eb VD |
41 | if (chdir(path.buf) < 0) |
42 | die_errno(_("could not switch to '%s'"), path.buf); | |
b4485574 | 43 | } else if (strbuf_getcwd(&path) < 0) |
d0feac4e DS |
44 | die(_("need a working directory")); |
45 | ||
46 | strbuf_trim_trailing_dir_sep(&path); | |
d0feac4e | 47 | |
65f6a9eb VD |
48 | /* check if currently in enlistment root with src/ workdir */ |
49 | len = path.len; | |
50 | strbuf_addstr(&path, "/src"); | |
51 | if (is_nonbare_repository_dir(&path)) { | |
52 | enlistment_is_repo_parent = 1; | |
53 | if (chdir(path.buf) < 0) | |
54 | die_errno(_("could not switch to '%s'"), path.buf); | |
55 | } | |
56 | strbuf_setlen(&path, len); | |
d0feac4e | 57 | |
65f6a9eb | 58 | setup_git_directory(); |
d0feac4e | 59 | |
65f6a9eb VD |
60 | if (!the_repository->worktree) |
61 | die(_("Scalar enlistments require a worktree")); | |
d0feac4e | 62 | |
65f6a9eb VD |
63 | if (enlistment_root) { |
64 | if (enlistment_is_repo_parent) | |
65 | strbuf_addbuf(enlistment_root, &path); | |
66 | else | |
67 | strbuf_addstr(enlistment_root, the_repository->worktree); | |
68 | } | |
d0feac4e DS |
69 | |
70 | strbuf_release(&path); | |
d0feac4e DS |
71 | } |
72 | ||
73 | static int run_git(const char *arg, ...) | |
74 | { | |
0e906739 | 75 | struct child_process cmd = CHILD_PROCESS_INIT; |
d0feac4e DS |
76 | va_list args; |
77 | const char *p; | |
d0feac4e DS |
78 | |
79 | va_start(args, arg); | |
0e906739 | 80 | strvec_push(&cmd.args, arg); |
d0feac4e | 81 | while ((p = va_arg(args, const char *))) |
0e906739 | 82 | strvec_push(&cmd.args, p); |
d0feac4e DS |
83 | va_end(args); |
84 | ||
0e906739 RS |
85 | cmd.git_cmd = 1; |
86 | return run_command(&cmd); | |
d0feac4e DS |
87 | } |
88 | ||
d934a11c VD |
89 | struct scalar_config { |
90 | const char *key; | |
91 | const char *value; | |
92 | int overwrite_on_reconfigure; | |
93 | }; | |
94 | ||
95 | static int set_scalar_config(const struct scalar_config *config, int reconfigure) | |
96 | { | |
97 | char *value = NULL; | |
98 | int res; | |
99 | ||
100 | if ((reconfigure && config->overwrite_on_reconfigure) || | |
101 | git_config_get_string(config->key, &value)) { | |
102 | trace2_data_string("scalar", the_repository, config->key, "created"); | |
103 | res = git_config_set_gently(config->key, config->value); | |
104 | } else { | |
105 | trace2_data_string("scalar", the_repository, config->key, "exists"); | |
106 | res = 0; | |
107 | } | |
108 | ||
109 | free(value); | |
110 | return res; | |
111 | } | |
112 | ||
3f1917dc MJC |
113 | static int have_fsmonitor_support(void) |
114 | { | |
115 | return fsmonitor_ipc__is_supported() && | |
116 | fsm_settings__get_reason(the_repository) == FSMONITOR_REASON_OK; | |
117 | } | |
118 | ||
cb59d55e | 119 | static int set_recommended_config(int reconfigure) |
d0feac4e | 120 | { |
d934a11c | 121 | struct scalar_config config[] = { |
cb59d55e JS |
122 | /* Required */ |
123 | { "am.keepCR", "true", 1 }, | |
124 | { "core.FSCache", "true", 1 }, | |
125 | { "core.multiPackIndex", "true", 1 }, | |
126 | { "core.preloadIndex", "true", 1 }, | |
d0feac4e | 127 | #ifndef WIN32 |
cb59d55e | 128 | { "core.untrackedCache", "true", 1 }, |
d0feac4e DS |
129 | #else |
130 | /* | |
131 | * Unfortunately, Scalar's Functional Tests demonstrated | |
132 | * that the untracked cache feature is unreliable on Windows | |
133 | * (which is a bummer because that platform would benefit the | |
134 | * most from it). For some reason, freshly created files seem | |
135 | * not to update the directory's `lastModified` time | |
136 | * immediately, but the untracked cache would need to rely on | |
137 | * that. | |
138 | * | |
139 | * Therefore, with a sad heart, we disable this very useful | |
140 | * feature on Windows. | |
141 | */ | |
cb59d55e | 142 | { "core.untrackedCache", "false", 1 }, |
d0feac4e | 143 | #endif |
cb59d55e JS |
144 | { "core.logAllRefUpdates", "true", 1 }, |
145 | { "credential.https://dev.azure.com.useHttpPath", "true", 1 }, | |
146 | { "credential.validate", "false", 1 }, /* GCM4W-only */ | |
147 | { "gc.auto", "0", 1 }, | |
148 | { "gui.GCWarning", "false", 1 }, | |
17194b19 | 149 | { "index.skipHash", "false", 1 }, |
cb59d55e JS |
150 | { "index.threads", "true", 1 }, |
151 | { "index.version", "4", 1 }, | |
152 | { "merge.stat", "false", 1 }, | |
153 | { "merge.renames", "true", 1 }, | |
154 | { "pack.useBitmaps", "false", 1 }, | |
155 | { "pack.useSparse", "true", 1 }, | |
156 | { "receive.autoGC", "false", 1 }, | |
cb59d55e JS |
157 | { "feature.manyFiles", "false", 1 }, |
158 | { "feature.experimental", "false", 1 }, | |
159 | { "fetch.unpackLimit", "1", 1 }, | |
160 | { "fetch.writeCommitGraph", "false", 1 }, | |
d0feac4e | 161 | #ifdef WIN32 |
cb59d55e | 162 | { "http.sslBackend", "schannel", 1 }, |
d0feac4e | 163 | #endif |
cb59d55e | 164 | /* Optional */ |
d0feac4e DS |
165 | { "status.aheadBehind", "false" }, |
166 | { "commitGraph.generationVersion", "1" }, | |
167 | { "core.autoCRLF", "false" }, | |
168 | { "core.safeCRLF", "false" }, | |
169 | { "fetch.showForcedUpdates", "false" }, | |
170 | { NULL, NULL }, | |
171 | }; | |
172 | int i; | |
173 | char *value; | |
174 | ||
175 | for (i = 0; config[i].key; i++) { | |
d934a11c VD |
176 | if (set_scalar_config(config + i, reconfigure)) |
177 | return error(_("could not configure %s=%s"), | |
178 | config[i].key, config[i].value); | |
d0feac4e DS |
179 | } |
180 | ||
3f1917dc MJC |
181 | if (have_fsmonitor_support()) { |
182 | struct scalar_config fsmonitor = { "core.fsmonitor", "true" }; | |
183 | if (set_scalar_config(&fsmonitor, reconfigure)) | |
184 | return error(_("could not configure %s=%s"), | |
185 | fsmonitor.key, fsmonitor.value); | |
d0feac4e DS |
186 | } |
187 | ||
188 | /* | |
189 | * The `log.excludeDecoration` setting is special because it allows | |
190 | * for multiple values. | |
191 | */ | |
192 | if (git_config_get_string("log.excludeDecoration", &value)) { | |
193 | trace2_data_string("scalar", the_repository, | |
194 | "log.excludeDecoration", "created"); | |
195 | if (git_config_set_multivar_gently("log.excludeDecoration", | |
196 | "refs/prefetch/*", | |
197 | CONFIG_REGEX_NONE, 0)) | |
198 | return error(_("could not configure " | |
199 | "log.excludeDecoration")); | |
200 | } else { | |
201 | trace2_data_string("scalar", the_repository, | |
202 | "log.excludeDecoration", "exists"); | |
203 | free(value); | |
204 | } | |
205 | ||
206 | return 0; | |
207 | } | |
208 | ||
c76a53eb | 209 | static int toggle_maintenance(int enable) |
d0feac4e | 210 | { |
d871b6c6 DS |
211 | return run_git("maintenance", |
212 | enable ? "start" : "unregister", | |
213 | enable ? NULL : "--force", | |
214 | NULL); | |
d0feac4e DS |
215 | } |
216 | ||
c76a53eb | 217 | static int add_or_remove_enlistment(int add) |
d0feac4e DS |
218 | { |
219 | int res; | |
220 | ||
221 | if (!the_repository->worktree) | |
222 | die(_("Scalar enlistments require a worktree")); | |
223 | ||
224 | res = run_git("config", "--global", "--get", "--fixed-value", | |
225 | "scalar.repo", the_repository->worktree, NULL); | |
226 | ||
227 | /* | |
c76a53eb DS |
228 | * If we want to add and the setting is already there, then do nothing. |
229 | * If we want to remove and the setting is not there, then do nothing. | |
d0feac4e | 230 | */ |
c76a53eb | 231 | if ((add && !res) || (!add && res)) |
d0feac4e DS |
232 | return 0; |
233 | ||
c76a53eb DS |
234 | return run_git("config", "--global", add ? "--add" : "--unset", |
235 | add ? "--no-fixed-value" : "--fixed-value", | |
d0feac4e DS |
236 | "scalar.repo", the_repository->worktree, NULL); |
237 | } | |
238 | ||
3f1917dc MJC |
239 | static int start_fsmonitor_daemon(void) |
240 | { | |
241 | assert(have_fsmonitor_support()); | |
242 | ||
243 | if (fsmonitor_ipc__get_state() != IPC_STATE__LISTENING) | |
244 | return run_git("fsmonitor--daemon", "start", NULL); | |
245 | ||
246 | return 0; | |
247 | } | |
248 | ||
ec4c2311 JS |
249 | static int stop_fsmonitor_daemon(void) |
250 | { | |
251 | assert(have_fsmonitor_support()); | |
252 | ||
253 | if (fsmonitor_ipc__get_state() == IPC_STATE__LISTENING) | |
254 | return run_git("fsmonitor--daemon", "stop", NULL); | |
255 | ||
256 | return 0; | |
257 | } | |
258 | ||
d0feac4e DS |
259 | static int register_dir(void) |
260 | { | |
d2a79bc9 VD |
261 | if (add_or_remove_enlistment(1)) |
262 | return error(_("could not add enlistment")); | |
d0feac4e | 263 | |
d2a79bc9 VD |
264 | if (set_recommended_config(0)) |
265 | return error(_("could not set recommended config")); | |
d0feac4e | 266 | |
d2a79bc9 | 267 | if (toggle_maintenance(1)) |
dea63088 | 268 | warning(_("could not turn on maintenance")); |
c76a53eb | 269 | |
3f1917dc MJC |
270 | if (have_fsmonitor_support() && start_fsmonitor_daemon()) { |
271 | return error(_("could not start the FSMonitor daemon")); | |
272 | } | |
273 | ||
d2a79bc9 | 274 | return 0; |
c76a53eb DS |
275 | } |
276 | ||
277 | static int unregister_dir(void) | |
278 | { | |
279 | int res = 0; | |
280 | ||
adedcee8 | 281 | if (toggle_maintenance(0)) |
d2a79bc9 | 282 | res = error(_("could not turn off maintenance")); |
c76a53eb | 283 | |
adedcee8 | 284 | if (add_or_remove_enlistment(0)) |
d2a79bc9 | 285 | res = error(_("could not remove enlistment")); |
d0feac4e DS |
286 | |
287 | return res; | |
288 | } | |
289 | ||
546f822d JS |
290 | /* printf-style interface, expects `<key>=<value>` argument */ |
291 | static int set_config(const char *fmt, ...) | |
292 | { | |
293 | struct strbuf buf = STRBUF_INIT; | |
294 | char *value; | |
295 | int res; | |
296 | va_list args; | |
297 | ||
298 | va_start(args, fmt); | |
299 | strbuf_vaddf(&buf, fmt, args); | |
300 | va_end(args); | |
301 | ||
302 | value = strchr(buf.buf, '='); | |
303 | if (value) | |
304 | *(value++) = '\0'; | |
305 | res = git_config_set_gently(buf.buf, value); | |
306 | strbuf_release(&buf); | |
307 | ||
308 | return res; | |
309 | } | |
310 | ||
311 | static char *remote_default_branch(const char *url) | |
312 | { | |
313 | struct child_process cp = CHILD_PROCESS_INIT; | |
314 | struct strbuf out = STRBUF_INIT; | |
315 | ||
316 | cp.git_cmd = 1; | |
317 | strvec_pushl(&cp.args, "ls-remote", "--symref", url, "HEAD", NULL); | |
318 | if (!pipe_command(&cp, NULL, 0, &out, 0, NULL, 0)) { | |
319 | const char *line = out.buf; | |
320 | ||
321 | while (*line) { | |
322 | const char *eol = strchrnul(line, '\n'), *p; | |
323 | size_t len = eol - line; | |
324 | char *branch; | |
325 | ||
326 | if (!skip_prefix(line, "ref: ", &p) || | |
327 | !strip_suffix_mem(line, &len, "\tHEAD")) { | |
328 | line = eol + (*eol == '\n'); | |
329 | continue; | |
330 | } | |
331 | ||
332 | eol = line + len; | |
333 | if (skip_prefix(p, "refs/heads/", &p)) { | |
334 | branch = xstrndup(p, eol - p); | |
335 | strbuf_release(&out); | |
336 | return branch; | |
337 | } | |
338 | ||
339 | error(_("remote HEAD is not a branch: '%.*s'"), | |
340 | (int)(eol - p), p); | |
341 | strbuf_release(&out); | |
342 | return NULL; | |
343 | } | |
344 | } | |
345 | warning(_("failed to get default branch name from remote; " | |
346 | "using local default")); | |
347 | strbuf_reset(&out); | |
348 | ||
349 | child_process_init(&cp); | |
350 | cp.git_cmd = 1; | |
351 | strvec_pushl(&cp.args, "symbolic-ref", "--short", "HEAD", NULL); | |
352 | if (!pipe_command(&cp, NULL, 0, &out, 0, NULL, 0)) { | |
353 | strbuf_trim(&out); | |
354 | return strbuf_detach(&out, NULL); | |
355 | } | |
356 | ||
357 | strbuf_release(&out); | |
358 | error(_("failed to get default branch name")); | |
359 | return NULL; | |
360 | } | |
361 | ||
d85ada7c MJC |
362 | static int delete_enlistment(struct strbuf *enlistment) |
363 | { | |
364 | #ifdef WIN32 | |
365 | struct strbuf parent = STRBUF_INIT; | |
65f6a9eb VD |
366 | size_t offset; |
367 | char *path_sep; | |
d85ada7c MJC |
368 | #endif |
369 | ||
370 | if (unregister_dir()) | |
9b24bb92 | 371 | return error(_("failed to unregister repository")); |
d85ada7c MJC |
372 | |
373 | #ifdef WIN32 | |
374 | /* | |
375 | * Change the current directory to one outside of the enlistment so | |
376 | * that we may delete everything underneath it. | |
377 | */ | |
65f6a9eb VD |
378 | offset = offset_1st_component(enlistment->buf); |
379 | path_sep = find_last_dir_sep(enlistment->buf + offset); | |
380 | strbuf_add(&parent, enlistment->buf, | |
381 | path_sep ? path_sep - enlistment->buf : offset); | |
9b24bb92 VD |
382 | if (chdir(parent.buf) < 0) { |
383 | int res = error_errno(_("could not switch to '%s'"), parent.buf); | |
384 | strbuf_release(&parent); | |
385 | return res; | |
386 | } | |
d85ada7c MJC |
387 | strbuf_release(&parent); |
388 | #endif | |
389 | ||
ec4c2311 JS |
390 | if (have_fsmonitor_support() && stop_fsmonitor_daemon()) |
391 | return error(_("failed to stop the FSMonitor daemon")); | |
392 | ||
d85ada7c | 393 | if (remove_dir_recursively(enlistment, 0)) |
9b24bb92 | 394 | return error(_("failed to delete enlistment directory")); |
d85ada7c MJC |
395 | |
396 | return 0; | |
397 | } | |
398 | ||
ddc35d83 JS |
399 | /* |
400 | * Dummy implementation; Using `get_version_info()` would cause a link error | |
401 | * without this. | |
402 | */ | |
403 | void load_builtin_commands(const char *prefix, struct cmdnames *cmds) | |
404 | { | |
405 | die("not implemented"); | |
406 | } | |
407 | ||
546f822d JS |
408 | static int cmd_clone(int argc, const char **argv) |
409 | { | |
410 | const char *branch = NULL; | |
4433bd24 | 411 | int full_clone = 0, single_branch = 0, show_progress = isatty(2); |
4527db8f | 412 | int src = 1; |
546f822d JS |
413 | struct option clone_options[] = { |
414 | OPT_STRING('b', "branch", &branch, N_("<branch>"), | |
415 | N_("branch to checkout after clone")), | |
416 | OPT_BOOL(0, "full-clone", &full_clone, | |
417 | N_("when cloning, create full working directory")), | |
4368e40b JS |
418 | OPT_BOOL(0, "single-branch", &single_branch, |
419 | N_("only download metadata for the branch that will " | |
420 | "be checked out")), | |
4527db8f DS |
421 | OPT_BOOL(0, "src", &src, |
422 | N_("create repository within 'src' directory")), | |
546f822d JS |
423 | OPT_END(), |
424 | }; | |
425 | const char * const clone_usage[] = { | |
4527db8f DS |
426 | N_("scalar clone [--single-branch] [--branch <main-branch>] [--full-clone]\n" |
427 | "\t[--[no-]src] <url> [<enlistment>]"), | |
546f822d JS |
428 | NULL |
429 | }; | |
430 | const char *url; | |
431 | char *enlistment = NULL, *dir = NULL; | |
432 | struct strbuf buf = STRBUF_INIT; | |
433 | int res; | |
434 | ||
435 | argc = parse_options(argc, argv, NULL, clone_options, clone_usage, 0); | |
436 | ||
437 | if (argc == 2) { | |
438 | url = argv[0]; | |
439 | enlistment = xstrdup(argv[1]); | |
440 | } else if (argc == 1) { | |
441 | url = argv[0]; | |
442 | ||
443 | strbuf_addstr(&buf, url); | |
444 | /* Strip trailing slashes, if any */ | |
445 | while (buf.len > 0 && is_dir_sep(buf.buf[buf.len - 1])) | |
446 | strbuf_setlen(&buf, buf.len - 1); | |
447 | /* Strip suffix `.git`, if any */ | |
448 | strbuf_strip_suffix(&buf, ".git"); | |
449 | ||
450 | enlistment = find_last_dir_sep(buf.buf); | |
451 | if (!enlistment) { | |
452 | die(_("cannot deduce worktree name from '%s'"), url); | |
453 | } | |
454 | enlistment = xstrdup(enlistment + 1); | |
455 | } else { | |
456 | usage_msg_opt(_("You must specify a repository to clone."), | |
457 | clone_usage, clone_options); | |
458 | } | |
459 | ||
460 | if (is_directory(enlistment)) | |
461 | die(_("directory '%s' exists already"), enlistment); | |
462 | ||
4527db8f DS |
463 | if (src) |
464 | dir = xstrfmt("%s/src", enlistment); | |
465 | else | |
466 | dir = xstrdup(enlistment); | |
546f822d JS |
467 | |
468 | strbuf_reset(&buf); | |
469 | if (branch) | |
470 | strbuf_addf(&buf, "init.defaultBranch=%s", branch); | |
471 | else { | |
472 | char *b = repo_default_branch_name(the_repository, 1); | |
473 | strbuf_addf(&buf, "init.defaultBranch=%s", b); | |
474 | free(b); | |
475 | } | |
476 | ||
477 | if ((res = run_git("-c", buf.buf, "init", "--", dir, NULL))) | |
478 | goto cleanup; | |
479 | ||
480 | if (chdir(dir) < 0) { | |
481 | res = error_errno(_("could not switch to '%s'"), dir); | |
482 | goto cleanup; | |
483 | } | |
484 | ||
485 | setup_git_directory(); | |
486 | ||
487 | /* common-main already logs `argv` */ | |
488 | trace2_def_repo(the_repository); | |
489 | ||
490 | if (!branch && !(branch = remote_default_branch(url))) { | |
491 | res = error(_("failed to get default branch for '%s'"), url); | |
492 | goto cleanup; | |
493 | } | |
494 | ||
495 | if (set_config("remote.origin.url=%s", url) || | |
496 | set_config("remote.origin.fetch=" | |
4368e40b JS |
497 | "+refs/heads/%s:refs/remotes/origin/%s", |
498 | single_branch ? branch : "*", | |
499 | single_branch ? branch : "*") || | |
546f822d JS |
500 | set_config("remote.origin.promisor=true") || |
501 | set_config("remote.origin.partialCloneFilter=blob:none")) { | |
502 | res = error(_("could not configure remote in '%s'"), dir); | |
503 | goto cleanup; | |
504 | } | |
505 | ||
506 | if (!full_clone && | |
507 | (res = run_git("sparse-checkout", "init", "--cone", NULL))) | |
508 | goto cleanup; | |
509 | ||
cb59d55e | 510 | if (set_recommended_config(0)) |
546f822d JS |
511 | return error(_("could not configure '%s'"), dir); |
512 | ||
4433bd24 ZH |
513 | if ((res = run_git("fetch", "--quiet", |
514 | show_progress ? "--progress" : "--no-progress", | |
515 | "origin", NULL))) { | |
546f822d JS |
516 | warning(_("partial clone failed; attempting full clone")); |
517 | ||
518 | if (set_config("remote.origin.promisor") || | |
519 | set_config("remote.origin.partialCloneFilter")) { | |
520 | res = error(_("could not configure for full clone")); | |
521 | goto cleanup; | |
522 | } | |
523 | ||
4433bd24 ZH |
524 | if ((res = run_git("fetch", "--quiet", |
525 | show_progress ? "--progress" : "--no-progress", | |
526 | "origin", NULL))) | |
546f822d JS |
527 | goto cleanup; |
528 | } | |
529 | ||
530 | if ((res = set_config("branch.%s.remote=origin", branch))) | |
531 | goto cleanup; | |
532 | if ((res = set_config("branch.%s.merge=refs/heads/%s", | |
533 | branch, branch))) | |
534 | goto cleanup; | |
535 | ||
536 | strbuf_reset(&buf); | |
537 | strbuf_addf(&buf, "origin/%s", branch); | |
538 | res = run_git("checkout", "-f", "-t", buf.buf, NULL); | |
539 | if (res) | |
540 | goto cleanup; | |
541 | ||
542 | res = register_dir(); | |
543 | ||
544 | cleanup: | |
545 | free(enlistment); | |
546 | free(dir); | |
547 | strbuf_release(&buf); | |
548 | return res; | |
549 | } | |
550 | ||
aa5c79a3 JS |
551 | static int cmd_diagnose(int argc, const char **argv) |
552 | { | |
553 | struct option options[] = { | |
554 | OPT_END(), | |
555 | }; | |
556 | const char * const usage[] = { | |
557 | N_("scalar diagnose [<enlistment>]"), | |
558 | NULL | |
559 | }; | |
672196a3 | 560 | struct strbuf diagnostics_root = STRBUF_INIT; |
aa5c79a3 JS |
561 | int res = 0; |
562 | ||
563 | argc = parse_options(argc, argv, NULL, options, | |
564 | usage, 0); | |
565 | ||
672196a3 VD |
566 | setup_enlistment_directory(argc, argv, usage, options, &diagnostics_root); |
567 | strbuf_addstr(&diagnostics_root, "/.scalarDiagnostics"); | |
aa5c79a3 | 568 | |
672196a3 VD |
569 | res = run_git("diagnose", "--mode=all", "-s", "%Y%m%d_%H%M%S", |
570 | "-o", diagnostics_root.buf, NULL); | |
aa5c79a3 | 571 | |
672196a3 | 572 | strbuf_release(&diagnostics_root); |
aa5c79a3 JS |
573 | return res; |
574 | } | |
575 | ||
6ba21fa6 | 576 | static int cmd_list(int argc, const char **argv UNUSED) |
2b710457 DS |
577 | { |
578 | if (argc != 1) | |
579 | die(_("`scalar list` does not take arguments")); | |
580 | ||
581 | if (run_git("config", "--global", "--get-all", "scalar.repo", NULL) < 0) | |
582 | return -1; | |
583 | return 0; | |
584 | } | |
585 | ||
d0feac4e DS |
586 | static int cmd_register(int argc, const char **argv) |
587 | { | |
588 | struct option options[] = { | |
589 | OPT_END(), | |
590 | }; | |
591 | const char * const usage[] = { | |
592 | N_("scalar register [<enlistment>]"), | |
593 | NULL | |
594 | }; | |
595 | ||
596 | argc = parse_options(argc, argv, NULL, options, | |
597 | usage, 0); | |
598 | ||
599 | setup_enlistment_directory(argc, argv, usage, options, NULL); | |
600 | ||
601 | return register_dir(); | |
602 | } | |
0a43fb22 | 603 | |
a4e7e317 GC |
604 | static int get_scalar_repos(const char *key, const char *value, |
605 | const struct config_context *ctx UNUSED, | |
606 | void *data) | |
45826760 JS |
607 | { |
608 | struct string_list *list = data; | |
609 | ||
610 | if (!strcmp(key, "scalar.repo")) | |
611 | string_list_append(list, value); | |
612 | ||
613 | return 0; | |
614 | } | |
615 | ||
c90db53d JS |
616 | static int remove_deleted_enlistment(struct strbuf *path) |
617 | { | |
618 | int res = 0; | |
619 | strbuf_realpath_forgiving(path, path->buf, 1); | |
620 | ||
621 | if (run_git("config", "--global", | |
622 | "--unset", "--fixed-value", | |
623 | "scalar.repo", path->buf, NULL) < 0) | |
624 | res = -1; | |
625 | ||
626 | if (run_git("config", "--global", | |
627 | "--unset", "--fixed-value", | |
628 | "maintenance.repo", path->buf, NULL) < 0) | |
629 | res = -1; | |
630 | ||
631 | return res; | |
632 | } | |
633 | ||
cb59d55e JS |
634 | static int cmd_reconfigure(int argc, const char **argv) |
635 | { | |
45826760 | 636 | int all = 0; |
cb59d55e | 637 | struct option options[] = { |
45826760 JS |
638 | OPT_BOOL('a', "all", &all, |
639 | N_("reconfigure all registered enlistments")), | |
cb59d55e JS |
640 | OPT_END(), |
641 | }; | |
642 | const char * const usage[] = { | |
45826760 | 643 | N_("scalar reconfigure [--all | <enlistment>]"), |
cb59d55e JS |
644 | NULL |
645 | }; | |
45826760 JS |
646 | struct string_list scalar_repos = STRING_LIST_INIT_DUP; |
647 | int i, res = 0; | |
648 | struct repository r = { NULL }; | |
649 | struct strbuf commondir = STRBUF_INIT, gitdir = STRBUF_INIT; | |
cb59d55e JS |
650 | |
651 | argc = parse_options(argc, argv, NULL, options, | |
652 | usage, 0); | |
653 | ||
45826760 JS |
654 | if (!all) { |
655 | setup_enlistment_directory(argc, argv, usage, options, NULL); | |
656 | ||
657 | return set_recommended_config(1); | |
658 | } | |
659 | ||
660 | if (argc > 0) | |
661 | usage_msg_opt(_("--all or <enlistment>, but not both"), | |
662 | usage, options); | |
663 | ||
664 | git_config(get_scalar_repos, &scalar_repos); | |
cb59d55e | 665 | |
45826760 | 666 | for (i = 0; i < scalar_repos.nr; i++) { |
f9a547d3 | 667 | int succeeded = 0; |
45826760 JS |
668 | const char *dir = scalar_repos.items[i].string; |
669 | ||
670 | strbuf_reset(&commondir); | |
671 | strbuf_reset(&gitdir); | |
672 | ||
673 | if (chdir(dir) < 0) { | |
c90db53d JS |
674 | struct strbuf buf = STRBUF_INIT; |
675 | ||
676 | if (errno != ENOENT) { | |
677 | warning_errno(_("could not switch to '%s'"), dir); | |
f9a547d3 | 678 | goto loop_end; |
c90db53d JS |
679 | } |
680 | ||
681 | strbuf_addstr(&buf, dir); | |
682 | if (remove_deleted_enlistment(&buf)) | |
f9a547d3 DS |
683 | error(_("could not remove stale " |
684 | "scalar.repo '%s'"), dir); | |
685 | else { | |
686 | warning(_("removed stale scalar.repo '%s'"), | |
c90db53d | 687 | dir); |
f9a547d3 DS |
688 | succeeded = 1; |
689 | } | |
c90db53d | 690 | strbuf_release(&buf); |
f9a547d3 DS |
691 | goto loop_end; |
692 | } | |
693 | ||
694 | switch (discover_git_directory_reason(&commondir, &gitdir)) { | |
695 | case GIT_DIR_INVALID_OWNERSHIP: | |
696 | warning(_("repository at '%s' has different owner"), dir); | |
697 | goto loop_end; | |
698 | ||
699 | case GIT_DIR_INVALID_GITFILE: | |
700 | case GIT_DIR_INVALID_FORMAT: | |
701 | warning(_("repository at '%s' has a format issue"), dir); | |
702 | goto loop_end; | |
45826760 | 703 | |
f9a547d3 DS |
704 | case GIT_DIR_DISCOVERED: |
705 | succeeded = 1; | |
706 | break; | |
45826760 | 707 | |
f9a547d3 DS |
708 | default: |
709 | warning(_("repository not found in '%s'"), dir); | |
710 | break; | |
711 | } | |
712 | ||
713 | git_config_clear(); | |
714 | ||
715 | the_repository = &r; | |
716 | r.commondir = commondir.buf; | |
717 | r.gitdir = gitdir.buf; | |
718 | ||
719 | if (set_recommended_config(1) >= 0) | |
720 | succeeded = 1; | |
721 | ||
722 | loop_end: | |
723 | if (!succeeded) { | |
724 | res = -1; | |
725 | warning(_("to unregister this repository from Scalar, run\n" | |
726 | "\tgit config --global --unset --fixed-value scalar.repo \"%s\""), | |
727 | dir); | |
45826760 JS |
728 | } |
729 | } | |
730 | ||
731 | string_list_clear(&scalar_repos, 1); | |
732 | strbuf_release(&commondir); | |
733 | strbuf_release(&gitdir); | |
734 | ||
735 | return res; | |
cb59d55e JS |
736 | } |
737 | ||
7020c88c DS |
738 | static int cmd_run(int argc, const char **argv) |
739 | { | |
740 | struct option options[] = { | |
741 | OPT_END(), | |
742 | }; | |
743 | struct { | |
744 | const char *arg, *task; | |
745 | } tasks[] = { | |
746 | { "config", NULL }, | |
747 | { "commit-graph", "commit-graph" }, | |
748 | { "fetch", "prefetch" }, | |
749 | { "loose-objects", "loose-objects" }, | |
750 | { "pack-files", "incremental-repack" }, | |
751 | { NULL, NULL } | |
752 | }; | |
753 | struct strbuf buf = STRBUF_INIT; | |
754 | const char *usagestr[] = { NULL, NULL }; | |
755 | int i; | |
756 | ||
757 | strbuf_addstr(&buf, N_("scalar run <task> [<enlistment>]\nTasks:\n")); | |
758 | for (i = 0; tasks[i].arg; i++) | |
759 | strbuf_addf(&buf, "\t%s\n", tasks[i].arg); | |
760 | usagestr[0] = buf.buf; | |
761 | ||
762 | argc = parse_options(argc, argv, NULL, options, | |
763 | usagestr, 0); | |
764 | ||
765 | if (!argc) | |
766 | usage_with_options(usagestr, options); | |
767 | ||
768 | if (!strcmp("all", argv[0])) { | |
769 | i = -1; | |
770 | } else { | |
771 | for (i = 0; tasks[i].arg && strcmp(tasks[i].arg, argv[0]); i++) | |
772 | ; /* keep looking for the task */ | |
773 | ||
774 | if (i > 0 && !tasks[i].arg) { | |
775 | error(_("no such task: '%s'"), argv[0]); | |
776 | usage_with_options(usagestr, options); | |
777 | } | |
778 | } | |
779 | ||
780 | argc--; | |
781 | argv++; | |
782 | setup_enlistment_directory(argc, argv, usagestr, options, NULL); | |
783 | strbuf_release(&buf); | |
784 | ||
785 | if (i == 0) | |
786 | return register_dir(); | |
787 | ||
788 | if (i > 0) | |
789 | return run_git("maintenance", "run", | |
790 | "--task", tasks[i].task, NULL); | |
791 | ||
792 | if (register_dir()) | |
793 | return -1; | |
794 | for (i = 1; tasks[i].arg; i++) | |
795 | if (run_git("maintenance", "run", | |
796 | "--task", tasks[i].task, NULL)) | |
797 | return -1; | |
798 | return 0; | |
799 | } | |
800 | ||
c76a53eb DS |
801 | static int cmd_unregister(int argc, const char **argv) |
802 | { | |
803 | struct option options[] = { | |
804 | OPT_END(), | |
805 | }; | |
806 | const char * const usage[] = { | |
807 | N_("scalar unregister [<enlistment>]"), | |
808 | NULL | |
809 | }; | |
810 | ||
811 | argc = parse_options(argc, argv, NULL, options, | |
812 | usage, 0); | |
813 | ||
f5f0842d JS |
814 | /* |
815 | * Be forgiving when the enlistment or worktree does not even exist any | |
816 | * longer; This can be the case if a user deleted the worktree by | |
817 | * mistake and _still_ wants to unregister the thing. | |
818 | */ | |
819 | if (argc == 1) { | |
820 | struct strbuf src_path = STRBUF_INIT, workdir_path = STRBUF_INIT; | |
821 | ||
822 | strbuf_addf(&src_path, "%s/src/.git", argv[0]); | |
823 | strbuf_addf(&workdir_path, "%s/.git", argv[0]); | |
824 | if (!is_directory(src_path.buf) && !is_directory(workdir_path.buf)) { | |
825 | /* remove possible matching registrations */ | |
826 | int res = -1; | |
827 | ||
828 | strbuf_strip_suffix(&src_path, "/.git"); | |
829 | res = remove_deleted_enlistment(&src_path) && res; | |
830 | ||
831 | strbuf_strip_suffix(&workdir_path, "/.git"); | |
832 | res = remove_deleted_enlistment(&workdir_path) && res; | |
833 | ||
834 | strbuf_release(&src_path); | |
835 | strbuf_release(&workdir_path); | |
836 | return res; | |
837 | } | |
838 | strbuf_release(&src_path); | |
839 | strbuf_release(&workdir_path); | |
840 | } | |
841 | ||
c76a53eb DS |
842 | setup_enlistment_directory(argc, argv, usage, options, NULL); |
843 | ||
844 | return unregister_dir(); | |
845 | } | |
846 | ||
d85ada7c MJC |
847 | static int cmd_delete(int argc, const char **argv) |
848 | { | |
849 | char *cwd = xgetcwd(); | |
850 | struct option options[] = { | |
851 | OPT_END(), | |
852 | }; | |
853 | const char * const usage[] = { | |
854 | N_("scalar delete <enlistment>"), | |
855 | NULL | |
856 | }; | |
857 | struct strbuf enlistment = STRBUF_INIT; | |
858 | int res = 0; | |
859 | ||
860 | argc = parse_options(argc, argv, NULL, options, | |
861 | usage, 0); | |
862 | ||
863 | if (argc != 1) | |
864 | usage_with_options(usage, options); | |
865 | ||
866 | setup_enlistment_directory(argc, argv, usage, options, &enlistment); | |
867 | ||
868 | if (dir_inside_of(cwd, enlistment.buf) >= 0) | |
869 | res = error(_("refusing to delete current working directory")); | |
870 | else { | |
871 | close_object_store(the_repository->objects); | |
872 | res = delete_enlistment(&enlistment); | |
873 | } | |
874 | strbuf_release(&enlistment); | |
875 | free(cwd); | |
876 | ||
877 | return res; | |
878 | } | |
879 | ||
951759d3 JS |
880 | static int cmd_help(int argc, const char **argv) |
881 | { | |
882 | struct option options[] = { | |
883 | OPT_END(), | |
884 | }; | |
885 | const char * const usage[] = { | |
886 | "scalar help", | |
887 | NULL | |
888 | }; | |
889 | ||
890 | argc = parse_options(argc, argv, NULL, options, | |
891 | usage, 0); | |
892 | ||
893 | if (argc != 0) | |
894 | usage_with_options(usage, options); | |
895 | ||
896 | return run_git("help", "scalar", NULL); | |
897 | } | |
898 | ||
ddc35d83 JS |
899 | static int cmd_version(int argc, const char **argv) |
900 | { | |
901 | int verbose = 0, build_options = 0; | |
902 | struct option options[] = { | |
903 | OPT__VERBOSE(&verbose, N_("include Git version")), | |
904 | OPT_BOOL(0, "build-options", &build_options, | |
905 | N_("include Git's build options")), | |
906 | OPT_END(), | |
907 | }; | |
908 | const char * const usage[] = { | |
909 | N_("scalar verbose [-v | --verbose] [--build-options]"), | |
910 | NULL | |
911 | }; | |
912 | struct strbuf buf = STRBUF_INIT; | |
913 | ||
914 | argc = parse_options(argc, argv, NULL, options, | |
915 | usage, 0); | |
916 | ||
917 | if (argc != 0) | |
918 | usage_with_options(usage, options); | |
919 | ||
920 | get_version_info(&buf, build_options); | |
921 | fprintf(stderr, "%s\n", buf.buf); | |
922 | strbuf_release(&buf); | |
923 | ||
924 | return 0; | |
925 | } | |
926 | ||
0a43fb22 JS |
927 | static struct { |
928 | const char *name; | |
929 | int (*fn)(int, const char **); | |
930 | } builtins[] = { | |
546f822d | 931 | { "clone", cmd_clone }, |
2b710457 | 932 | { "list", cmd_list }, |
d0feac4e | 933 | { "register", cmd_register }, |
c76a53eb | 934 | { "unregister", cmd_unregister }, |
7020c88c | 935 | { "run", cmd_run }, |
cb59d55e | 936 | { "reconfigure", cmd_reconfigure }, |
d85ada7c | 937 | { "delete", cmd_delete }, |
951759d3 | 938 | { "help", cmd_help }, |
ddc35d83 | 939 | { "version", cmd_version }, |
aa5c79a3 | 940 | { "diagnose", cmd_diagnose }, |
0a43fb22 JS |
941 | { NULL, NULL}, |
942 | }; | |
943 | ||
944 | int cmd_main(int argc, const char **argv) | |
945 | { | |
946 | struct strbuf scalar_usage = STRBUF_INIT; | |
947 | int i; | |
948 | ||
2ae8eb5d JS |
949 | while (argc > 1 && *argv[1] == '-') { |
950 | if (!strcmp(argv[1], "-C")) { | |
951 | if (argc < 3) | |
952 | die(_("-C requires a <directory>")); | |
953 | if (chdir(argv[2]) < 0) | |
954 | die_errno(_("could not change to '%s'"), | |
955 | argv[2]); | |
956 | argc -= 2; | |
957 | argv += 2; | |
958 | } else if (!strcmp(argv[1], "-c")) { | |
959 | if (argc < 3) | |
960 | die(_("-c requires a <key>=<value> argument")); | |
961 | git_config_push_parameter(argv[2]); | |
962 | argc -= 2; | |
963 | argv += 2; | |
964 | } else | |
965 | break; | |
966 | } | |
967 | ||
0a43fb22 JS |
968 | if (argc > 1) { |
969 | argv++; | |
970 | argc--; | |
971 | ||
972 | for (i = 0; builtins[i].name; i++) | |
973 | if (!strcmp(builtins[i].name, argv[0])) | |
974 | return !!builtins[i].fn(argc, argv); | |
975 | } | |
976 | ||
977 | strbuf_addstr(&scalar_usage, | |
2ae8eb5d JS |
978 | N_("scalar [-C <directory>] [-c <key>=<value>] " |
979 | "<command> [<options>]\n\nCommands:\n")); | |
0a43fb22 JS |
980 | for (i = 0; builtins[i].name; i++) |
981 | strbuf_addf(&scalar_usage, "\t%s\n", builtins[i].name); | |
982 | ||
983 | usage(scalar_usage.buf); | |
984 | } |