clone.c 24.5 KB
Newer Older
D
Daniel Barkalow 已提交
1 2 3 4 5 6 7 8 9 10
/*
 * Builtin "git clone"
 *
 * Copyright (c) 2007 Kristian Høgsberg <krh@redhat.com>,
 *		 2008 Daniel Barkalow <barkalow@iabervon.org>
 * Based on git-commit.sh by Junio C Hamano and Linus Torvalds
 *
 * Clone a repository into a different directory that does not yet exist.
 */

S
Stephen Boyd 已提交
11
#include "builtin.h"
D
Daniel Barkalow 已提交
12 13 14 15 16 17 18 19 20
#include "parse-options.h"
#include "fetch-pack.h"
#include "refs.h"
#include "tree.h"
#include "tree-walk.h"
#include "unpack-trees.h"
#include "transport.h"
#include "strbuf.h"
#include "dir.h"
J
Johan Herland 已提交
21
#include "pack-refs.h"
22
#include "sigchain.h"
23
#include "branch.h"
J
Jay Soffian 已提交
24
#include "remote.h"
25
#include "run-command.h"
D
Daniel Barkalow 已提交
26 27 28 29 30 31 32 33 34 35

/*
 * Overall FIXMEs:
 *  - respect DB_ENVIRONMENT for .git/objects.
 *
 * Implementation notes:
 *  - dropping use-separate-remote and no-separate-remote compatibility
 *
 */
static const char * const builtin_clone_usage[] = {
36
	N_("git clone [options] [--] <repo> [<dir>]"),
D
Daniel Barkalow 已提交
37 38 39
	NULL
};

40
static int option_no_checkout, option_bare, option_mirror, option_single_branch = -1;
41
static int option_local = -1, option_no_hardlinks, option_shared, option_recursive;
42
static char *option_template, *option_depth;
D
Daniel Barkalow 已提交
43
static char *option_origin = NULL;
44
static char *option_branch = NULL;
45
static const char *real_git_dir;
D
Daniel Barkalow 已提交
46
static char *option_upload_pack = "git-upload-pack";
47
static int option_verbosity;
48
static int option_progress = -1;
49
static struct string_list option_config;
50 51 52 53 54 55 56 57 58 59
static struct string_list option_reference;

static int opt_parse_reference(const struct option *opt, const char *arg, int unset)
{
	struct string_list *option_reference = opt->value;
	if (!arg)
		return -1;
	string_list_append(option_reference, arg);
	return 0;
}
D
Daniel Barkalow 已提交
60 61

static struct option builtin_clone_options[] = {
62
	OPT__VERBOSITY(&option_verbosity),
63
	OPT_BOOL(0, "progress", &option_progress,
64
		 N_("force progress reporting")),
D
Daniel Barkalow 已提交
65
	OPT_BOOLEAN('n', "no-checkout", &option_no_checkout,
66 67
		    N_("don't create a checkout")),
	OPT_BOOLEAN(0, "bare", &option_bare, N_("create a bare repository")),
J
Junio C Hamano 已提交
68
	{ OPTION_BOOLEAN, 0, "naked", &option_bare, NULL,
69
		N_("create a bare repository"),
J
Junio C Hamano 已提交
70
		PARSE_OPT_NOARG | PARSE_OPT_HIDDEN },
71
	OPT_BOOLEAN(0, "mirror", &option_mirror,
72
		    N_("create a mirror repository (implies bare)")),
73
	OPT_BOOL('l', "local", &option_local,
74
		N_("to clone from a local repository")),
D
Daniel Barkalow 已提交
75
	OPT_BOOLEAN(0, "no-hardlinks", &option_no_hardlinks,
76
		    N_("don't use local hardlinks, always copy")),
D
Daniel Barkalow 已提交
77
	OPT_BOOLEAN('s', "shared", &option_shared,
78
		    N_("setup as shared repository")),
79
	OPT_BOOLEAN(0, "recursive", &option_recursive,
80
		    N_("initialize submodules in the clone")),
81
	OPT_BOOLEAN(0, "recurse-submodules", &option_recursive,
82 83 84 85 86 87 88 89 90 91 92 93 94
		    N_("initialize submodules in the clone")),
	OPT_STRING(0, "template", &option_template, N_("template-directory"),
		   N_("directory from which templates will be used")),
	OPT_CALLBACK(0 , "reference", &option_reference, N_("repo"),
		     N_("reference repository"), &opt_parse_reference),
	OPT_STRING('o', "origin", &option_origin, N_("name"),
		   N_("use <name> instead of 'origin' to track upstream")),
	OPT_STRING('b', "branch", &option_branch, N_("branch"),
		   N_("checkout <branch> instead of the remote's HEAD")),
	OPT_STRING('u', "upload-pack", &option_upload_pack, N_("path"),
		   N_("path to git-upload-pack on the remote")),
	OPT_STRING(0, "depth", &option_depth, N_("depth"),
		    N_("create a shallow clone of that depth")),
95
	OPT_BOOL(0, "single-branch", &option_single_branch,
96 97 98 99 100
		    N_("clone only one branch, HEAD or --branch")),
	OPT_STRING(0, "separate-git-dir", &real_git_dir, N_("gitdir"),
		   N_("separate git dir from working tree")),
	OPT_STRING_LIST('c', "config", &option_config, N_("key=value"),
			N_("set config inside the new repository")),
D
Daniel Barkalow 已提交
101 102 103
	OPT_END()
};

104 105 106 107
static const char *argv_submodule[] = {
	"submodule", "update", "--init", "--recursive", NULL
};

D
Daniel Barkalow 已提交
108 109
static char *get_repo_path(const char *repo, int *is_bundle)
{
110
	static char *suffix[] = { "/.git", "", ".git/.git", ".git" };
D
Daniel Barkalow 已提交
111 112 113 114 115 116 117
	static char *bundle_suffix[] = { ".bundle", "" };
	struct stat st;
	int i;

	for (i = 0; i < ARRAY_SIZE(suffix); i++) {
		const char *path;
		path = mkpath("%s%s", repo, suffix[i]);
118 119
		if (stat(path, &st))
			continue;
120
		if (S_ISDIR(st.st_mode) && is_git_directory(path)) {
D
Daniel Barkalow 已提交
121
			*is_bundle = 0;
122
			return xstrdup(absolute_path(path));
123 124 125 126 127 128 129 130 131 132 133 134 135 136 137
		} else if (S_ISREG(st.st_mode) && st.st_size > 8) {
			/* Is it a "gitfile"? */
			char signature[8];
			int len, fd = open(path, O_RDONLY);
			if (fd < 0)
				continue;
			len = read_in_full(fd, signature, 8);
			close(fd);
			if (len != 8 || strncmp(signature, "gitdir: ", 8))
				continue;
			path = read_gitfile(path);
			if (path) {
				*is_bundle = 0;
				return xstrdup(absolute_path(path));
			}
D
Daniel Barkalow 已提交
138 139 140 141 142 143 144 145
		}
	}

	for (i = 0; i < ARRAY_SIZE(bundle_suffix); i++) {
		const char *path;
		path = mkpath("%s%s", repo, bundle_suffix[i]);
		if (!stat(path, &st) && S_ISREG(st.st_mode)) {
			*is_bundle = 1;
146
			return xstrdup(absolute_path(path));
D
Daniel Barkalow 已提交
147 148 149 150 151 152
		}
	}

	return NULL;
}

153
static char *guess_dir_name(const char *repo, int is_bundle, int is_bare)
D
Daniel Barkalow 已提交
154
{
155
	const char *end = repo + strlen(repo), *start;
156
	char *dir;
157 158

	/*
159
	 * Strip trailing spaces, slashes and /.git
160
	 */
161
	while (repo < end && (is_dir_sep(end[-1]) || isspace(end[-1])))
162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187
		end--;
	if (end - repo > 5 && is_dir_sep(end[-5]) &&
	    !strncmp(end - 4, ".git", 4)) {
		end -= 5;
		while (repo < end && is_dir_sep(end[-1]))
			end--;
	}

	/*
	 * Find last component, but be prepared that repo could have
	 * the form  "remote.example.com:foo.git", i.e. no slash
	 * in the directory part.
	 */
	start = end;
	while (repo < start && !is_dir_sep(start[-1]) && start[-1] != ':')
		start--;

	/*
	 * Strip .{bundle,git}.
	 */
	if (is_bundle) {
		if (end - start > 7 && !strncmp(end - 7, ".bundle", 7))
			end -= 7;
	} else {
		if (end - start > 4 && !strncmp(end - 4, ".git", 4))
			end -= 4;
D
Daniel Barkalow 已提交
188 189
	}

190
	if (is_bare) {
191 192
		struct strbuf result = STRBUF_INIT;
		strbuf_addf(&result, "%.*s.git", (int)(end - start), start);
193
		dir = strbuf_detach(&result, NULL);
194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217
	} else
		dir = xstrndup(start, end - start);
	/*
	 * Replace sequences of 'control' characters and whitespace
	 * with one ascii space, remove leading and trailing spaces.
	 */
	if (*dir) {
		char *out = dir;
		int prev_space = 1 /* strip leading whitespace */;
		for (end = dir; *end; ++end) {
			char ch = *end;
			if ((unsigned char)ch < '\x20')
				ch = '\x20';
			if (isspace(ch)) {
				if (prev_space)
					continue;
				prev_space = 1;
			} else
				prev_space = 0;
			*out++ = ch;
		}
		*out = '\0';
		if (out > dir && prev_space)
			out[-1] = '\0';
218
	}
219
	return dir;
D
Daniel Barkalow 已提交
220 221
}

222 223 224 225 226 227 228 229 230
static void strip_trailing_slashes(char *dir)
{
	char *end = dir + strlen(dir);

	while (dir < end - 1 && is_dir_sep(end[-1]))
		end--;
	*end = '\0';
}

231
static int add_one_reference(struct string_list_item *item, void *cb_data)
D
Daniel Barkalow 已提交
232
{
233 234
	char *ref_git;
	struct strbuf alternate = STRBUF_INIT;
D
Daniel Barkalow 已提交
235

236 237 238
	/* Beware: real_path() and mkpath() return static buffer */
	ref_git = xstrdup(real_path(item->string));
	if (is_directory(mkpath("%s/.git/objects", ref_git))) {
239
		char *ref_git_git = mkpathdup("%s/.git", ref_git);
240 241 242
		free(ref_git);
		ref_git = ref_git_git;
	} else if (!is_directory(mkpath("%s/objects", ref_git)))
243
		die(_("reference repository '%s' is not a local directory."),
244
		    item->string);
D
Daniel Barkalow 已提交
245

246 247 248 249
	strbuf_addf(&alternate, "%s/objects", ref_git);
	add_to_alternates_file(alternate.buf);
	strbuf_release(&alternate);
	free(ref_git);
250 251
	return 0;
}
D
Daniel Barkalow 已提交
252

253 254 255
static void setup_reference(void)
{
	for_each_string_list(&option_reference, add_one_reference, NULL);
D
Daniel Barkalow 已提交
256 257
}

258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293
static void copy_alternates(struct strbuf *src, struct strbuf *dst,
			    const char *src_repo)
{
	/*
	 * Read from the source objects/info/alternates file
	 * and copy the entries to corresponding file in the
	 * destination repository with add_to_alternates_file().
	 * Both src and dst have "$path/objects/info/alternates".
	 *
	 * Instead of copying bit-for-bit from the original,
	 * we need to append to existing one so that the already
	 * created entry via "clone -s" is not lost, and also
	 * to turn entries with paths relative to the original
	 * absolute, so that they can be used in the new repository.
	 */
	FILE *in = fopen(src->buf, "r");
	struct strbuf line = STRBUF_INIT;

	while (strbuf_getline(&line, in, '\n') != EOF) {
		char *abs_path, abs_buf[PATH_MAX];
		if (!line.len || line.buf[0] == '#')
			continue;
		if (is_absolute_path(line.buf)) {
			add_to_alternates_file(line.buf);
			continue;
		}
		abs_path = mkpath("%s/objects/%s", src_repo, line.buf);
		normalize_path_copy(abs_buf, abs_path);
		add_to_alternates_file(abs_buf);
	}
	strbuf_release(&line);
	fclose(in);
}

static void copy_or_link_directory(struct strbuf *src, struct strbuf *dest,
				   const char *src_repo, int src_baselen)
D
Daniel Barkalow 已提交
294 295 296 297 298 299
{
	struct dirent *de;
	struct stat buf;
	int src_len, dest_len;
	DIR *dir;

300
	dir = opendir(src->buf);
D
Daniel Barkalow 已提交
301
	if (!dir)
302
		die_errno(_("failed to open '%s'"), src->buf);
D
Daniel Barkalow 已提交
303

304
	if (mkdir(dest->buf, 0777)) {
D
Daniel Barkalow 已提交
305
		if (errno != EEXIST)
306
			die_errno(_("failed to create directory '%s'"), dest->buf);
307
		else if (stat(dest->buf, &buf))
308
			die_errno(_("failed to stat '%s'"), dest->buf);
D
Daniel Barkalow 已提交
309
		else if (!S_ISDIR(buf.st_mode))
310
			die(_("%s exists and is not a directory"), dest->buf);
D
Daniel Barkalow 已提交
311 312
	}

313 314 315 316
	strbuf_addch(src, '/');
	src_len = src->len;
	strbuf_addch(dest, '/');
	dest_len = dest->len;
D
Daniel Barkalow 已提交
317 318

	while ((de = readdir(dir)) != NULL) {
319 320 321 322 323
		strbuf_setlen(src, src_len);
		strbuf_addstr(src, de->d_name);
		strbuf_setlen(dest, dest_len);
		strbuf_addstr(dest, de->d_name);
		if (stat(src->buf, &buf)) {
324
			warning (_("failed to stat %s\n"), src->buf);
D
Daniel Barkalow 已提交
325 326 327 328
			continue;
		}
		if (S_ISDIR(buf.st_mode)) {
			if (de->d_name[0] != '.')
329 330 331 332 333 334 335 336
				copy_or_link_directory(src, dest,
						       src_repo, src_baselen);
			continue;
		}

		/* Files that cannot be copied bit-for-bit... */
		if (!strcmp(src->buf + src_baselen, "/info/alternates")) {
			copy_alternates(src, dest, src_repo);
D
Daniel Barkalow 已提交
337 338 339
			continue;
		}

340
		if (unlink(dest->buf) && errno != ENOENT)
341
			die_errno(_("failed to unlink '%s'"), dest->buf);
342
		if (!option_no_hardlinks) {
343
			if (!link(src->buf, dest->buf))
344
				continue;
345
			if (option_local > 0)
346
				die_errno(_("failed to create link '%s'"), dest->buf);
347
			option_no_hardlinks = 1;
D
Daniel Barkalow 已提交
348
		}
349
		if (copy_file_with_time(dest->buf, src->buf, 0666))
350
			die_errno(_("failed to copy file to '%s'"), dest->buf);
D
Daniel Barkalow 已提交
351
	}
352
	closedir(dir);
D
Daniel Barkalow 已提交
353 354
}

355
static void clone_local(const char *src_repo, const char *dest_repo)
D
Daniel Barkalow 已提交
356
{
357 358 359 360 361 362 363 364
	if (option_shared) {
		struct strbuf alt = STRBUF_INIT;
		strbuf_addf(&alt, "%s/objects", src_repo);
		add_to_alternates_file(alt.buf);
		strbuf_release(&alt);
	} else {
		struct strbuf src = STRBUF_INIT;
		struct strbuf dest = STRBUF_INIT;
365 366
		strbuf_addf(&src, "%s/objects", src_repo);
		strbuf_addf(&dest, "%s/objects", dest_repo);
367
		copy_or_link_directory(&src, &dest, src_repo, src.len);
368 369
		strbuf_release(&src);
		strbuf_release(&dest);
D
Daniel Barkalow 已提交
370 371
	}

372
	if (0 <= option_verbosity)
373
		printf(_("done.\n"));
D
Daniel Barkalow 已提交
374 375 376 377
}

static const char *junk_work_tree;
static const char *junk_git_dir;
378
static pid_t junk_pid;
D
Daniel Barkalow 已提交
379 380 381

static void remove_junk(void)
{
382
	struct strbuf sb = STRBUF_INIT;
D
Daniel Barkalow 已提交
383 384 385 386 387 388 389 390 391 392 393 394 395 396 397 398 399
	if (getpid() != junk_pid)
		return;
	if (junk_git_dir) {
		strbuf_addstr(&sb, junk_git_dir);
		remove_dir_recursively(&sb, 0);
		strbuf_reset(&sb);
	}
	if (junk_work_tree) {
		strbuf_addstr(&sb, junk_work_tree);
		remove_dir_recursively(&sb, 0);
		strbuf_reset(&sb);
	}
}

static void remove_junk_on_signal(int signo)
{
	remove_junk();
400
	sigchain_pop(signo);
D
Daniel Barkalow 已提交
401 402 403
	raise(signo);
}

404 405 406 407 408 409 410 411
static struct ref *find_remote_branch(const struct ref *refs, const char *branch)
{
	struct ref *ref;
	struct strbuf head = STRBUF_INIT;
	strbuf_addstr(&head, "refs/heads/");
	strbuf_addstr(&head, branch);
	ref = find_ref_by_name(refs, head.buf);
	strbuf_release(&head);
412 413 414 415 416 417 418 419 420

	if (ref)
		return ref;

	strbuf_addstr(&head, "refs/tags/");
	strbuf_addstr(&head, branch);
	ref = find_ref_by_name(refs, head.buf);
	strbuf_release(&head);

421 422 423
	return ref;
}

424 425
static struct ref *wanted_peer_refs(const struct ref *refs,
		struct refspec *refspec)
D
Daniel Barkalow 已提交
426
{
J
Jeff King 已提交
427 428 429
	struct ref *head = copy_ref(find_ref_by_name(refs, "HEAD"));
	struct ref *local_refs = head;
	struct ref **tail = head ? &head->next : &local_refs;
D
Daniel Barkalow 已提交
430

431 432 433 434 435
	if (option_single_branch) {
		struct ref *remote_head = NULL;

		if (!option_branch)
			remote_head = guess_remote_head(head, refs, 0);
436 437 438 439 440
		else {
			local_refs = NULL;
			tail = &local_refs;
			remote_head = copy_ref(find_remote_branch(refs, option_branch));
		}
441 442 443 444

		if (!remote_head && option_branch)
			warning(_("Could not find remote branch %s to clone."),
				option_branch);
445
		else {
446
			get_fetch_map(remote_head, refspec, &tail, 0);
447 448 449 450

			/* if --branch=tag, pull the requested tag explicitly */
			get_fetch_map(remote_head, tag_refspec, &tail, 0);
		}
451 452 453 454
	} else
		get_fetch_map(refs, refspec, &tail, 0);

	if (!option_mirror && !option_single_branch)
455
		get_fetch_map(refs, tag_refspec, &tail, 0);
D
Daniel Barkalow 已提交
456

457 458 459 460 461 462 463
	return local_refs;
}

static void write_remote_refs(const struct ref *local_refs)
{
	const struct ref *r;

J
Jeff King 已提交
464 465 466
	for (r = local_refs; r; r = r->next) {
		if (!r->peer_ref)
			continue;
467
		add_packed_ref(r->peer_ref->name, r->old_sha1);
J
Jeff King 已提交
468
	}
J
Johan Herland 已提交
469 470

	pack_refs(PACK_REFS_ALL);
D
Daniel Barkalow 已提交
471 472
}

473 474 475 476 477 478 479 480 481 482 483 484 485 486 487
static void write_followtags(const struct ref *refs, const char *msg)
{
	const struct ref *ref;
	for (ref = refs; ref; ref = ref->next) {
		if (prefixcmp(ref->name, "refs/tags/"))
			continue;
		if (!suffixcmp(ref->name, "^{}"))
			continue;
		if (!has_sha1_file(ref->old_sha1))
			continue;
		update_ref(msg, ref->name, ref->old_sha1,
			   NULL, 0, DIE_ON_ERR);
	}
}

488 489 490 491 492 493 494 495 496 497 498 499 500 501 502 503 504 505 506 507 508 509
static void update_remote_refs(const struct ref *refs,
			       const struct ref *mapped_refs,
			       const struct ref *remote_head_points_at,
			       const char *branch_top,
			       const char *msg)
{
	if (refs) {
		write_remote_refs(mapped_refs);
		if (option_single_branch)
			write_followtags(refs, msg);
	}

	if (remote_head_points_at && !option_bare) {
		struct strbuf head_ref = STRBUF_INIT;
		strbuf_addstr(&head_ref, branch_top);
		strbuf_addstr(&head_ref, "HEAD");
		create_symref(head_ref.buf,
			      remote_head_points_at->peer_ref->name,
			      msg);
	}
}

510 511 512
static void update_head(const struct ref *our, const struct ref *remote,
			const char *msg)
{
513
	if (our && !prefixcmp(our->name, "refs/heads/")) {
514 515 516 517 518 519 520
		/* Local default branch link */
		create_symref("HEAD", our->name, NULL);
		if (!option_bare) {
			const char *head = skip_prefix(our->name, "refs/heads/");
			update_ref(msg, "HEAD", our->old_sha1, NULL, 0, DIE_ON_ERR);
			install_branch_config(0, head, option_origin, our->name);
		}
521 522 523 524 525
	} else if (our) {
		struct commit *c = lookup_commit_reference(our->old_sha1);
		/* --branch specifies a non-branch (i.e. tags), detach HEAD */
		update_ref(msg, "HEAD", c->object.sha1,
			   NULL, REF_NODEREF, DIE_ON_ERR);
526 527 528
	} else if (remote) {
		/*
		 * We know remote HEAD points to a non-branch, or
529
		 * HEAD points to a branch but we don't know which one.
530 531 532 533 534 535 536
		 * Detach HEAD in all these cases.
		 */
		update_ref(msg, "HEAD", remote->old_sha1,
			   NULL, REF_NODEREF, DIE_ON_ERR);
	}
}

537 538 539 540 541 542 543 544 545 546 547 548 549 550 551 552 553 554 555
static int checkout(void)
{
	unsigned char sha1[20];
	char *head;
	struct lock_file *lock_file;
	struct unpack_trees_options opts;
	struct tree *tree;
	struct tree_desc t;
	int err = 0, fd;

	if (option_no_checkout)
		return 0;

	head = resolve_refdup("HEAD", sha1, 1, NULL);
	if (!head) {
		warning(_("remote HEAD refers to nonexistent ref, "
			  "unable to checkout.\n"));
		return 0;
	}
556 557 558 559
	if (!strcmp(head, "HEAD")) {
		if (advice_detached_head)
			detach_advice(sha1_to_hex(sha1));
	} else {
560 561 562 563 564 565 566 567 568 569 570 571 572 573 574
		if (prefixcmp(head, "refs/heads/"))
			die(_("HEAD not found below refs/heads!"));
	}
	free(head);

	/* We need to be in the new work tree for the checkout */
	setup_work_tree();

	lock_file = xcalloc(1, sizeof(struct lock_file));
	fd = hold_locked_index(lock_file, 1);

	memset(&opts, 0, sizeof opts);
	opts.update = 1;
	opts.merge = 1;
	opts.fn = oneway_merge;
575
	opts.verbose_update = (option_verbosity >= 0);
576 577 578 579 580 581
	opts.src_index = &the_index;
	opts.dst_index = &the_index;

	tree = parse_tree_indirect(sha1);
	parse_tree(tree);
	init_tree_desc(&t, tree->buffer, tree->size);
582 583
	if (unpack_trees(1, &t, &opts) < 0)
		die(_("unable to checkout working tree"));
584 585 586 587 588 589 590 591 592 593 594 595 596 597

	if (write_cache(fd, active_cache, active_nr) ||
	    commit_locked_index(lock_file))
		die(_("unable to write new index file"));

	err |= run_hook(NULL, "post-checkout", sha1_to_hex(null_sha1),
			sha1_to_hex(sha1), "1", NULL);

	if (!err && option_recursive)
		err = run_command_v_opt(argv_submodule, RUN_GIT_CMD);

	return err;
}

598 599 600 601 602 603 604 605 606 607 608 609 610 611 612 613
static int write_one_config(const char *key, const char *value, void *data)
{
	return git_config_set_multivar(key, value ? value : "true", "^$", 0);
}

static void write_config(struct string_list *config)
{
	int i;

	for (i = 0; i < config->nr; i++) {
		if (git_config_parse_parameter(config->items[i].string,
					       write_one_config, NULL) < 0)
			die("unable to write parameters to config file");
	}
}

614 615 616 617 618 619 620 621 622 623 624 625 626 627 628 629 630 631 632 633 634 635 636 637 638 639 640 641 642 643 644 645 646 647 648 649 650 651 652 653 654 655 656 657 658 659 660 661
static void write_refspec_config(const char* src_ref_prefix,
		const struct ref* our_head_points_at,
		const struct ref* remote_head_points_at, struct strbuf* branch_top)
{
	struct strbuf key = STRBUF_INIT;
	struct strbuf value = STRBUF_INIT;

	if (option_mirror || !option_bare) {
		if (option_single_branch && !option_mirror) {
			if (option_branch) {
				if (strstr(our_head_points_at->name, "refs/tags/"))
					strbuf_addf(&value, "+%s:%s", our_head_points_at->name,
						our_head_points_at->name);
				else
					strbuf_addf(&value, "+%s:%s%s", our_head_points_at->name,
						branch_top->buf, option_branch);
			} else if (remote_head_points_at) {
				strbuf_addf(&value, "+%s:%s%s", remote_head_points_at->name,
						branch_top->buf,
						skip_prefix(remote_head_points_at->name, "refs/heads/"));
			}
			/*
			 * otherwise, the next "git fetch" will
			 * simply fetch from HEAD without updating
			 * any remote tracking branch, which is what
			 * we want.
			 */
		} else {
			strbuf_addf(&value, "+%s*:%s*", src_ref_prefix, branch_top->buf);
		}
		/* Configure the remote */
		if (value.len) {
			strbuf_addf(&key, "remote.%s.fetch", option_origin);
			git_config_set_multivar(key.buf, value.buf, "^$", 0);
			strbuf_reset(&key);

			if (option_mirror) {
				strbuf_addf(&key, "remote.%s.mirror", option_origin);
				git_config_set(key.buf, "true");
				strbuf_reset(&key);
			}
		}
	}

	strbuf_release(&key);
	strbuf_release(&value);
}

D
Daniel Barkalow 已提交
662 663
int cmd_clone(int argc, const char **argv, const char *prefix)
{
664
	int is_bundle = 0, is_local;
D
Daniel Barkalow 已提交
665 666 667
	struct stat buf;
	const char *repo_name, *repo, *work_tree, *git_dir;
	char *path, *dir;
668
	int dest_exists;
D
Daniel Barkalow 已提交
669
	const struct ref *refs, *remote_head;
670 671
	const struct ref *remote_head_points_at;
	const struct ref *our_head_points_at;
D
Daniel Barkalow 已提交
672
	struct ref *mapped_refs;
673
	const struct ref *ref;
674 675
	struct strbuf key = STRBUF_INIT, value = STRBUF_INIT;
	struct strbuf branch_top = STRBUF_INIT, reflog_msg = STRBUF_INIT;
676
	struct transport *transport = NULL;
677
	const char *src_ref_prefix = "refs/heads/";
678
	struct remote *remote;
679
	int err = 0, complete_refs_before_fetch = 1;
D
Daniel Barkalow 已提交
680

681 682
	struct refspec *refspec;
	const char *fetch_pattern;
D
Daniel Barkalow 已提交
683 684 685

	junk_pid = getpid();

J
Jeff King 已提交
686
	packet_trace_identity("clone");
687
	argc = parse_options(argc, argv, prefix, builtin_clone_options,
D
Daniel Barkalow 已提交
688 689
			     builtin_clone_usage, 0);

J
Jonathan Nieder 已提交
690
	if (argc > 2)
691
		usage_msg_opt(_("Too many arguments."),
J
Jonathan Nieder 已提交
692 693
			builtin_clone_usage, builtin_clone_options);

D
Daniel Barkalow 已提交
694
	if (argc == 0)
695
		usage_msg_opt(_("You must specify a repository to clone."),
J
Jonathan Nieder 已提交
696
			builtin_clone_usage, builtin_clone_options);
D
Daniel Barkalow 已提交
697

698 699 700
	if (option_single_branch == -1)
		option_single_branch = option_depth ? 1 : 0;

701 702 703
	if (option_mirror)
		option_bare = 1;

D
Daniel Barkalow 已提交
704 705
	if (option_bare) {
		if (option_origin)
706
			die(_("--bare and --origin %s options are incompatible."),
D
Daniel Barkalow 已提交
707
			    option_origin);
708 709
		if (real_git_dir)
			die(_("--bare and --separate-git-dir are incompatible."));
D
Daniel Barkalow 已提交
710 711 712 713 714 715 716 717 718 719
		option_no_checkout = 1;
	}

	if (!option_origin)
		option_origin = "origin";

	repo_name = argv[0];

	path = get_repo_path(repo_name, &is_bundle);
	if (path)
720
		repo = xstrdup(absolute_path(repo_name));
D
Daniel Barkalow 已提交
721
	else if (!strchr(repo_name, ':'))
722
		die(_("repository '%s' does not exist"), repo_name);
D
Daniel Barkalow 已提交
723 724
	else
		repo = repo_name;
725
	is_local = option_local != 0 && path && !is_bundle;
726
	if (is_local && option_depth)
727
		warning(_("--depth is ignored in local clones; use file:// instead."));
D
Daniel Barkalow 已提交
728 729 730 731

	if (argc == 2)
		dir = xstrdup(argv[1]);
	else
732
		dir = guess_dir_name(repo_name, is_bundle, option_bare);
733
	strip_trailing_slashes(dir);
D
Daniel Barkalow 已提交
734

735 736
	dest_exists = !stat(dir, &buf);
	if (dest_exists && !is_empty_dir(dir))
737 738
		die(_("destination path '%s' already exists and is not "
			"an empty directory."), dir);
D
Daniel Barkalow 已提交
739 740 741 742 743 744 745 746

	strbuf_addf(&reflog_msg, "clone: from %s", repo);

	if (option_bare)
		work_tree = NULL;
	else {
		work_tree = getenv("GIT_WORK_TREE");
		if (work_tree && !stat(work_tree, &buf))
747
			die(_("working tree '%s' already exists."), work_tree);
D
Daniel Barkalow 已提交
748 749 750 751 752 753
	}

	if (option_bare || work_tree)
		git_dir = xstrdup(dir);
	else {
		work_tree = dir;
754
		git_dir = mkpathdup("%s/.git", dir);
D
Daniel Barkalow 已提交
755 756 757 758
	}

	if (!option_bare) {
		junk_work_tree = work_tree;
759
		if (safe_create_leading_directories_const(work_tree) < 0)
760
			die_errno(_("could not create leading directories of '%s'"),
761
				  work_tree);
762
		if (!dest_exists && mkdir(work_tree, 0777))
763
			die_errno(_("could not create work tree dir '%s'."),
764
				  work_tree);
D
Daniel Barkalow 已提交
765 766 767 768
		set_git_work_tree(work_tree);
	}
	junk_git_dir = git_dir;
	atexit(remove_junk);
769
	sigchain_push_common(remove_junk_on_signal);
D
Daniel Barkalow 已提交
770

771
	if (safe_create_leading_directories_const(git_dir) < 0)
772
		die(_("could not create leading directories of '%s'"), git_dir);
773 774

	set_git_dir_init(git_dir, real_git_dir, 0);
775
	if (real_git_dir) {
776
		git_dir = real_git_dir;
777 778
		junk_git_dir = real_git_dir;
	}
D
Daniel Barkalow 已提交
779

780 781
	if (0 <= option_verbosity) {
		if (option_bare)
782
			printf(_("Cloning into bare repository '%s'...\n"), dir);
783
		else
784
			printf(_("Cloning into '%s'...\n"), dir);
785
	}
786
	init_db(option_template, INIT_DB_QUIET);
787
	write_config(&option_config);
D
Daniel Barkalow 已提交
788

J
Junio C Hamano 已提交
789
	git_config(git_default_config, NULL);
D
Daniel Barkalow 已提交
790 791

	if (option_bare) {
792 793
		if (option_mirror)
			src_ref_prefix = "refs/";
794
		strbuf_addstr(&branch_top, src_ref_prefix);
D
Daniel Barkalow 已提交
795 796 797

		git_config_set("core.bare", "true");
	} else {
798
		strbuf_addf(&branch_top, "refs/remotes/%s/", option_origin);
799
	}
D
Daniel Barkalow 已提交
800

801
	strbuf_addf(&value, "+%s*:%s*", src_ref_prefix, branch_top.buf);
802 803 804 805
	strbuf_addf(&key, "remote.%s.url", option_origin);
	git_config_set(key.buf, repo);
	strbuf_reset(&key);

806 807
	if (option_reference.nr)
		setup_reference();
808

809 810 811 812
	fetch_pattern = value.buf;
	refspec = parse_fetch_refspec(1, &fetch_pattern);

	strbuf_reset(&value);
D
Daniel Barkalow 已提交
813

814 815
	remote = remote_get(option_origin);
	transport = transport_get(remote, remote->url[0]);
D
Daniel Barkalow 已提交
816

817
	if (!is_local) {
818
		if (!transport->get_refs_list || !transport->fetch)
819
			die(_("Don't know how to clone %s"), transport->url);
820

D
Daniel Barkalow 已提交
821 822 823 824 825
		transport_set_option(transport, TRANS_OPT_KEEP, "yes");

		if (option_depth)
			transport_set_option(transport, TRANS_OPT_DEPTH,
					     option_depth);
826 827
		if (option_single_branch)
			transport_set_option(transport, TRANS_OPT_FOLLOWTAGS, "1");
D
Daniel Barkalow 已提交
828

829
		transport_set_verbosity(transport, option_verbosity, option_progress);
D
Daniel Barkalow 已提交
830

831 832 833
		if (option_upload_pack)
			transport_set_option(transport, TRANS_OPT_UPLOADPACK,
					     option_upload_pack);
D
Daniel Barkalow 已提交
834 835
	}

836
	refs = transport_get_remote_refs(transport);
D
Daniel Barkalow 已提交
837

838 839 840 841 842 843 844 845 846 847 848 849 850 851 852 853 854
	if (refs) {
		mapped_refs = wanted_peer_refs(refs, refspec);
		/*
		 * transport_get_remote_refs() may return refs with null sha-1
		 * in mapped_refs (see struct transport->get_refs_list
		 * comment). In that case we need fetch it early because
		 * remote_head code below relies on it.
		 *
		 * for normal clones, transport_get_remote_refs() should
		 * return reliable ref set, we can delay cloning until after
		 * remote HEAD check.
		 */
		for (ref = refs; ref; ref = ref->next)
			if (is_null_sha1(ref->old_sha1)) {
				complete_refs_before_fetch = 0;
				break;
			}
855

856 857
		if (!is_local && !complete_refs_before_fetch)
			transport_fetch_refs(transport, mapped_refs);
D
Daniel Barkalow 已提交
858

859
		remote_head = find_ref_by_name(refs, "HEAD");
860 861 862 863 864
		remote_head_points_at =
			guess_remote_head(remote_head, mapped_refs, 0);

		if (option_branch) {
			our_head_points_at =
865
				find_remote_branch(mapped_refs, option_branch);
866

867 868 869
			if (!our_head_points_at)
				die(_("Remote branch %s not found in upstream %s"),
				    option_branch, option_origin);
870 871 872
		}
		else
			our_head_points_at = remote_head_points_at;
873 874
	}
	else {
875
		warning(_("You appear to have cloned an empty repository."));
876
		mapped_refs = NULL;
877 878
		our_head_points_at = NULL;
		remote_head_points_at = NULL;
879 880
		remote_head = NULL;
		option_no_checkout = 1;
881
		if (!option_bare)
882
			install_branch_config(0, "master", option_origin,
883
					      "refs/heads/master");
884
	}
D
Daniel Barkalow 已提交
885

886 887 888
	write_refspec_config(src_ref_prefix, our_head_points_at,
			remote_head_points_at, &branch_top);

889 890
	if (is_local)
		clone_local(path, git_dir);
891
	else if (refs && complete_refs_before_fetch)
892
		transport_fetch_refs(transport, mapped_refs);
D
Daniel Barkalow 已提交
893

894 895
	update_remote_refs(refs, mapped_refs, remote_head_points_at,
			   branch_top.buf, reflog_msg.buf);
D
Daniel Barkalow 已提交
896

897
	update_head(our_head_points_at, remote_head, reflog_msg.buf);
898

899 900
	transport_unlock_pack(transport);
	transport_disconnect(transport);
901

902
	err = checkout();
D
Daniel Barkalow 已提交
903 904

	strbuf_release(&reflog_msg);
905 906 907
	strbuf_release(&branch_top);
	strbuf_release(&key);
	strbuf_release(&value);
D
Daniel Barkalow 已提交
908
	junk_pid = 0;
909
	return err;
D
Daniel Barkalow 已提交
910
}