clone.c 25.9 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"
26
#include "connected.h"
D
Daniel Barkalow 已提交
27 28 29 30 31 32 33 34 35 36

/*
 * 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[] = {
37
	N_("git clone [options] [--] <repo> [<dir>]"),
D
Daniel Barkalow 已提交
38 39 40
	NULL
};

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

static struct option builtin_clone_options[] = {
63
	OPT__VERBOSITY(&option_verbosity),
64
	OPT_BOOL(0, "progress", &option_progress,
65
		 N_("force progress reporting")),
D
Daniel Barkalow 已提交
66
	OPT_BOOLEAN('n', "no-checkout", &option_no_checkout,
67 68
		    N_("don't create a checkout")),
	OPT_BOOLEAN(0, "bare", &option_bare, N_("create a bare repository")),
J
Junio C Hamano 已提交
69
	{ OPTION_BOOLEAN, 0, "naked", &option_bare, NULL,
70
		N_("create a bare repository"),
J
Junio C Hamano 已提交
71
		PARSE_OPT_NOARG | PARSE_OPT_HIDDEN },
72
	OPT_BOOLEAN(0, "mirror", &option_mirror,
73
		    N_("create a mirror repository (implies bare)")),
74
	OPT_BOOL('l', "local", &option_local,
75
		N_("to clone from a local repository")),
D
Daniel Barkalow 已提交
76
	OPT_BOOLEAN(0, "no-hardlinks", &option_no_hardlinks,
77
		    N_("don't use local hardlinks, always copy")),
D
Daniel Barkalow 已提交
78
	OPT_BOOLEAN('s', "shared", &option_shared,
79
		    N_("setup as shared repository")),
80
	OPT_BOOLEAN(0, "recursive", &option_recursive,
81
		    N_("initialize submodules in the clone")),
82
	OPT_BOOLEAN(0, "recurse-submodules", &option_recursive,
83 84 85 86 87 88 89 90 91 92 93 94 95
		    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")),
96
	OPT_BOOL(0, "single-branch", &option_single_branch,
97 98 99 100 101
		    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 已提交
102 103 104
	OPT_END()
};

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

D
Daniel Barkalow 已提交
109 110
static char *get_repo_path(const char *repo, int *is_bundle)
{
111
	static char *suffix[] = { "/.git", "", ".git/.git", ".git" };
D
Daniel Barkalow 已提交
112 113 114 115 116 117 118
	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]);
119 120
		if (stat(path, &st))
			continue;
121
		if (S_ISDIR(st.st_mode) && is_git_directory(path)) {
D
Daniel Barkalow 已提交
122
			*is_bundle = 0;
123
			return xstrdup(absolute_path(path));
124 125 126 127 128 129 130 131 132 133 134 135 136 137 138
		} 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 已提交
139 140 141 142 143 144 145 146
		}
	}

	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;
147
			return xstrdup(absolute_path(path));
D
Daniel Barkalow 已提交
148 149 150 151 152 153
		}
	}

	return NULL;
}

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

	/*
160
	 * Strip trailing spaces, slashes and /.git
161
	 */
162
	while (repo < end && (is_dir_sep(end[-1]) || isspace(end[-1])))
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 188
		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 已提交
189 190
	}

191
	if (is_bare) {
192 193
		struct strbuf result = STRBUF_INIT;
		strbuf_addf(&result, "%.*s.git", (int)(end - start), start);
194
		dir = strbuf_detach(&result, NULL);
195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218
	} 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';
219
	}
220
	return dir;
D
Daniel Barkalow 已提交
221 222
}

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

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

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

238
	/* Beware: read_gitfile(), real_path() and mkpath() return static buffer */
239
	ref_git = xstrdup(real_path(item->string));
240 241 242 243 244 245 246 247 248 249

	repo = read_gitfile(ref_git);
	if (!repo)
		repo = read_gitfile(mkpath("%s/.git", ref_git));
	if (repo) {
		free(ref_git);
		ref_git = xstrdup(repo);
	}

	if (!repo && is_directory(mkpath("%s/.git/objects", ref_git))) {
250
		char *ref_git_git = mkpathdup("%s/.git", ref_git);
251 252 253
		free(ref_git);
		ref_git = ref_git_git;
	} else if (!is_directory(mkpath("%s/objects", ref_git)))
254
		die(_("reference repository '%s' is not a local repository."),
255
		    item->string);
D
Daniel Barkalow 已提交
256

257 258 259 260
	strbuf_addf(&alternate, "%s/objects", ref_git);
	add_to_alternates_file(alternate.buf);
	strbuf_release(&alternate);
	free(ref_git);
261 262
	return 0;
}
D
Daniel Barkalow 已提交
263

264 265 266
static void setup_reference(void)
{
	for_each_string_list(&option_reference, add_one_reference, NULL);
D
Daniel Barkalow 已提交
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 294 295 296 297 298 299 300 301 302 303 304
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 已提交
305 306 307 308 309 310
{
	struct dirent *de;
	struct stat buf;
	int src_len, dest_len;
	DIR *dir;

311
	dir = opendir(src->buf);
D
Daniel Barkalow 已提交
312
	if (!dir)
313
		die_errno(_("failed to open '%s'"), src->buf);
D
Daniel Barkalow 已提交
314

315
	if (mkdir(dest->buf, 0777)) {
D
Daniel Barkalow 已提交
316
		if (errno != EEXIST)
317
			die_errno(_("failed to create directory '%s'"), dest->buf);
318
		else if (stat(dest->buf, &buf))
319
			die_errno(_("failed to stat '%s'"), dest->buf);
D
Daniel Barkalow 已提交
320
		else if (!S_ISDIR(buf.st_mode))
321
			die(_("%s exists and is not a directory"), dest->buf);
D
Daniel Barkalow 已提交
322 323
	}

324 325 326 327
	strbuf_addch(src, '/');
	src_len = src->len;
	strbuf_addch(dest, '/');
	dest_len = dest->len;
D
Daniel Barkalow 已提交
328 329

	while ((de = readdir(dir)) != NULL) {
330 331 332 333 334
		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)) {
335
			warning (_("failed to stat %s\n"), src->buf);
D
Daniel Barkalow 已提交
336 337 338 339
			continue;
		}
		if (S_ISDIR(buf.st_mode)) {
			if (de->d_name[0] != '.')
340 341 342 343 344 345 346 347
				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 已提交
348 349 350
			continue;
		}

351
		if (unlink(dest->buf) && errno != ENOENT)
352
			die_errno(_("failed to unlink '%s'"), dest->buf);
353
		if (!option_no_hardlinks) {
354
			if (!link(src->buf, dest->buf))
355
				continue;
356
			if (option_local > 0)
357
				die_errno(_("failed to create link '%s'"), dest->buf);
358
			option_no_hardlinks = 1;
D
Daniel Barkalow 已提交
359
		}
360
		if (copy_file_with_time(dest->buf, src->buf, 0666))
361
			die_errno(_("failed to copy file to '%s'"), dest->buf);
D
Daniel Barkalow 已提交
362
	}
363
	closedir(dir);
D
Daniel Barkalow 已提交
364 365
}

366
static void clone_local(const char *src_repo, const char *dest_repo)
D
Daniel Barkalow 已提交
367
{
368 369 370 371 372 373 374 375
	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;
376 377
		strbuf_addf(&src, "%s/objects", src_repo);
		strbuf_addf(&dest, "%s/objects", dest_repo);
378
		copy_or_link_directory(&src, &dest, src_repo, src.len);
379 380
		strbuf_release(&src);
		strbuf_release(&dest);
D
Daniel Barkalow 已提交
381 382
	}

383
	if (0 <= option_verbosity)
384
		printf(_("done.\n"));
D
Daniel Barkalow 已提交
385 386 387 388
}

static const char *junk_work_tree;
static const char *junk_git_dir;
389
static pid_t junk_pid;
390
static enum {
391 392 393 394 395 396 397 398 399
	JUNK_LEAVE_NONE,
	JUNK_LEAVE_REPO,
	JUNK_LEAVE_ALL
} junk_mode = JUNK_LEAVE_NONE;

static const char junk_leave_repo_msg[] =
N_("Clone succeeded, but checkout failed.\n"
   "You can inspect what was checked out with 'git status'\n"
   "and retry the checkout with 'git checkout -f HEAD'\n");
D
Daniel Barkalow 已提交
400 401 402

static void remove_junk(void)
{
403
	struct strbuf sb = STRBUF_INIT;
404 405 406 407 408 409 410 411 412 413 414 415

	switch (junk_mode) {
	case JUNK_LEAVE_REPO:
		warning("%s", _(junk_leave_repo_msg));
		/* fall-through */
	case JUNK_LEAVE_ALL:
		return;
	default:
		/* proceed to removal */
		break;
	}

D
Daniel Barkalow 已提交
416 417 418 419 420 421 422 423 424 425 426 427 428 429 430 431 432
	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();
433
	sigchain_pop(signo);
D
Daniel Barkalow 已提交
434 435 436
	raise(signo);
}

437 438 439 440 441 442 443 444
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);
445 446 447 448 449 450 451 452 453

	if (ref)
		return ref;

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

454 455 456
	return ref;
}

457 458
static struct ref *wanted_peer_refs(const struct ref *refs,
		struct refspec *refspec)
D
Daniel Barkalow 已提交
459
{
J
Jeff King 已提交
460 461 462
	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 已提交
463

464 465 466 467 468
	if (option_single_branch) {
		struct ref *remote_head = NULL;

		if (!option_branch)
			remote_head = guess_remote_head(head, refs, 0);
469 470 471 472 473
		else {
			local_refs = NULL;
			tail = &local_refs;
			remote_head = copy_ref(find_remote_branch(refs, option_branch));
		}
474 475 476 477

		if (!remote_head && option_branch)
			warning(_("Could not find remote branch %s to clone."),
				option_branch);
478
		else {
479
			get_fetch_map(remote_head, refspec, &tail, 0);
480 481 482 483

			/* if --branch=tag, pull the requested tag explicitly */
			get_fetch_map(remote_head, tag_refspec, &tail, 0);
		}
484 485 486 487
	} else
		get_fetch_map(refs, refspec, &tail, 0);

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

490 491 492 493 494 495 496
	return local_refs;
}

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

J
Jeff King 已提交
497 498 499
	for (r = local_refs; r; r = r->next) {
		if (!r->peer_ref)
			continue;
500
		add_packed_ref(r->peer_ref->name, r->old_sha1);
J
Jeff King 已提交
501
	}
J
Johan Herland 已提交
502 503

	pack_refs(PACK_REFS_ALL);
D
Daniel Barkalow 已提交
504 505
}

506 507 508 509 510 511 512 513 514 515 516 517 518 519 520
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);
	}
}

521 522 523 524 525 526 527 528 529 530 531 532 533 534 535 536 537 538 539 540
static int iterate_ref_map(void *cb_data, unsigned char sha1[20])
{
	struct ref **rm = cb_data;
	struct ref *ref = *rm;

	/*
	 * Skip anything missing a peer_ref, which we are not
	 * actually going to write a ref for.
	 */
	while (ref && !ref->peer_ref)
		ref = ref->next;
	/* Returning -1 notes "end of list" to the caller. */
	if (!ref)
		return -1;

	hashcpy(sha1, ref->old_sha1);
	*rm = ref->next;
	return 0;
}

541 542 543 544 545 546
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)
{
547 548
	const struct ref *rm = mapped_refs;

549 550
	if (0 <= option_verbosity)
		printf(_("Checking connectivity... "));
551 552
	if (check_everything_connected(iterate_ref_map, 0, &rm))
		die(_("remote did not send all necessary objects"));
553 554
	if (0 <= option_verbosity)
		printf(_("done\n"));
555

556 557 558 559 560 561 562 563 564 565 566 567 568 569 570 571
	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);
	}
}

572 573 574
static void update_head(const struct ref *our, const struct ref *remote,
			const char *msg)
{
575
	if (our && !prefixcmp(our->name, "refs/heads/")) {
576 577 578 579 580 581 582
		/* 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);
		}
583 584 585 586 587
	} 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);
588 589 590
	} else if (remote) {
		/*
		 * We know remote HEAD points to a non-branch, or
591
		 * HEAD points to a branch but we don't know which one.
592 593 594 595 596 597 598
		 * Detach HEAD in all these cases.
		 */
		update_ref(msg, "HEAD", remote->old_sha1,
			   NULL, REF_NODEREF, DIE_ON_ERR);
	}
}

599 600 601 602 603 604 605 606 607 608 609 610 611 612 613 614 615 616 617
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;
	}
618 619 620 621
	if (!strcmp(head, "HEAD")) {
		if (advice_detached_head)
			detach_advice(sha1_to_hex(sha1));
	} else {
622 623 624 625 626 627 628 629 630 631 632 633 634 635 636
		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;
637
	opts.verbose_update = (option_verbosity >= 0);
638 639 640 641 642 643
	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);
644 645
	if (unpack_trees(1, &t, &opts) < 0)
		die(_("unable to checkout working tree"));
646 647 648 649 650 651 652 653 654 655 656 657 658 659

	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;
}

660 661 662 663 664 665 666 667 668 669 670 671 672 673 674 675
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");
	}
}

676 677 678 679 680 681 682 683 684 685 686 687 688 689 690 691 692 693 694 695 696 697 698 699 700 701 702 703 704 705 706 707 708 709 710 711 712 713 714 715 716 717 718 719 720 721 722 723
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 已提交
724 725
int cmd_clone(int argc, const char **argv, const char *prefix)
{
726
	int is_bundle = 0, is_local;
D
Daniel Barkalow 已提交
727 728 729
	struct stat buf;
	const char *repo_name, *repo, *work_tree, *git_dir;
	char *path, *dir;
730
	int dest_exists;
D
Daniel Barkalow 已提交
731
	const struct ref *refs, *remote_head;
732 733
	const struct ref *remote_head_points_at;
	const struct ref *our_head_points_at;
D
Daniel Barkalow 已提交
734
	struct ref *mapped_refs;
735
	const struct ref *ref;
736 737
	struct strbuf key = STRBUF_INIT, value = STRBUF_INIT;
	struct strbuf branch_top = STRBUF_INIT, reflog_msg = STRBUF_INIT;
738
	struct transport *transport = NULL;
739
	const char *src_ref_prefix = "refs/heads/";
740
	struct remote *remote;
741
	int err = 0, complete_refs_before_fetch = 1;
D
Daniel Barkalow 已提交
742

743 744
	struct refspec *refspec;
	const char *fetch_pattern;
D
Daniel Barkalow 已提交
745 746 747

	junk_pid = getpid();

J
Jeff King 已提交
748
	packet_trace_identity("clone");
749
	argc = parse_options(argc, argv, prefix, builtin_clone_options,
D
Daniel Barkalow 已提交
750 751
			     builtin_clone_usage, 0);

J
Jonathan Nieder 已提交
752
	if (argc > 2)
753
		usage_msg_opt(_("Too many arguments."),
J
Jonathan Nieder 已提交
754 755
			builtin_clone_usage, builtin_clone_options);

D
Daniel Barkalow 已提交
756
	if (argc == 0)
757
		usage_msg_opt(_("You must specify a repository to clone."),
J
Jonathan Nieder 已提交
758
			builtin_clone_usage, builtin_clone_options);
D
Daniel Barkalow 已提交
759

760 761 762
	if (option_single_branch == -1)
		option_single_branch = option_depth ? 1 : 0;

763 764 765
	if (option_mirror)
		option_bare = 1;

D
Daniel Barkalow 已提交
766 767
	if (option_bare) {
		if (option_origin)
768
			die(_("--bare and --origin %s options are incompatible."),
D
Daniel Barkalow 已提交
769
			    option_origin);
770 771
		if (real_git_dir)
			die(_("--bare and --separate-git-dir are incompatible."));
D
Daniel Barkalow 已提交
772 773 774 775 776 777 778 779 780 781
		option_no_checkout = 1;
	}

	if (!option_origin)
		option_origin = "origin";

	repo_name = argv[0];

	path = get_repo_path(repo_name, &is_bundle);
	if (path)
782
		repo = xstrdup(absolute_path(repo_name));
D
Daniel Barkalow 已提交
783
	else if (!strchr(repo_name, ':'))
784
		die(_("repository '%s' does not exist"), repo_name);
D
Daniel Barkalow 已提交
785 786
	else
		repo = repo_name;
787
	is_local = option_local != 0 && path && !is_bundle;
788
	if (is_local && option_depth)
789
		warning(_("--depth is ignored in local clones; use file:// instead."));
D
Daniel Barkalow 已提交
790 791 792 793

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

797 798
	dest_exists = !stat(dir, &buf);
	if (dest_exists && !is_empty_dir(dir))
799 800
		die(_("destination path '%s' already exists and is not "
			"an empty directory."), dir);
D
Daniel Barkalow 已提交
801 802 803 804 805 806 807 808

	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))
809
			die(_("working tree '%s' already exists."), work_tree);
D
Daniel Barkalow 已提交
810 811 812 813 814 815
	}

	if (option_bare || work_tree)
		git_dir = xstrdup(dir);
	else {
		work_tree = dir;
816
		git_dir = mkpathdup("%s/.git", dir);
D
Daniel Barkalow 已提交
817 818 819 820
	}

	if (!option_bare) {
		junk_work_tree = work_tree;
821
		if (safe_create_leading_directories_const(work_tree) < 0)
822
			die_errno(_("could not create leading directories of '%s'"),
823
				  work_tree);
824
		if (!dest_exists && mkdir(work_tree, 0777))
825
			die_errno(_("could not create work tree dir '%s'."),
826
				  work_tree);
D
Daniel Barkalow 已提交
827 828 829 830
		set_git_work_tree(work_tree);
	}
	junk_git_dir = git_dir;
	atexit(remove_junk);
831
	sigchain_push_common(remove_junk_on_signal);
D
Daniel Barkalow 已提交
832

833
	if (safe_create_leading_directories_const(git_dir) < 0)
834
		die(_("could not create leading directories of '%s'"), git_dir);
835 836

	set_git_dir_init(git_dir, real_git_dir, 0);
837
	if (real_git_dir) {
838
		git_dir = real_git_dir;
839 840
		junk_git_dir = real_git_dir;
	}
D
Daniel Barkalow 已提交
841

842 843
	if (0 <= option_verbosity) {
		if (option_bare)
844
			printf(_("Cloning into bare repository '%s'...\n"), dir);
845
		else
846
			printf(_("Cloning into '%s'...\n"), dir);
847
	}
848
	init_db(option_template, INIT_DB_QUIET);
849
	write_config(&option_config);
D
Daniel Barkalow 已提交
850

J
Junio C Hamano 已提交
851
	git_config(git_default_config, NULL);
D
Daniel Barkalow 已提交
852 853

	if (option_bare) {
854 855
		if (option_mirror)
			src_ref_prefix = "refs/";
856
		strbuf_addstr(&branch_top, src_ref_prefix);
D
Daniel Barkalow 已提交
857 858 859

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

863
	strbuf_addf(&value, "+%s*:%s*", src_ref_prefix, branch_top.buf);
864 865 866 867
	strbuf_addf(&key, "remote.%s.url", option_origin);
	git_config_set(key.buf, repo);
	strbuf_reset(&key);

868 869
	if (option_reference.nr)
		setup_reference();
870

871 872 873 874
	fetch_pattern = value.buf;
	refspec = parse_fetch_refspec(1, &fetch_pattern);

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

876 877
	remote = remote_get(option_origin);
	transport = transport_get(remote, remote->url[0]);
D
Daniel Barkalow 已提交
878

879
	if (!is_local) {
880
		if (!transport->get_refs_list || !transport->fetch)
881
			die(_("Don't know how to clone %s"), transport->url);
882

D
Daniel Barkalow 已提交
883 884 885 886 887
		transport_set_option(transport, TRANS_OPT_KEEP, "yes");

		if (option_depth)
			transport_set_option(transport, TRANS_OPT_DEPTH,
					     option_depth);
888 889
		if (option_single_branch)
			transport_set_option(transport, TRANS_OPT_FOLLOWTAGS, "1");
D
Daniel Barkalow 已提交
890

891
		transport_set_verbosity(transport, option_verbosity, option_progress);
D
Daniel Barkalow 已提交
892

893 894 895
		if (option_upload_pack)
			transport_set_option(transport, TRANS_OPT_UPLOADPACK,
					     option_upload_pack);
D
Daniel Barkalow 已提交
896 897
	}

898
	refs = transport_get_remote_refs(transport);
D
Daniel Barkalow 已提交
899

900 901 902 903 904 905 906 907 908 909 910 911 912 913 914 915 916
	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;
			}
917

918 919
		if (!is_local && !complete_refs_before_fetch)
			transport_fetch_refs(transport, mapped_refs);
D
Daniel Barkalow 已提交
920

921
		remote_head = find_ref_by_name(refs, "HEAD");
922 923 924 925 926
		remote_head_points_at =
			guess_remote_head(remote_head, mapped_refs, 0);

		if (option_branch) {
			our_head_points_at =
927
				find_remote_branch(mapped_refs, option_branch);
928

929 930 931
			if (!our_head_points_at)
				die(_("Remote branch %s not found in upstream %s"),
				    option_branch, option_origin);
932 933 934
		}
		else
			our_head_points_at = remote_head_points_at;
935 936
	}
	else {
937
		warning(_("You appear to have cloned an empty repository."));
938
		mapped_refs = NULL;
939 940
		our_head_points_at = NULL;
		remote_head_points_at = NULL;
941 942
		remote_head = NULL;
		option_no_checkout = 1;
943
		if (!option_bare)
944
			install_branch_config(0, "master", option_origin,
945
					      "refs/heads/master");
946
	}
D
Daniel Barkalow 已提交
947

948 949 950
	write_refspec_config(src_ref_prefix, our_head_points_at,
			remote_head_points_at, &branch_top);

951 952
	if (is_local)
		clone_local(path, git_dir);
953
	else if (refs && complete_refs_before_fetch)
954
		transport_fetch_refs(transport, mapped_refs);
D
Daniel Barkalow 已提交
955

956 957
	update_remote_refs(refs, mapped_refs, remote_head_points_at,
			   branch_top.buf, reflog_msg.buf);
D
Daniel Barkalow 已提交
958

959
	update_head(our_head_points_at, remote_head, reflog_msg.buf);
960

961 962
	transport_unlock_pack(transport);
	transport_disconnect(transport);
963

964
	junk_mode = JUNK_LEAVE_REPO;
965
	err = checkout();
D
Daniel Barkalow 已提交
966 967

	strbuf_release(&reflog_msg);
968 969 970
	strbuf_release(&branch_top);
	strbuf_release(&key);
	strbuf_release(&value);
971
	junk_mode = JUNK_LEAVE_ALL;
972
	return err;
D
Daniel Barkalow 已提交
973
}