Merge branch 'jk/tighten-alloc'
[git/gitweb.git] / builtin / submodule--helper.c
blobf4c3eff179b5f25e5d09c26348c107d0582455f2
1 #include "builtin.h"
2 #include "cache.h"
3 #include "parse-options.h"
4 #include "quote.h"
5 #include "pathspec.h"
6 #include "dir.h"
7 #include "utf8.h"
8 #include "submodule.h"
9 #include "submodule-config.h"
10 #include "string-list.h"
11 #include "run-command.h"
13 struct module_list {
14 const struct cache_entry **entries;
15 int alloc, nr;
17 #define MODULE_LIST_INIT { NULL, 0, 0 }
19 static int module_list_compute(int argc, const char **argv,
20 const char *prefix,
21 struct pathspec *pathspec,
22 struct module_list *list)
24 int i, result = 0;
25 char *max_prefix, *ps_matched = NULL;
26 int max_prefix_len;
27 parse_pathspec(pathspec, 0,
28 PATHSPEC_PREFER_FULL |
29 PATHSPEC_STRIP_SUBMODULE_SLASH_CHEAP,
30 prefix, argv);
32 /* Find common prefix for all pathspec's */
33 max_prefix = common_prefix(pathspec);
34 max_prefix_len = max_prefix ? strlen(max_prefix) : 0;
36 if (pathspec->nr)
37 ps_matched = xcalloc(pathspec->nr, 1);
39 if (read_cache() < 0)
40 die(_("index file corrupt"));
42 for (i = 0; i < active_nr; i++) {
43 const struct cache_entry *ce = active_cache[i];
45 if (!S_ISGITLINK(ce->ce_mode) ||
46 !match_pathspec(pathspec, ce->name, ce_namelen(ce),
47 max_prefix_len, ps_matched, 1))
48 continue;
50 ALLOC_GROW(list->entries, list->nr + 1, list->alloc);
51 list->entries[list->nr++] = ce;
52 while (i + 1 < active_nr &&
53 !strcmp(ce->name, active_cache[i + 1]->name))
55 * Skip entries with the same name in different stages
56 * to make sure an entry is returned only once.
58 i++;
60 free(max_prefix);
62 if (ps_matched && report_path_error(ps_matched, pathspec, prefix))
63 result = -1;
65 free(ps_matched);
67 return result;
70 static int module_list(int argc, const char **argv, const char *prefix)
72 int i;
73 struct pathspec pathspec;
74 struct module_list list = MODULE_LIST_INIT;
76 struct option module_list_options[] = {
77 OPT_STRING(0, "prefix", &prefix,
78 N_("path"),
79 N_("alternative anchor for relative paths")),
80 OPT_END()
83 const char *const git_submodule_helper_usage[] = {
84 N_("git submodule--helper list [--prefix=<path>] [<path>...]"),
85 NULL
88 argc = parse_options(argc, argv, prefix, module_list_options,
89 git_submodule_helper_usage, 0);
91 if (module_list_compute(argc, argv, prefix, &pathspec, &list) < 0) {
92 printf("#unmatched\n");
93 return 1;
96 for (i = 0; i < list.nr; i++) {
97 const struct cache_entry *ce = list.entries[i];
99 if (ce_stage(ce))
100 printf("%06o %s U\t", ce->ce_mode, sha1_to_hex(null_sha1));
101 else
102 printf("%06o %s %d\t", ce->ce_mode, sha1_to_hex(ce->sha1), ce_stage(ce));
104 utf8_fprintf(stdout, "%s\n", ce->name);
106 return 0;
109 static int module_name(int argc, const char **argv, const char *prefix)
111 const struct submodule *sub;
113 if (argc != 2)
114 usage(_("git submodule--helper name <path>"));
116 gitmodules_config();
117 sub = submodule_from_path(null_sha1, argv[1]);
119 if (!sub)
120 die(_("no submodule mapping found in .gitmodules for path '%s'"),
121 argv[1]);
123 printf("%s\n", sub->name);
125 return 0;
127 static int clone_submodule(const char *path, const char *gitdir, const char *url,
128 const char *depth, const char *reference, int quiet)
130 struct child_process cp;
131 child_process_init(&cp);
133 argv_array_push(&cp.args, "clone");
134 argv_array_push(&cp.args, "--no-checkout");
135 if (quiet)
136 argv_array_push(&cp.args, "--quiet");
137 if (depth && *depth)
138 argv_array_pushl(&cp.args, "--depth", depth, NULL);
139 if (reference && *reference)
140 argv_array_pushl(&cp.args, "--reference", reference, NULL);
141 if (gitdir && *gitdir)
142 argv_array_pushl(&cp.args, "--separate-git-dir", gitdir, NULL);
144 argv_array_push(&cp.args, url);
145 argv_array_push(&cp.args, path);
147 cp.git_cmd = 1;
148 cp.env = local_repo_env;
149 cp.no_stdin = 1;
151 return run_command(&cp);
154 static int module_clone(int argc, const char **argv, const char *prefix)
156 const char *path = NULL, *name = NULL, *url = NULL;
157 const char *reference = NULL, *depth = NULL;
158 int quiet = 0;
159 FILE *submodule_dot_git;
160 char *sm_gitdir, *cwd, *p;
161 struct strbuf rel_path = STRBUF_INIT;
162 struct strbuf sb = STRBUF_INIT;
164 struct option module_clone_options[] = {
165 OPT_STRING(0, "prefix", &prefix,
166 N_("path"),
167 N_("alternative anchor for relative paths")),
168 OPT_STRING(0, "path", &path,
169 N_("path"),
170 N_("where the new submodule will be cloned to")),
171 OPT_STRING(0, "name", &name,
172 N_("string"),
173 N_("name of the new submodule")),
174 OPT_STRING(0, "url", &url,
175 N_("string"),
176 N_("url where to clone the submodule from")),
177 OPT_STRING(0, "reference", &reference,
178 N_("string"),
179 N_("reference repository")),
180 OPT_STRING(0, "depth", &depth,
181 N_("string"),
182 N_("depth for shallow clones")),
183 OPT__QUIET(&quiet, "Suppress output for cloning a submodule"),
184 OPT_END()
187 const char *const git_submodule_helper_usage[] = {
188 N_("git submodule--helper clone [--prefix=<path>] [--quiet] "
189 "[--reference <repository>] [--name <name>] [--url <url>]"
190 "[--depth <depth>] [--] [<path>...]"),
191 NULL
194 argc = parse_options(argc, argv, prefix, module_clone_options,
195 git_submodule_helper_usage, 0);
197 strbuf_addf(&sb, "%s/modules/%s", get_git_dir(), name);
198 sm_gitdir = strbuf_detach(&sb, NULL);
200 if (!file_exists(sm_gitdir)) {
201 if (safe_create_leading_directories_const(sm_gitdir) < 0)
202 die(_("could not create directory '%s'"), sm_gitdir);
203 if (clone_submodule(path, sm_gitdir, url, depth, reference, quiet))
204 die(_("clone of '%s' into submodule path '%s' failed"),
205 url, path);
206 } else {
207 if (safe_create_leading_directories_const(path) < 0)
208 die(_("could not create directory '%s'"), path);
209 strbuf_addf(&sb, "%s/index", sm_gitdir);
210 unlink_or_warn(sb.buf);
211 strbuf_reset(&sb);
214 /* Write a .git file in the submodule to redirect to the superproject. */
215 if (safe_create_leading_directories_const(path) < 0)
216 die(_("could not create directory '%s'"), path);
218 if (path && *path)
219 strbuf_addf(&sb, "%s/.git", path);
220 else
221 strbuf_addstr(&sb, ".git");
223 if (safe_create_leading_directories_const(sb.buf) < 0)
224 die(_("could not create leading directories of '%s'"), sb.buf);
225 submodule_dot_git = fopen(sb.buf, "w");
226 if (!submodule_dot_git)
227 die_errno(_("cannot open file '%s'"), sb.buf);
229 fprintf(submodule_dot_git, "gitdir: %s\n",
230 relative_path(sm_gitdir, path, &rel_path));
231 if (fclose(submodule_dot_git))
232 die(_("could not close file %s"), sb.buf);
233 strbuf_reset(&sb);
234 strbuf_reset(&rel_path);
236 cwd = xgetcwd();
237 /* Redirect the worktree of the submodule in the superproject's config */
238 if (!is_absolute_path(sm_gitdir)) {
239 strbuf_addf(&sb, "%s/%s", cwd, sm_gitdir);
240 free(sm_gitdir);
241 sm_gitdir = strbuf_detach(&sb, NULL);
244 strbuf_addf(&sb, "%s/%s", cwd, path);
245 p = git_pathdup_submodule(path, "config");
246 if (!p)
247 die(_("could not get submodule directory for '%s'"), path);
248 git_config_set_in_file(p, "core.worktree",
249 relative_path(sb.buf, sm_gitdir, &rel_path));
250 strbuf_release(&sb);
251 strbuf_release(&rel_path);
252 free(sm_gitdir);
253 free(cwd);
254 free(p);
255 return 0;
258 struct cmd_struct {
259 const char *cmd;
260 int (*fn)(int, const char **, const char *);
263 static struct cmd_struct commands[] = {
264 {"list", module_list},
265 {"name", module_name},
266 {"clone", module_clone},
269 int cmd_submodule__helper(int argc, const char **argv, const char *prefix)
271 int i;
272 if (argc < 2)
273 die(_("fatal: submodule--helper subcommand must be "
274 "called with a subcommand"));
276 for (i = 0; i < ARRAY_SIZE(commands); i++)
277 if (!strcmp(argv[1], commands[i].cmd))
278 return commands[i].fn(argc - 1, argv + 1, prefix);
280 die(_("fatal: '%s' is not a valid submodule--helper "
281 "subcommand"), argv[1]);