am.c 58.9 KB
Newer Older
1 2 3 4 5
/*
 * Builtin "git am"
 *
 * Based on git-am.sh by Junio C Hamano.
 */
6
#define USE_THE_INDEX_COMPATIBILITY_MACROS
7
#include "cache.h"
8
#include "config.h"
9
#include "builtin.h"
10
#include "exec-cmd.h"
11 12
#include "parse-options.h"
#include "dir.h"
13
#include "run-command.h"
14
#include "quote.h"
J
Junio C Hamano 已提交
15
#include "tempfile.h"
16
#include "lockfile.h"
17 18 19
#include "cache-tree.h"
#include "refs.h"
#include "commit.h"
20 21
#include "diff.h"
#include "diffcore.h"
P
Paul Tan 已提交
22 23
#include "unpack-trees.h"
#include "branch.h"
P
Paul Tan 已提交
24
#include "sequencer.h"
P
Paul Tan 已提交
25 26
#include "revision.h"
#include "merge-recursive.h"
27 28
#include "revision.h"
#include "log-tree.h"
29
#include "notes-utils.h"
P
Paul Tan 已提交
30
#include "rerere.h"
31
#include "prompt.h"
J
Junio C Hamano 已提交
32
#include "mailinfo.h"
33
#include "apply.h"
34
#include "string-list.h"
J
Jonathan Tan 已提交
35
#include "packfile.h"
36
#include "repository.h"
37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52

/**
 * Returns 1 if the file is empty or does not exist, 0 otherwise.
 */
static int is_empty_file(const char *filename)
{
	struct stat st;

	if (stat(filename, &st) < 0) {
		if (errno == ENOENT)
			return 1;
		die_errno(_("could not stat %s"), filename);
	}

	return !st.st_size;
}
53

54 55 56 57 58 59 60 61
/**
 * Returns the length of the first line of msg.
 */
static int linelen(const char *msg)
{
	return strchrnul(msg, '\n') - msg;
}

62 63 64 65 66 67 68 69 70 71 72 73
/**
 * Returns true if `str` consists of only whitespace, false otherwise.
 */
static int str_isspace(const char *str)
{
	for (; *str; str++)
		if (!isspace(*str))
			return 0;

	return 1;
}

74 75
enum patch_format {
	PATCH_FORMAT_UNKNOWN = 0,
76
	PATCH_FORMAT_MBOX,
77
	PATCH_FORMAT_STGIT,
78
	PATCH_FORMAT_STGIT_SERIES,
E
Eric Wong 已提交
79 80
	PATCH_FORMAT_HG,
	PATCH_FORMAT_MBOXRD
81
};
82

83 84 85 86 87 88
enum keep_type {
	KEEP_FALSE = 0,
	KEEP_TRUE,      /* pass -k flag to git-mailinfo */
	KEEP_NON_PATCH  /* pass -b flag to git-mailinfo */
};

P
Paul Tan 已提交
89 90 91 92 93 94
enum scissors_type {
	SCISSORS_UNSET = -1,
	SCISSORS_FALSE = 0,  /* pass --no-scissors to git-mailinfo */
	SCISSORS_TRUE        /* pass --scissors to git-mailinfo */
};

95 96 97 98 99 100
enum signoff_type {
	SIGNOFF_FALSE = 0,
	SIGNOFF_TRUE = 1,
	SIGNOFF_EXPLICIT /* --signoff was set on the command-line */
};

101 102 103 104 105 106 107
struct am_state {
	/* state directory path */
	char *dir;

	/* current and last patch numbers, 1-indexed */
	int cur;
	int last;
108

109 110 111 112 113 114 115
	/* commit metadata and message */
	char *author_name;
	char *author_email;
	char *author_date;
	char *msg;
	size_t msg_len;

P
Paul Tan 已提交
116
	/* when --rebasing, records the original commit the patch came from */
117
	struct object_id orig_commit;
P
Paul Tan 已提交
118

119 120
	/* number of digits in patch filename */
	int prec;
P
Paul Tan 已提交
121 122

	/* various operating modes and command line options */
123
	int interactive;
P
Paul Tan 已提交
124
	int threeway;
P
Paul Tan 已提交
125
	int quiet;
126
	int signoff; /* enum signoff_type */
P
Paul Tan 已提交
127
	int utf8;
128
	int keep; /* enum keep_type */
129
	int message_id;
P
Paul Tan 已提交
130
	int scissors; /* enum scissors_type */
131
	struct argv_array git_apply_opts;
132
	const char *resolvemsg;
133
	int committer_date_is_author_date;
P
Paul Tan 已提交
134
	int ignore_date;
P
Paul Tan 已提交
135
	int allow_rerere_autoupdate;
136
	const char *sign_commit;
P
Paul Tan 已提交
137
	int rebasing;
138 139 140
};

/**
141
 * Initializes am_state with the default values.
142
 */
143
static void am_state_init(struct am_state *state)
144
{
145 146
	int gpgsign;

147 148
	memset(state, 0, sizeof(*state));

149
	state->dir = git_pathdup("rebase-apply");
150 151

	state->prec = 4;
P
Paul Tan 已提交
152

153 154
	git_config_get_bool("am.threeway", &state->threeway);

P
Paul Tan 已提交
155
	state->utf8 = 1;
156 157

	git_config_get_bool("am.messageid", &state->message_id);
P
Paul Tan 已提交
158 159

	state->scissors = SCISSORS_UNSET;
160 161

	argv_array_init(&state->git_apply_opts);
162 163 164

	if (!git_config_get_bool("commit.gpgsign", &gpgsign))
		state->sign_commit = gpgsign ? "" : NULL;
165 166 167 168 169 170 171 172
}

/**
 * Releases memory allocated by an am_state.
 */
static void am_state_release(struct am_state *state)
{
	free(state->dir);
173 174 175 176
	free(state->author_name);
	free(state->author_email);
	free(state->author_date);
	free(state->msg);
177
	argv_array_clear(&state->git_apply_opts);
178 179 180 181 182 183 184 185 186 187
}

/**
 * Returns path relative to the am_state directory.
 */
static inline const char *am_path(const struct am_state *state, const char *path)
{
	return mkpath("%s/%s", state->dir, path);
}

188 189 190
/**
 * For convenience to call write_file()
 */
191 192
static void write_state_text(const struct am_state *state,
			     const char *name, const char *string)
193
{
194
	write_file(am_path(state, name), "%s", string);
195 196
}

197 198
static void write_state_count(const struct am_state *state,
			      const char *name, int value)
199
{
200
	write_file(am_path(state, name), "%d", value);
201 202
}

203 204
static void write_state_bool(const struct am_state *state,
			     const char *name, int value)
205
{
206
	write_state_text(state, name, value ? "t" : "f");
207 208
}

P
Paul Tan 已提交
209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224
/**
 * If state->quiet is false, calls fprintf(fp, fmt, ...), and appends a newline
 * at the end.
 */
static void say(const struct am_state *state, FILE *fp, const char *fmt, ...)
{
	va_list ap;

	va_start(ap, fmt);
	if (!state->quiet) {
		vfprintf(fp, fmt, ap);
		putc('\n', fp);
	}
	va_end(ap);
}

225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263
/**
 * Returns 1 if there is an am session in progress, 0 otherwise.
 */
static int am_in_progress(const struct am_state *state)
{
	struct stat st;

	if (lstat(state->dir, &st) < 0 || !S_ISDIR(st.st_mode))
		return 0;
	if (lstat(am_path(state, "last"), &st) || !S_ISREG(st.st_mode))
		return 0;
	if (lstat(am_path(state, "next"), &st) || !S_ISREG(st.st_mode))
		return 0;
	return 1;
}

/**
 * Reads the contents of `file` in the `state` directory into `sb`. Returns the
 * number of bytes read on success, -1 if the file does not exist. If `trim` is
 * set, trailing whitespace will be removed.
 */
static int read_state_file(struct strbuf *sb, const struct am_state *state,
			const char *file, int trim)
{
	strbuf_reset(sb);

	if (strbuf_read_file(sb, am_path(state, file), 0) >= 0) {
		if (trim)
			strbuf_trim(sb);

		return sb->len;
	}

	if (errno == ENOENT)
		return -1;

	die_errno(_("could not read '%s'"), am_path(state, file));
}

264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279
/**
 * Reads and parses the state directory's "author-script" file, and sets
 * state->author_name, state->author_email and state->author_date accordingly.
 * Returns 0 on success, -1 if the file could not be parsed.
 *
 * The author script is of the format:
 *
 *	GIT_AUTHOR_NAME='$author_name'
 *	GIT_AUTHOR_EMAIL='$author_email'
 *	GIT_AUTHOR_DATE='$author_date'
 *
 * where $author_name, $author_email and $author_date are quoted. We are strict
 * with our parsing, as the file was meant to be eval'd in the old git-am.sh
 * script, and thus if the file differs from what this function expects, it is
 * better to bail out than to do something that the user does not expect.
 */
P
Phillip Wood 已提交
280
static int read_am_author_script(struct am_state *state)
281 282 283 284 285 286 287
{
	const char *filename = am_path(state, "author-script");

	assert(!state->author_name);
	assert(!state->author_email);
	assert(!state->author_date);

P
Phillip Wood 已提交
288 289
	return read_author_script(filename, &state->author_name,
				  &state->author_email, &state->author_date, 1);
290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311
}

/**
 * Saves state->author_name, state->author_email and state->author_date in the
 * state directory's "author-script" file.
 */
static void write_author_script(const struct am_state *state)
{
	struct strbuf sb = STRBUF_INIT;

	strbuf_addstr(&sb, "GIT_AUTHOR_NAME=");
	sq_quote_buf(&sb, state->author_name);
	strbuf_addch(&sb, '\n');

	strbuf_addstr(&sb, "GIT_AUTHOR_EMAIL=");
	sq_quote_buf(&sb, state->author_email);
	strbuf_addch(&sb, '\n');

	strbuf_addstr(&sb, "GIT_AUTHOR_DATE=");
	sq_quote_buf(&sb, state->author_date);
	strbuf_addch(&sb, '\n');

312
	write_state_text(state, "author-script", sb.buf);
313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336 337 338 339 340 341 342 343 344

	strbuf_release(&sb);
}

/**
 * Reads the commit message from the state directory's "final-commit" file,
 * setting state->msg to its contents and state->msg_len to the length of its
 * contents in bytes.
 *
 * Returns 0 on success, -1 if the file does not exist.
 */
static int read_commit_msg(struct am_state *state)
{
	struct strbuf sb = STRBUF_INIT;

	assert(!state->msg);

	if (read_state_file(&sb, state, "final-commit", 0) < 0) {
		strbuf_release(&sb);
		return -1;
	}

	state->msg = strbuf_detach(&sb, &state->msg_len);
	return 0;
}

/**
 * Saves state->msg in the state directory's "final-commit" file.
 */
static void write_commit_msg(const struct am_state *state)
{
	const char *filename = am_path(state, "final-commit");
J
Jeff King 已提交
345
	write_file_buf(filename, state->msg, state->msg_len);
346 347
}

348 349 350 351 352 353 354 355
/**
 * Loads state from disk.
 */
static void am_load(struct am_state *state)
{
	struct strbuf sb = STRBUF_INIT;

	if (read_state_file(&sb, state, "next", 1) < 0)
356
		BUG("state file 'next' does not exist");
357 358 359
	state->cur = strtol(sb.buf, NULL, 10);

	if (read_state_file(&sb, state, "last", 1) < 0)
360
		BUG("state file 'last' does not exist");
361 362
	state->last = strtol(sb.buf, NULL, 10);

P
Phillip Wood 已提交
363
	if (read_am_author_script(state) < 0)
364 365 366 367
		die(_("could not parse author script"));

	read_commit_msg(state);

P
Paul Tan 已提交
368
	if (read_state_file(&sb, state, "original-commit", 1) < 0)
369 370
		oidclr(&state->orig_commit);
	else if (get_oid_hex(sb.buf, &state->orig_commit) < 0)
P
Paul Tan 已提交
371 372
		die(_("could not parse %s"), am_path(state, "original-commit"));

P
Paul Tan 已提交
373 374 375
	read_state_file(&sb, state, "threeway", 1);
	state->threeway = !strcmp(sb.buf, "t");

P
Paul Tan 已提交
376 377 378
	read_state_file(&sb, state, "quiet", 1);
	state->quiet = !strcmp(sb.buf, "t");

P
Paul Tan 已提交
379 380 381
	read_state_file(&sb, state, "sign", 1);
	state->signoff = !strcmp(sb.buf, "t");

P
Paul Tan 已提交
382 383 384
	read_state_file(&sb, state, "utf8", 1);
	state->utf8 = !strcmp(sb.buf, "t");

385 386 387 388 389 390 391 392
	if (file_exists(am_path(state, "rerere-autoupdate"))) {
		read_state_file(&sb, state, "rerere-autoupdate", 1);
		state->allow_rerere_autoupdate = strcmp(sb.buf, "t") ?
			RERERE_NOAUTOUPDATE : RERERE_AUTOUPDATE;
	} else {
		state->allow_rerere_autoupdate = 0;
	}

393 394 395 396 397 398 399 400
	read_state_file(&sb, state, "keep", 1);
	if (!strcmp(sb.buf, "t"))
		state->keep = KEEP_TRUE;
	else if (!strcmp(sb.buf, "b"))
		state->keep = KEEP_NON_PATCH;
	else
		state->keep = KEEP_FALSE;

401 402 403
	read_state_file(&sb, state, "messageid", 1);
	state->message_id = !strcmp(sb.buf, "t");

P
Paul Tan 已提交
404 405 406 407 408 409 410 411
	read_state_file(&sb, state, "scissors", 1);
	if (!strcmp(sb.buf, "t"))
		state->scissors = SCISSORS_TRUE;
	else if (!strcmp(sb.buf, "f"))
		state->scissors = SCISSORS_FALSE;
	else
		state->scissors = SCISSORS_UNSET;

412 413 414 415 416
	read_state_file(&sb, state, "apply-opt", 1);
	argv_array_clear(&state->git_apply_opts);
	if (sq_dequote_to_argv_array(sb.buf, &state->git_apply_opts) < 0)
		die(_("could not parse %s"), am_path(state, "apply-opt"));

P
Paul Tan 已提交
417 418
	state->rebasing = !!file_exists(am_path(state, "rebasing"));

419 420 421 422 423 424 425 426 427 428 429 430 431 432 433 434
	strbuf_release(&sb);
}

/**
 * Removes the am_state directory, forcefully terminating the current am
 * session.
 */
static void am_destroy(const struct am_state *state)
{
	struct strbuf sb = STRBUF_INIT;

	strbuf_addstr(&sb, state->dir);
	remove_dir_recursively(&sb, 0);
	strbuf_release(&sb);
}

435 436 437 438 439 440 441 442 443 444 445
/**
 * Runs applypatch-msg hook. Returns its exit code.
 */
static int run_applypatch_msg_hook(struct am_state *state)
{
	int ret;

	assert(state->msg);
	ret = run_hook_le(NULL, "applypatch-msg", am_path(state, "final-commit"), NULL);

	if (!ret) {
446
		FREE_AND_NULL(state->msg);
447 448 449 450 451 452 453 454
		if (read_commit_msg(state) < 0)
			die(_("'%s' was deleted by the applypatch-msg hook"),
				am_path(state, "final-commit"));
	}

	return ret;
}

P
Paul Tan 已提交
455 456 457 458 459 460 461 462 463 464 465 466 467 468 469 470 471 472 473 474 475 476 477 478
/**
 * Runs post-rewrite hook. Returns it exit code.
 */
static int run_post_rewrite_hook(const struct am_state *state)
{
	struct child_process cp = CHILD_PROCESS_INIT;
	const char *hook = find_hook("post-rewrite");
	int ret;

	if (!hook)
		return 0;

	argv_array_push(&cp.args, hook);
	argv_array_push(&cp.args, "rebase");

	cp.in = xopen(am_path(state, "rewritten"), O_RDONLY);
	cp.stdout_to_stderr = 1;

	ret = run_command(&cp);

	close(cp.in);
	return ret;
}

479 480 481 482 483 484 485 486 487 488 489 490 491 492 493 494 495 496 497 498 499 500 501
/**
 * Reads the state directory's "rewritten" file, and copies notes from the old
 * commits listed in the file to their rewritten commits.
 *
 * Returns 0 on success, -1 on failure.
 */
static int copy_notes_for_rebase(const struct am_state *state)
{
	struct notes_rewrite_cfg *c;
	struct strbuf sb = STRBUF_INIT;
	const char *invalid_line = _("Malformed input line: '%s'.");
	const char *msg = "Notes added by 'git rebase'";
	FILE *fp;
	int ret = 0;

	assert(state->rebasing);

	c = init_copy_notes_for_rewrite("rebase");
	if (!c)
		return 0;

	fp = xfopen(am_path(state, "rewritten"), "r");

502
	while (!strbuf_getline_lf(&sb, fp)) {
503
		struct object_id from_obj, to_obj;
504 505 506 507 508 509

		if (sb.len != GIT_SHA1_HEXSZ * 2 + 1) {
			ret = error(invalid_line, sb.buf);
			goto finish;
		}

510
		if (get_oid_hex(sb.buf, &from_obj)) {
511 512 513 514 515 516 517 518 519
			ret = error(invalid_line, sb.buf);
			goto finish;
		}

		if (sb.buf[GIT_SHA1_HEXSZ] != ' ') {
			ret = error(invalid_line, sb.buf);
			goto finish;
		}

520
		if (get_oid_hex(sb.buf + GIT_SHA1_HEXSZ + 1, &to_obj)) {
521 522 523 524
			ret = error(invalid_line, sb.buf);
			goto finish;
		}

525
		if (copy_note_for_rewrite(c, &from_obj, &to_obj))
526
			ret = error(_("Failed to copy notes from '%s' to '%s'"),
527
					oid_to_hex(&from_obj), oid_to_hex(&to_obj));
528 529 530
	}

finish:
531
	finish_copy_notes_for_rewrite(the_repository, c, msg);
532 533 534 535 536
	fclose(fp);
	strbuf_release(&sb);
	return ret;
}

P
Paul Tan 已提交
537 538 539 540 541 542 543 544 545 546 547 548 549 550 551 552 553 554 555 556
/**
 * Determines if the file looks like a piece of RFC2822 mail by grabbing all
 * non-indented lines and checking if they look like they begin with valid
 * header field names.
 *
 * Returns 1 if the file looks like a piece of mail, 0 otherwise.
 */
static int is_mail(FILE *fp)
{
	const char *header_regex = "^[!-9;-~]+:";
	struct strbuf sb = STRBUF_INIT;
	regex_t regex;
	int ret = 1;

	if (fseek(fp, 0L, SEEK_SET))
		die_errno(_("fseek failed"));

	if (regcomp(&regex, header_regex, REG_NOSUB | REG_EXTENDED))
		die("invalid pattern: %s", header_regex);

557
	while (!strbuf_getline(&sb, fp)) {
P
Paul Tan 已提交
558 559 560 561 562 563 564 565 566 567 568 569 570 571 572 573 574 575 576 577 578 579 580 581 582 583 584 585 586
		if (!sb.len)
			break; /* End of header */

		/* Ignore indented folded lines */
		if (*sb.buf == '\t' || *sb.buf == ' ')
			continue;

		/* It's a header if it matches header_regex */
		if (regexec(&regex, sb.buf, 0, NULL, 0)) {
			ret = 0;
			goto done;
		}
	}

done:
	regfree(&regex);
	strbuf_release(&sb);
	return ret;
}

/**
 * Attempts to detect the patch_format of the patches contained in `paths`,
 * returning the PATCH_FORMAT_* enum value. Returns PATCH_FORMAT_UNKNOWN if
 * detection fails.
 */
static int detect_patch_format(const char **paths)
{
	enum patch_format ret = PATCH_FORMAT_UNKNOWN;
	struct strbuf l1 = STRBUF_INIT;
587 588
	struct strbuf l2 = STRBUF_INIT;
	struct strbuf l3 = STRBUF_INIT;
P
Paul Tan 已提交
589 590 591 592 593 594 595 596 597 598 599 600 601 602 603
	FILE *fp;

	/*
	 * We default to mbox format if input is from stdin and for directories
	 */
	if (!*paths || !strcmp(*paths, "-") || is_directory(*paths))
		return PATCH_FORMAT_MBOX;

	/*
	 * Otherwise, check the first few lines of the first patch, starting
	 * from the first non-blank line, to try to detect its format.
	 */

	fp = xfopen(*paths, "r");

604
	while (!strbuf_getline(&l1, fp)) {
P
Paul Tan 已提交
605 606 607 608 609 610 611 612 613
		if (l1.len)
			break;
	}

	if (starts_with(l1.buf, "From ") || starts_with(l1.buf, "From: ")) {
		ret = PATCH_FORMAT_MBOX;
		goto done;
	}

614 615 616 617 618
	if (starts_with(l1.buf, "# This series applies on GIT commit")) {
		ret = PATCH_FORMAT_STGIT_SERIES;
		goto done;
	}

619 620 621 622 623
	if (!strcmp(l1.buf, "# HG changeset patch")) {
		ret = PATCH_FORMAT_HG;
		goto done;
	}

624 625
	strbuf_getline(&l2, fp);
	strbuf_getline(&l3, fp);
626 627 628 629 630 631 632 633 634 635 636 637 638

	/*
	 * If the second line is empty and the third is a From, Author or Date
	 * entry, this is likely an StGit patch.
	 */
	if (l1.len && !l2.len &&
		(starts_with(l3.buf, "From:") ||
		 starts_with(l3.buf, "Author:") ||
		 starts_with(l3.buf, "Date:"))) {
		ret = PATCH_FORMAT_STGIT;
		goto done;
	}

P
Paul Tan 已提交
639 640 641 642 643 644 645 646
	if (l1.len && is_mail(fp)) {
		ret = PATCH_FORMAT_MBOX;
		goto done;
	}

done:
	fclose(fp);
	strbuf_release(&l1);
647 648
	strbuf_release(&l2);
	strbuf_release(&l3);
P
Paul Tan 已提交
649 650 651
	return ret;
}

652 653 654 655
/**
 * Splits out individual email patches from `paths`, where each path is either
 * a mbox file or a Maildir. Returns 0 on success, -1 on failure.
 */
E
Eric Wong 已提交
656 657
static int split_mail_mbox(struct am_state *state, const char **paths,
				int keep_cr, int mboxrd)
658 659 660
{
	struct child_process cp = CHILD_PROCESS_INIT;
	struct strbuf last = STRBUF_INIT;
661
	int ret;
662 663 664 665 666 667

	cp.git_cmd = 1;
	argv_array_push(&cp.args, "mailsplit");
	argv_array_pushf(&cp.args, "-d%d", state->prec);
	argv_array_pushf(&cp.args, "-o%s", state->dir);
	argv_array_push(&cp.args, "-b");
668 669
	if (keep_cr)
		argv_array_push(&cp.args, "--keep-cr");
E
Eric Wong 已提交
670 671
	if (mboxrd)
		argv_array_push(&cp.args, "--mboxrd");
672 673 674
	argv_array_push(&cp.args, "--");
	argv_array_pushv(&cp.args, paths);

675 676 677
	ret = capture_command(&cp, &last, 8);
	if (ret)
		goto exit;
678 679 680 681

	state->cur = 1;
	state->last = strtol(last.buf, NULL, 10);

682 683 684
exit:
	strbuf_release(&last);
	return ret ? -1 : 0;
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
/**
 * Callback signature for split_mail_conv(). The foreign patch should be
 * read from `in`, and the converted patch (in RFC2822 mail format) should be
 * written to `out`. Return 0 on success, or -1 on failure.
 */
typedef int (*mail_conv_fn)(FILE *out, FILE *in, int keep_cr);

/**
 * Calls `fn` for each file in `paths` to convert the foreign patch to the
 * RFC2822 mail format suitable for parsing with git-mailinfo.
 *
 * Returns 0 on success, -1 on failure.
 */
static int split_mail_conv(mail_conv_fn fn, struct am_state *state,
			const char **paths, int keep_cr)
{
	static const char *stdin_only[] = {"-", NULL};
	int i;

	if (!*paths)
		paths = stdin_only;

	for (i = 0; *paths; paths++, i++) {
		FILE *in, *out;
		const char *mail;
		int ret;

		if (!strcmp(*paths, "-"))
			in = stdin;
		else
			in = fopen(*paths, "r");

		if (!in)
720 721
			return error_errno(_("could not open '%s' for reading"),
					   *paths);
722 723 724 725

		mail = mkpath("%s/%0*d", state->dir, state->prec, i + 1);

		out = fopen(mail, "w");
726 727 728
		if (!out) {
			if (in != stdin)
				fclose(in);
729 730
			return error_errno(_("could not open '%s' for writing"),
					   mail);
731
		}
732 733 734 735

		ret = fn(out, in, keep_cr);

		fclose(out);
736 737
		if (in != stdin)
			fclose(in);
738 739 740 741 742 743 744 745 746 747 748 749 750 751 752 753 754 755 756

		if (ret)
			return error(_("could not parse patch '%s'"), *paths);
	}

	state->cur = 1;
	state->last = i;
	return 0;
}

/**
 * A split_mail_conv() callback that converts an StGit patch to an RFC2822
 * message suitable for parsing with git-mailinfo.
 */
static int stgit_patch_to_mail(FILE *out, FILE *in, int keep_cr)
{
	struct strbuf sb = STRBUF_INIT;
	int subject_printed = 0;

757
	while (!strbuf_getline_lf(&sb, in)) {
758 759 760 761 762 763 764 765 766 767 768 769 770 771 772 773 774 775 776 777 778 779 780 781 782 783 784
		const char *str;

		if (str_isspace(sb.buf))
			continue;
		else if (skip_prefix(sb.buf, "Author:", &str))
			fprintf(out, "From:%s\n", str);
		else if (starts_with(sb.buf, "From") || starts_with(sb.buf, "Date"))
			fprintf(out, "%s\n", sb.buf);
		else if (!subject_printed) {
			fprintf(out, "Subject: %s\n", sb.buf);
			subject_printed = 1;
		} else {
			fprintf(out, "\n%s\n", sb.buf);
			break;
		}
	}

	strbuf_reset(&sb);
	while (strbuf_fread(&sb, 8192, in) > 0) {
		fwrite(sb.buf, 1, sb.len, out);
		strbuf_reset(&sb);
	}

	strbuf_release(&sb);
	return 0;
}

785 786 787 788 789 790 791 792 793 794 795 796 797 798 799 800 801 802 803 804 805 806 807 808 809 810 811
/**
 * This function only supports a single StGit series file in `paths`.
 *
 * Given an StGit series file, converts the StGit patches in the series into
 * RFC2822 messages suitable for parsing with git-mailinfo, and queues them in
 * the state directory.
 *
 * Returns 0 on success, -1 on failure.
 */
static int split_mail_stgit_series(struct am_state *state, const char **paths,
					int keep_cr)
{
	const char *series_dir;
	char *series_dir_buf;
	FILE *fp;
	struct argv_array patches = ARGV_ARRAY_INIT;
	struct strbuf sb = STRBUF_INIT;
	int ret;

	if (!paths[0] || paths[1])
		return error(_("Only one StGIT patch series can be applied at once"));

	series_dir_buf = xstrdup(*paths);
	series_dir = dirname(series_dir_buf);

	fp = fopen(*paths, "r");
	if (!fp)
812
		return error_errno(_("could not open '%s' for reading"), *paths);
813

814
	while (!strbuf_getline_lf(&sb, fp)) {
815 816 817 818 819 820 821 822 823 824 825 826 827 828 829 830
		if (*sb.buf == '#')
			continue; /* skip comment lines */

		argv_array_push(&patches, mkpath("%s/%s", series_dir, sb.buf));
	}

	fclose(fp);
	strbuf_release(&sb);
	free(series_dir_buf);

	ret = split_mail_conv(stgit_patch_to_mail, state, patches.argv, keep_cr);

	argv_array_clear(&patches);
	return ret;
}

831 832 833 834 835 836 837
/**
 * A split_patches_conv() callback that converts a mercurial patch to a RFC2822
 * message suitable for parsing with git-mailinfo.
 */
static int hg_patch_to_mail(FILE *out, FILE *in, int keep_cr)
{
	struct strbuf sb = STRBUF_INIT;
838
	int rc = 0;
839

840
	while (!strbuf_getline_lf(&sb, in)) {
841 842 843 844 845
		const char *str;

		if (skip_prefix(sb.buf, "# User ", &str))
			fprintf(out, "From: %s\n", str);
		else if (skip_prefix(sb.buf, "# Date ", &str)) {
846
			timestamp_t timestamp;
847 848 849 850
			long tz, tz2;
			char *end;

			errno = 0;
851
			timestamp = parse_timestamp(str, &end, 10);
852 853 854 855
			if (errno) {
				rc = error(_("invalid timestamp"));
				goto exit;
			}
856

857 858 859 860
			if (!skip_prefix(end, " ", &str)) {
				rc = error(_("invalid Date line"));
				goto exit;
			}
861 862 863

			errno = 0;
			tz = strtol(str, &end, 10);
864 865 866 867
			if (errno) {
				rc = error(_("invalid timezone offset"));
				goto exit;
			}
868

869 870 871 872
			if (*end) {
				rc = error(_("invalid Date line"));
				goto exit;
			}
873 874 875 876 877 878 879 880 881 882 883 884 885 886 887 888 889 890 891 892 893 894 895 896

			/*
			 * mercurial's timezone is in seconds west of UTC,
			 * however git's timezone is in hours + minutes east of
			 * UTC. Convert it.
			 */
			tz2 = labs(tz) / 3600 * 100 + labs(tz) % 3600 / 60;
			if (tz > 0)
				tz2 = -tz2;

			fprintf(out, "Date: %s\n", show_date(timestamp, tz2, DATE_MODE(RFC2822)));
		} else if (starts_with(sb.buf, "# ")) {
			continue;
		} else {
			fprintf(out, "\n%s\n", sb.buf);
			break;
		}
	}

	strbuf_reset(&sb);
	while (strbuf_fread(&sb, 8192, in) > 0) {
		fwrite(sb.buf, 1, sb.len, out);
		strbuf_reset(&sb);
	}
897
exit:
898
	strbuf_release(&sb);
899
	return rc;
900 901
}

902 903 904 905 906 907 908 909 910 911 912 913
/**
 * Splits a list of files/directories into individual email patches. Each path
 * in `paths` must be a file/directory that is formatted according to
 * `patch_format`.
 *
 * Once split out, the individual email patches will be stored in the state
 * directory, with each patch's filename being its index, padded to state->prec
 * digits.
 *
 * state->cur will be set to the index of the first mail, and state->last will
 * be set to the index of the last mail.
 *
914 915 916
 * Set keep_cr to 0 to convert all lines ending with \r\n to end with \n, 1
 * to disable this behavior, -1 to use the default configured setting.
 *
917 918 919
 * Returns 0 on success, -1 on failure.
 */
static int split_mail(struct am_state *state, enum patch_format patch_format,
920
			const char **paths, int keep_cr)
921
{
922 923 924 925 926
	if (keep_cr < 0) {
		keep_cr = 0;
		git_config_get_bool("am.keepcr", &keep_cr);
	}

927 928
	switch (patch_format) {
	case PATCH_FORMAT_MBOX:
E
Eric Wong 已提交
929
		return split_mail_mbox(state, paths, keep_cr, 0);
930 931
	case PATCH_FORMAT_STGIT:
		return split_mail_conv(stgit_patch_to_mail, state, paths, keep_cr);
932 933
	case PATCH_FORMAT_STGIT_SERIES:
		return split_mail_stgit_series(state, paths, keep_cr);
934 935
	case PATCH_FORMAT_HG:
		return split_mail_conv(hg_patch_to_mail, state, paths, keep_cr);
E
Eric Wong 已提交
936 937
	case PATCH_FORMAT_MBOXRD:
		return split_mail_mbox(state, paths, keep_cr, 1);
938
	default:
939
		BUG("invalid patch_format");
940 941 942 943
	}
	return -1;
}

944 945 946
/**
 * Setup a new am session for applying patches
 */
947
static void am_setup(struct am_state *state, enum patch_format patch_format,
948
			const char **paths, int keep_cr)
949
{
950
	struct object_id curr_head;
951
	const char *str;
952
	struct strbuf sb = STRBUF_INIT;
P
Paul Tan 已提交
953

P
Paul Tan 已提交
954 955 956 957 958 959 960 961
	if (!patch_format)
		patch_format = detect_patch_format(paths);

	if (!patch_format) {
		fprintf_ln(stderr, _("Patch format detection failed."));
		exit(128);
	}

962 963
	if (mkdir(state->dir, 0777) < 0 && errno != EEXIST)
		die_errno(_("failed to create directory '%s'"), state->dir);
964
	delete_ref(NULL, "REBASE_HEAD", NULL, REF_NO_DEREF);
965

966
	if (split_mail(state, patch_format, paths, keep_cr) < 0) {
967 968 969 970
		am_destroy(state);
		die(_("Failed to split patches."));
	}

P
Paul Tan 已提交
971 972 973
	if (state->rebasing)
		state->threeway = 1;

974 975 976 977
	write_state_bool(state, "threeway", state->threeway);
	write_state_bool(state, "quiet", state->quiet);
	write_state_bool(state, "sign", state->signoff);
	write_state_bool(state, "utf8", state->utf8);
P
Paul Tan 已提交
978

979 980 981 982
	if (state->allow_rerere_autoupdate)
		write_state_bool(state, "rerere-autoupdate",
			 state->allow_rerere_autoupdate == RERERE_AUTOUPDATE);

983 984 985 986 987 988 989 990 991 992 993
	switch (state->keep) {
	case KEEP_FALSE:
		str = "f";
		break;
	case KEEP_TRUE:
		str = "t";
		break;
	case KEEP_NON_PATCH:
		str = "b";
		break;
	default:
994
		BUG("invalid value for state->keep");
995 996
	}

997 998
	write_state_text(state, "keep", str);
	write_state_bool(state, "messageid", state->message_id);
999

P
Paul Tan 已提交
1000 1001 1002 1003 1004 1005 1006 1007 1008 1009 1010
	switch (state->scissors) {
	case SCISSORS_UNSET:
		str = "";
		break;
	case SCISSORS_FALSE:
		str = "f";
		break;
	case SCISSORS_TRUE:
		str = "t";
		break;
	default:
1011
		BUG("invalid value for state->scissors");
P
Paul Tan 已提交
1012
	}
1013
	write_state_text(state, "scissors", str);
P
Paul Tan 已提交
1014

J
Jeff King 已提交
1015
	sq_quote_argv(&sb, state->git_apply_opts.argv);
1016
	write_state_text(state, "apply-opt", sb.buf);
1017

P
Paul Tan 已提交
1018
	if (state->rebasing)
1019
		write_state_text(state, "rebasing", "");
P
Paul Tan 已提交
1020
	else
1021
		write_state_text(state, "applying", "");
P
Paul Tan 已提交
1022

1023 1024
	if (!get_oid("HEAD", &curr_head)) {
		write_state_text(state, "abort-safety", oid_to_hex(&curr_head));
P
Paul Tan 已提交
1025
		if (!state->rebasing)
1026 1027
			update_ref("am", "ORIG_HEAD", &curr_head, NULL, 0,
				   UPDATE_REFS_DIE_ON_ERR);
P
Paul Tan 已提交
1028
	} else {
1029
		write_state_text(state, "abort-safety", "");
P
Paul Tan 已提交
1030
		if (!state->rebasing)
1031
			delete_ref(NULL, "ORIG_HEAD", NULL, 0);
P
Paul Tan 已提交
1032 1033
	}

1034 1035 1036 1037 1038
	/*
	 * NOTE: Since the "next" and "last" files determine if an am_state
	 * session is in progress, they should be written last.
	 */

1039 1040
	write_state_count(state, "next", state->cur);
	write_state_count(state, "last", state->last);
1041 1042

	strbuf_release(&sb);
1043 1044 1045 1046 1047 1048 1049 1050
}

/**
 * Increments the patch pointer, and cleans am_state for the application of the
 * next patch.
 */
static void am_next(struct am_state *state)
{
1051
	struct object_id head;
P
Paul Tan 已提交
1052

1053 1054 1055 1056
	FREE_AND_NULL(state->author_name);
	FREE_AND_NULL(state->author_email);
	FREE_AND_NULL(state->author_date);
	FREE_AND_NULL(state->msg);
1057 1058 1059 1060 1061
	state->msg_len = 0;

	unlink(am_path(state, "author-script"));
	unlink(am_path(state, "final-commit"));

1062
	oidclr(&state->orig_commit);
P
Paul Tan 已提交
1063
	unlink(am_path(state, "original-commit"));
1064
	delete_ref(NULL, "REBASE_HEAD", NULL, REF_NO_DEREF);
P
Paul Tan 已提交
1065

1066 1067
	if (!get_oid("HEAD", &head))
		write_state_text(state, "abort-safety", oid_to_hex(&head));
P
Paul Tan 已提交
1068
	else
1069
		write_state_text(state, "abort-safety", "");
P
Paul Tan 已提交
1070

1071
	state->cur++;
1072
	write_state_count(state, "next", state->cur);
1073 1074
}

1075 1076 1077 1078 1079 1080 1081 1082 1083 1084 1085 1086 1087
/**
 * Returns the filename of the current patch email.
 */
static const char *msgnum(const struct am_state *state)
{
	static struct strbuf sb = STRBUF_INIT;

	strbuf_reset(&sb);
	strbuf_addf(&sb, "%0*d", state->prec, state->cur);

	return sb.buf;
}

1088 1089 1090 1091 1092
/**
 * Refresh and write index.
 */
static void refresh_and_write_cache(void)
{
1093
	struct lock_file lock_file = LOCK_INIT;
1094

1095
	hold_locked_index(&lock_file, LOCK_DIE_ON_ERROR);
1096
	refresh_cache(REFRESH_QUIET);
1097
	if (write_locked_index(&the_index, &lock_file, COMMIT_LOCK))
1098 1099 1100
		die(_("unable to write index file"));
}

1101 1102 1103 1104 1105 1106 1107 1108 1109
/**
 * Dies with a user-friendly message on how to proceed after resolving the
 * problem. This message can be overridden with state->resolvemsg.
 */
static void NORETURN die_user_resolve(const struct am_state *state)
{
	if (state->resolvemsg) {
		printf_ln("%s", state->resolvemsg);
	} else {
1110
		const char *cmdline = state->interactive ? "git am -i" : "git am";
1111 1112 1113 1114 1115 1116 1117 1118 1119

		printf_ln(_("When you have resolved this problem, run \"%s --continue\"."), cmdline);
		printf_ln(_("If you prefer to skip this patch, run \"%s --skip\" instead."), cmdline);
		printf_ln(_("To restore the original branch and stop patching, run \"%s --abort\"."), cmdline);
	}

	exit(128);
}

1120 1121 1122 1123
/**
 * Appends signoff to the "msg" field of the am_state.
 */
static void am_append_signoff(struct am_state *state)
1124
{
1125
	struct strbuf sb = STRBUF_INIT;
1126

1127
	strbuf_attach(&sb, state->msg, state->msg_len, state->msg_len);
1128
	append_signoff(&sb, 0, 0);
1129 1130 1131
	state->msg = strbuf_detach(&sb, &state->msg_len);
}

1132 1133 1134 1135 1136 1137 1138 1139 1140 1141 1142 1143 1144 1145 1146 1147 1148 1149
/**
 * Parses `mail` using git-mailinfo, extracting its patch and authorship info.
 * state->msg will be set to the patch message. state->author_name,
 * state->author_email and state->author_date will be set to the patch author's
 * name, email and date respectively. The patch body will be written to the
 * state directory's "patch" file.
 *
 * Returns 1 if the patch should be skipped, 0 otherwise.
 */
static int parse_mail(struct am_state *state, const char *mail)
{
	FILE *fp;
	struct strbuf sb = STRBUF_INIT;
	struct strbuf msg = STRBUF_INIT;
	struct strbuf author_name = STRBUF_INIT;
	struct strbuf author_date = STRBUF_INIT;
	struct strbuf author_email = STRBUF_INIT;
	int ret = 0;
J
Junio C Hamano 已提交
1150
	struct mailinfo mi;
1151

J
Junio C Hamano 已提交
1152
	setup_mailinfo(&mi);
1153

J
Junio C Hamano 已提交
1154 1155 1156 1157
	if (state->utf8)
		mi.metainfo_charset = get_commit_output_encoding();
	else
		mi.metainfo_charset = NULL;
1158 1159 1160 1161 1162

	switch (state->keep) {
	case KEEP_FALSE:
		break;
	case KEEP_TRUE:
J
Junio C Hamano 已提交
1163
		mi.keep_subject = 1;
1164 1165
		break;
	case KEEP_NON_PATCH:
J
Junio C Hamano 已提交
1166
		mi.keep_non_patch_brackets_in_subject = 1;
1167 1168
		break;
	default:
1169
		BUG("invalid value for state->keep");
1170 1171
	}

1172
	if (state->message_id)
J
Junio C Hamano 已提交
1173
		mi.add_message_id = 1;
1174

P
Paul Tan 已提交
1175 1176 1177 1178
	switch (state->scissors) {
	case SCISSORS_UNSET:
		break;
	case SCISSORS_FALSE:
J
Junio C Hamano 已提交
1179
		mi.use_scissors = 0;
P
Paul Tan 已提交
1180 1181
		break;
	case SCISSORS_TRUE:
J
Junio C Hamano 已提交
1182
		mi.use_scissors = 1;
P
Paul Tan 已提交
1183 1184
		break;
	default:
1185
		BUG("invalid value for state->scissors");
P
Paul Tan 已提交
1186 1187
	}

1188 1189
	mi.input = xfopen(mail, "r");
	mi.output = xfopen(am_path(state, "info"), "w");
J
Junio C Hamano 已提交
1190
	if (mailinfo(&mi, am_path(state, "msg"), am_path(state, "patch")))
1191 1192
		die("could not parse patch");

J
Junio C Hamano 已提交
1193 1194
	fclose(mi.input);
	fclose(mi.output);
1195

R
René Scharfe 已提交
1196 1197 1198 1199
	if (mi.format_flowed)
		warning(_("Patch sent with format=flowed; "
			  "space at the end of lines might be lost."));

1200 1201
	/* Extract message and author information */
	fp = xfopen(am_path(state, "info"), "r");
1202
	while (!strbuf_getline_lf(&sb, fp)) {
1203 1204 1205 1206 1207 1208 1209 1210 1211 1212 1213 1214 1215 1216 1217 1218 1219 1220 1221 1222 1223 1224
		const char *x;

		if (skip_prefix(sb.buf, "Subject: ", &x)) {
			if (msg.len)
				strbuf_addch(&msg, '\n');
			strbuf_addstr(&msg, x);
		} else if (skip_prefix(sb.buf, "Author: ", &x))
			strbuf_addstr(&author_name, x);
		else if (skip_prefix(sb.buf, "Email: ", &x))
			strbuf_addstr(&author_email, x);
		else if (skip_prefix(sb.buf, "Date: ", &x))
			strbuf_addstr(&author_date, x);
	}
	fclose(fp);

	/* Skip pine's internal folder data */
	if (!strcmp(author_name.buf, "Mail System Internal Data")) {
		ret = 1;
		goto finish;
	}

	if (is_empty_file(am_path(state, "patch"))) {
1225
		printf_ln(_("Patch is empty."));
1226
		die_user_resolve(state);
1227 1228 1229
	}

	strbuf_addstr(&msg, "\n\n");
J
Junio C Hamano 已提交
1230
	strbuf_addbuf(&msg, &mi.log_message);
1231
	strbuf_stripspace(&msg, 0);
1232 1233 1234 1235 1236 1237 1238 1239 1240 1241 1242 1243 1244 1245 1246 1247 1248 1249 1250

	assert(!state->author_name);
	state->author_name = strbuf_detach(&author_name, NULL);

	assert(!state->author_email);
	state->author_email = strbuf_detach(&author_email, NULL);

	assert(!state->author_date);
	state->author_date = strbuf_detach(&author_date, NULL);

	assert(!state->msg);
	state->msg = strbuf_detach(&msg, &state->msg_len);

finish:
	strbuf_release(&msg);
	strbuf_release(&author_date);
	strbuf_release(&author_email);
	strbuf_release(&author_name);
	strbuf_release(&sb);
J
Junio C Hamano 已提交
1251
	clear_mailinfo(&mi);
1252 1253 1254
	return ret;
}

1255 1256 1257 1258
/**
 * Sets commit_id to the commit hash where the mail was generated from.
 * Returns 0 on success, -1 on failure.
 */
1259
static int get_mail_commit_oid(struct object_id *commit_id, const char *mail)
1260 1261 1262 1263
{
	struct strbuf sb = STRBUF_INIT;
	FILE *fp = xfopen(mail, "r");
	const char *x;
1264
	int ret = 0;
1265

1266 1267 1268 1269
	if (strbuf_getline_lf(&sb, fp) ||
	    !skip_prefix(sb.buf, "From ", &x) ||
	    get_oid_hex(x, commit_id) < 0)
		ret = -1;
1270 1271 1272

	strbuf_release(&sb);
	fclose(fp);
1273
	return ret;
1274 1275 1276 1277 1278 1279 1280 1281
}

/**
 * Sets state->msg, state->author_name, state->author_email, state->author_date
 * to the commit's respective info.
 */
static void get_commit_info(struct am_state *state, struct commit *commit)
{
1282
	const char *buffer, *ident_line, *msg;
1283
	size_t ident_len;
1284
	struct ident_split id;
1285 1286 1287 1288 1289

	buffer = logmsg_reencode(commit, NULL, get_commit_output_encoding());

	ident_line = find_commit_header(buffer, "author", &ident_len);

1290
	if (split_ident_line(&id, ident_line, ident_len) < 0)
1291
		die(_("invalid ident line: %.*s"), (int)ident_len, ident_line);
1292 1293

	assert(!state->author_name);
1294
	if (id.name_begin)
1295
		state->author_name =
1296 1297
			xmemdupz(id.name_begin, id.name_end - id.name_begin);
	else
1298 1299 1300
		state->author_name = xstrdup("");

	assert(!state->author_email);
1301
	if (id.mail_begin)
1302
		state->author_email =
1303 1304
			xmemdupz(id.mail_begin, id.mail_end - id.mail_begin);
	else
1305 1306 1307
		state->author_email = xstrdup("");

	assert(!state->author_date);
1308
	state->author_date = xstrdup(show_ident_date(&id, DATE_MODE(NORMAL)));
1309 1310 1311 1312

	assert(!state->msg);
	msg = strstr(buffer, "\n\n");
	if (!msg)
1313
		die(_("unable to parse commit %s"), oid_to_hex(&commit->object.oid));
1314 1315
	state->msg = xstrdup(msg + 2);
	state->msg_len = strlen(state->msg);
1316
	unuse_commit_buffer(commit, buffer);
1317 1318 1319 1320 1321 1322 1323 1324 1325 1326 1327
}

/**
 * Writes `commit` as a patch to the state directory's "patch" file.
 */
static void write_commit_patch(const struct am_state *state, struct commit *commit)
{
	struct rev_info rev_info;
	FILE *fp;

	fp = xfopen(am_path(state, "patch"), "w");
1328
	repo_init_revisions(the_repository, &rev_info, NULL);
1329 1330 1331 1332 1333 1334
	rev_info.diff = 1;
	rev_info.abbrev = 0;
	rev_info.disable_stdin = 1;
	rev_info.show_root_diff = 1;
	rev_info.diffopt.output_format = DIFF_FORMAT_PATCH;
	rev_info.no_commit_id = 1;
1335 1336
	rev_info.diffopt.flags.binary = 1;
	rev_info.diffopt.flags.full_index = 1;
1337 1338 1339 1340 1341 1342 1343 1344
	rev_info.diffopt.use_color = 0;
	rev_info.diffopt.file = fp;
	rev_info.diffopt.close_file = 1;
	add_pending_object(&rev_info, &commit->object, "");
	diff_setup_done(&rev_info.diffopt);
	log_tree_commit(&rev_info, commit);
}

1345 1346 1347 1348 1349 1350 1351
/**
 * Writes the diff of the index against HEAD as a patch to the state
 * directory's "patch" file.
 */
static void write_index_patch(const struct am_state *state)
{
	struct tree *tree;
1352
	struct object_id head;
1353 1354 1355
	struct rev_info rev_info;
	FILE *fp;

1356
	if (!get_oid_tree("HEAD", &head))
1357
		tree = lookup_tree(the_repository, &head);
1358
	else
1359 1360
		tree = lookup_tree(the_repository,
				   the_repository->hash_algo->empty_tree);
1361 1362

	fp = xfopen(am_path(state, "patch"), "w");
1363
	repo_init_revisions(the_repository, &rev_info, NULL);
1364 1365 1366 1367 1368 1369 1370 1371 1372 1373 1374 1375
	rev_info.diff = 1;
	rev_info.disable_stdin = 1;
	rev_info.no_commit_id = 1;
	rev_info.diffopt.output_format = DIFF_FORMAT_PATCH;
	rev_info.diffopt.use_color = 0;
	rev_info.diffopt.file = fp;
	rev_info.diffopt.close_file = 1;
	add_pending_object(&rev_info, &tree->object, "");
	diff_setup_done(&rev_info.diffopt);
	run_diff_index(&rev_info, 1);
}

1376 1377 1378 1379 1380
/**
 * Like parse_mail(), but parses the mail by looking up its commit ID
 * directly. This is used in --rebasing mode to bypass git-mailinfo's munging
 * of patches.
 *
P
Paul Tan 已提交
1381 1382
 * state->orig_commit will be set to the original commit ID.
 *
1383 1384 1385 1386 1387
 * Will always return 0 as the patch should never be skipped.
 */
static int parse_mail_rebase(struct am_state *state, const char *mail)
{
	struct commit *commit;
1388
	struct object_id commit_oid;
1389

1390
	if (get_mail_commit_oid(&commit_oid, mail) < 0)
1391 1392
		die(_("could not parse %s"), mail);

1393
	commit = lookup_commit_or_die(&commit_oid, mail);
1394 1395 1396 1397 1398

	get_commit_info(state, commit);

	write_commit_patch(state, commit);

1399 1400
	oidcpy(&state->orig_commit, &commit_oid);
	write_state_text(state, "original-commit", oid_to_hex(&commit_oid));
1401 1402
	update_ref("am", "REBASE_HEAD", &commit_oid,
		   NULL, REF_NO_DEREF, UPDATE_REFS_DIE_ON_ERR);
P
Paul Tan 已提交
1403

1404 1405 1406
	return 0;
}

1407
/**
P
Paul Tan 已提交
1408 1409
 * Applies current patch with git-apply. Returns 0 on success, -1 otherwise. If
 * `index_file` is not NULL, the patch will be applied to that index.
1410
 */
P
Paul Tan 已提交
1411
static int run_apply(const struct am_state *state, const char *index_file)
1412
{
1413 1414 1415 1416 1417 1418 1419
	struct argv_array apply_paths = ARGV_ARRAY_INIT;
	struct argv_array apply_opts = ARGV_ARRAY_INIT;
	struct apply_state apply_state;
	int res, opts_left;
	int force_apply = 0;
	int options = 0;

1420
	if (init_apply_state(&apply_state, the_repository, NULL))
1421
		BUG("init_apply_state() failed");
1422 1423 1424 1425 1426 1427 1428 1429 1430 1431 1432 1433 1434 1435 1436 1437

	argv_array_push(&apply_opts, "apply");
	argv_array_pushv(&apply_opts, state->git_apply_opts.argv);

	opts_left = apply_parse_options(apply_opts.argc, apply_opts.argv,
					&apply_state, &force_apply, &options,
					NULL);

	if (opts_left != 0)
		die("unknown option passed through to git apply");

	if (index_file) {
		apply_state.index_file = index_file;
		apply_state.cached = 1;
	} else
		apply_state.check_index = 1;
P
Paul Tan 已提交
1438 1439 1440 1441 1442

	/*
	 * If we are allowed to fall back on 3-way merge, don't give false
	 * errors during the initial attempt.
	 */
1443 1444
	if (state->threeway && !index_file)
		apply_state.apply_verbosity = verbosity_silent;
P
Paul Tan 已提交
1445

1446
	if (check_apply_state(&apply_state, force_apply))
1447
		BUG("check_apply_state() failed");
P
Paul Tan 已提交
1448

1449
	argv_array_push(&apply_paths, am_path(state, "patch"));
1450

1451
	res = apply_all_patches(&apply_state, apply_paths.argc, apply_paths.argv, options);
P
Paul Tan 已提交
1452

1453 1454 1455
	argv_array_clear(&apply_paths);
	argv_array_clear(&apply_opts);
	clear_apply_state(&apply_state);
1456

1457 1458
	if (res)
		return res;
1459

1460 1461 1462 1463 1464
	if (index_file) {
		/* Reload index as apply_all_patches() will have modified it. */
		discard_cache();
		read_cache_from(index_file);
	}
P
Paul Tan 已提交
1465 1466 1467 1468 1469 1470 1471 1472 1473 1474 1475 1476 1477

	return 0;
}

/**
 * Builds an index that contains just the blobs needed for a 3way merge.
 */
static int build_fake_ancestor(const struct am_state *state, const char *index_file)
{
	struct child_process cp = CHILD_PROCESS_INIT;

	cp.git_cmd = 1;
	argv_array_push(&cp.args, "apply");
1478
	argv_array_pushv(&cp.args, state->git_apply_opts.argv);
P
Paul Tan 已提交
1479 1480 1481 1482 1483 1484 1485 1486 1487 1488 1489 1490 1491 1492
	argv_array_pushf(&cp.args, "--build-fake-ancestor=%s", index_file);
	argv_array_push(&cp.args, am_path(state, "patch"));

	if (run_command(&cp))
		return -1;

	return 0;
}

/**
 * Attempt a threeway merge, using index_path as the temporary index.
 */
static int fall_back_threeway(const struct am_state *state, const char *index_path)
{
1493 1494 1495 1496 1497
	struct object_id orig_tree, their_tree, our_tree;
	const struct object_id *bases[1] = { &orig_tree };
	struct merge_options o;
	struct commit *result;
	char *their_tree_name;
P
Paul Tan 已提交
1498

1499
	if (get_oid("HEAD", &our_tree) < 0)
1500
		oidcpy(&our_tree, the_hash_algo->empty_tree);
P
Paul Tan 已提交
1501 1502 1503 1504 1505 1506 1507

	if (build_fake_ancestor(state, index_path))
		return error("could not build fake ancestor");

	discard_cache();
	read_cache_from(index_path);

1508
	if (write_index_as_tree(&orig_tree, &the_index, index_path, 0, NULL))
P
Paul Tan 已提交
1509 1510 1511 1512 1513 1514 1515 1516 1517 1518 1519 1520
		return error(_("Repository lacks necessary blobs to fall back on 3-way merge."));

	say(state, stdout, _("Using index info to reconstruct a base tree..."));

	if (!state->quiet) {
		/*
		 * List paths that needed 3-way fallback, so that the user can
		 * review them with extra care to spot mismerges.
		 */
		struct rev_info rev_info;
		const char *diff_filter_str = "--diff-filter=AM";

1521
		repo_init_revisions(the_repository, &rev_info, NULL);
P
Paul Tan 已提交
1522
		rev_info.diffopt.output_format = DIFF_FORMAT_NAME_STATUS;
1523
		diff_opt_parse(&rev_info.diffopt, &diff_filter_str, 1, rev_info.prefix);
1524
		add_pending_oid(&rev_info, "HEAD", &our_tree, 0);
P
Paul Tan 已提交
1525 1526 1527 1528 1529 1530 1531 1532
		diff_setup_done(&rev_info.diffopt);
		run_diff_index(&rev_info, 1);
	}

	if (run_apply(state, index_path))
		return error(_("Did you hand edit your patch?\n"
				"It does not apply to blobs recorded in its index."));

1533
	if (write_index_as_tree(&their_tree, &the_index, index_path, 0, NULL))
P
Paul Tan 已提交
1534 1535 1536 1537
		return error("could not write tree");

	say(state, stdout, _("Falling back to patching base and 3-way merge..."));

1538 1539 1540
	discard_cache();
	read_cache();

P
Paul Tan 已提交
1541 1542
	/*
	 * This is not so wrong. Depending on which base we picked, orig_tree
J
Johannes Schindelin 已提交
1543
	 * may be wildly different from ours, but their_tree has the same set of
P
Paul Tan 已提交
1544 1545 1546 1547 1548
	 * wildly different changes in parts the patch did not touch, so
	 * recursive ends up canceling them, saying that we reverted all those
	 * changes.
	 */

1549
	init_merge_options(&o, the_repository);
1550 1551 1552 1553

	o.branch1 = "HEAD";
	their_tree_name = xstrfmt("%.*s", linelen(state->msg), state->msg);
	o.branch2 = their_tree_name;
1554
	o.detect_directory_renames = 0;
1555 1556 1557 1558 1559

	if (state->quiet)
		o.verbosity = 0;

	if (merge_recursive_generic(&o, &our_tree, &their_tree, 1, bases, &result)) {
1560
		repo_rerere(the_repository, state->allow_rerere_autoupdate);
1561
		free(their_tree_name);
P
Paul Tan 已提交
1562 1563 1564
		return error(_("Failed to merge in the changes."));
	}

1565
	free(their_tree_name);
1566 1567 1568
	return 0;
}

1569 1570 1571 1572 1573 1574 1575
/**
 * Commits the current index with state->msg as the commit message and
 * state->author_name, state->author_email and state->author_date as the author
 * information.
 */
static void do_commit(const struct am_state *state)
{
1576 1577
	struct object_id tree, parent, commit;
	const struct object_id *old_oid;
1578 1579 1580 1581
	struct commit_list *parents = NULL;
	const char *reflog_msg, *author;
	struct strbuf sb = STRBUF_INIT;

1582 1583 1584
	if (run_hook_le(NULL, "pre-applypatch", NULL))
		exit(1);

1585
	if (write_cache_as_tree(&tree, 0, NULL))
1586 1587
		die(_("git write-tree failed to write a tree"));

1588
	if (!get_oid_commit("HEAD", &parent)) {
1589
		old_oid = &parent;
1590 1591
		commit_list_insert(lookup_commit(the_repository, &parent),
				   &parents);
1592
	} else {
1593
		old_oid = NULL;
P
Paul Tan 已提交
1594
		say(state, stderr, _("applying to an empty history"));
1595 1596 1597
	}

	author = fmt_ident(state->author_name, state->author_email,
P
Paul Tan 已提交
1598 1599
			state->ignore_date ? NULL : state->author_date,
			IDENT_STRICT);
1600

1601 1602 1603 1604
	if (state->committer_date_is_author_date)
		setenv("GIT_COMMITTER_DATE",
			state->ignore_date ? "" : state->author_date, 1);

1605 1606
	if (commit_tree(state->msg, state->msg_len, &tree, parents, &commit,
			author, state->sign_commit))
1607 1608 1609 1610 1611 1612 1613 1614 1615
		die(_("failed to write commit object"));

	reflog_msg = getenv("GIT_REFLOG_ACTION");
	if (!reflog_msg)
		reflog_msg = "am";

	strbuf_addf(&sb, "%s: %.*s", reflog_msg, linelen(state->msg),
			state->msg);

1616 1617
	update_ref(sb.buf, "HEAD", &commit, old_oid, 0,
		   UPDATE_REFS_DIE_ON_ERR);
1618

P
Paul Tan 已提交
1619 1620 1621
	if (state->rebasing) {
		FILE *fp = xfopen(am_path(state, "rewritten"), "a");

1622 1623 1624
		assert(!is_null_oid(&state->orig_commit));
		fprintf(fp, "%s ", oid_to_hex(&state->orig_commit));
		fprintf(fp, "%s\n", oid_to_hex(&commit));
P
Paul Tan 已提交
1625 1626 1627
		fclose(fp);
	}

1628 1629
	run_hook_le(NULL, "post-applypatch", NULL);

1630 1631 1632
	strbuf_release(&sb);
}

1633 1634 1635 1636 1637 1638 1639 1640 1641 1642 1643 1644 1645 1646 1647
/**
 * Validates the am_state for resuming -- the "msg" and authorship fields must
 * be filled up.
 */
static void validate_resume_state(const struct am_state *state)
{
	if (!state->msg)
		die(_("cannot resume: %s does not exist."),
			am_path(state, "final-commit"));

	if (!state->author_name || !state->author_email || !state->author_date)
		die(_("cannot resume: %s does not exist."),
			am_path(state, "author-script"));
}

1648 1649 1650 1651 1652 1653 1654 1655 1656 1657 1658 1659 1660 1661 1662 1663 1664 1665 1666 1667 1668 1669 1670 1671 1672 1673 1674 1675 1676 1677 1678 1679 1680 1681 1682 1683 1684 1685 1686 1687 1688 1689 1690 1691 1692 1693 1694 1695 1696 1697 1698 1699
/**
 * Interactively prompt the user on whether the current patch should be
 * applied.
 *
 * Returns 0 if the user chooses to apply the patch, 1 if the user chooses to
 * skip it.
 */
static int do_interactive(struct am_state *state)
{
	assert(state->msg);

	if (!isatty(0))
		die(_("cannot be interactive without stdin connected to a terminal."));

	for (;;) {
		const char *reply;

		puts(_("Commit Body is:"));
		puts("--------------------------");
		printf("%s", state->msg);
		puts("--------------------------");

		/*
		 * TRANSLATORS: Make sure to include [y], [n], [e], [v] and [a]
		 * in your translation. The program will only accept English
		 * input at this point.
		 */
		reply = git_prompt(_("Apply? [y]es/[n]o/[e]dit/[v]iew patch/[a]ccept all: "), PROMPT_ECHO);

		if (!reply) {
			continue;
		} else if (*reply == 'y' || *reply == 'Y') {
			return 0;
		} else if (*reply == 'a' || *reply == 'A') {
			state->interactive = 0;
			return 0;
		} else if (*reply == 'n' || *reply == 'N') {
			return 1;
		} else if (*reply == 'e' || *reply == 'E') {
			struct strbuf msg = STRBUF_INIT;

			if (!launch_editor(am_path(state, "final-commit"), &msg, NULL)) {
				free(state->msg);
				state->msg = strbuf_detach(&msg, &state->msg_len);
			}
			strbuf_release(&msg);
		} else if (*reply == 'v' || *reply == 'V') {
			const char *pager = git_pager(1);
			struct child_process cp = CHILD_PROCESS_INIT;

			if (!pager)
				pager = "cat";
J
Junio C Hamano 已提交
1700
			prepare_pager_args(&cp, pager);
1701 1702 1703 1704 1705 1706
			argv_array_push(&cp.args, am_path(state, "patch"));
			run_command(&cp);
		}
	}
}

1707 1708
/**
 * Applies all queued mail.
1709 1710 1711 1712
 *
 * If `resume` is true, we are "resuming". The "msg" and authorship fields, as
 * well as the state directory's "patch" file is used as-is for applying the
 * patch and committing it.
1713
 */
1714
static void am_run(struct am_state *state, int resume)
1715
{
1716
	const char *argv_gc_auto[] = {"gc", "--auto", NULL};
1717
	struct strbuf sb = STRBUF_INIT;
1718

P
Paul Tan 已提交
1719 1720
	unlink(am_path(state, "dirtyindex"));

1721 1722
	refresh_and_write_cache();

1723
	if (repo_index_has_changes(the_repository, NULL, &sb)) {
1724
		write_state_bool(state, "dirtyindex", 1);
1725
		die(_("Dirty index: cannot apply patches (dirty: %s)"), sb.buf);
P
Paul Tan 已提交
1726
	}
1727 1728 1729

	strbuf_release(&sb);

1730
	while (state->cur <= state->last) {
1731
		const char *mail = am_path(state, msgnum(state));
P
Paul Tan 已提交
1732
		int apply_status;
1733

1734 1735
		reset_ident_date();

1736 1737 1738
		if (!file_exists(mail))
			goto next;

1739 1740 1741
		if (resume) {
			validate_resume_state(state);
		} else {
1742 1743 1744 1745 1746 1747 1748 1749
			int skip;

			if (state->rebasing)
				skip = parse_mail_rebase(state, mail);
			else
				skip = parse_mail(state, mail);

			if (skip)
1750
				goto next; /* mail should be skipped */
1751

1752 1753 1754
			if (state->signoff)
				am_append_signoff(state);

1755 1756 1757
			write_author_script(state);
			write_commit_msg(state);
		}
1758

1759 1760 1761
		if (state->interactive && do_interactive(state))
			goto next;

1762 1763 1764
		if (run_applypatch_msg_hook(state))
			exit(1);

P
Paul Tan 已提交
1765
		say(state, stdout, _("Applying: %.*s"), linelen(state->msg), state->msg);
1766

P
Paul Tan 已提交
1767 1768 1769 1770 1771 1772 1773 1774 1775 1776 1777 1778 1779
		apply_status = run_apply(state, NULL);

		if (apply_status && state->threeway) {
			struct strbuf sb = STRBUF_INIT;

			strbuf_addstr(&sb, am_path(state, "patch-merge-index"));
			apply_status = fall_back_threeway(state, sb.buf);
			strbuf_release(&sb);

			/*
			 * Applying the patch to an earlier tree and merging
			 * the result may have produced the same tree as ours.
			 */
1780
			if (!apply_status &&
1781
			    !repo_index_has_changes(the_repository, NULL, NULL)) {
P
Paul Tan 已提交
1782 1783 1784 1785 1786 1787
				say(state, stdout, _("No changes -- Patch already applied."));
				goto next;
			}
		}

		if (apply_status) {
1788 1789 1790 1791
			printf_ln(_("Patch failed at %s %.*s"), msgnum(state),
				linelen(state->msg), state->msg);

			if (advice_amworkdir)
1792
				advise(_("Use 'git am --show-current-patch' to see the failed patch"));
1793

1794
			die_user_resolve(state);
1795 1796
		}

1797
		do_commit(state);
1798

1799
next:
1800
		am_next(state);
1801 1802 1803 1804

		if (resume)
			am_load(state);
		resume = 0;
1805 1806
	}

P
Paul Tan 已提交
1807 1808
	if (!is_empty_file(am_path(state, "rewritten"))) {
		assert(state->rebasing);
1809
		copy_notes_for_rebase(state);
P
Paul Tan 已提交
1810 1811 1812
		run_post_rewrite_hook(state);
	}

P
Paul Tan 已提交
1813 1814 1815 1816 1817 1818
	/*
	 * In rebasing mode, it's up to the caller to take care of
	 * housekeeping.
	 */
	if (!state->rebasing) {
		am_destroy(state);
1819
		close_all_packs(the_repository->objects);
P
Paul Tan 已提交
1820 1821
		run_command_v_opt(argv_gc_auto, RUN_GIT_CMD);
	}
1822
}
1823

1824 1825 1826 1827 1828 1829 1830 1831 1832
/**
 * Resume the current am session after patch application failure. The user did
 * all the hard work, and we do not have to do any patch application. Just
 * trust and commit what the user has in the index and working tree.
 */
static void am_resolve(struct am_state *state)
{
	validate_resume_state(state);

P
Paul Tan 已提交
1833
	say(state, stdout, _("Applying: %.*s"), linelen(state->msg), state->msg);
1834

1835
	if (!repo_index_has_changes(the_repository, NULL, NULL)) {
1836 1837 1838
		printf_ln(_("No changes - did you forget to use 'git add'?\n"
			"If there is nothing left to stage, chances are that something else\n"
			"already introduced the same changes; you might want to skip this patch."));
1839
		die_user_resolve(state);
1840 1841 1842 1843
	}

	if (unmerged_cache()) {
		printf_ln(_("You still have unmerged paths in your index.\n"
1844 1845
			"You should 'git add' each file with resolved conflicts to mark them as such.\n"
			"You might run `git rm` on a file to accept \"deleted by them\" for it."));
1846
		die_user_resolve(state);
1847 1848
	}

1849 1850 1851 1852 1853 1854
	if (state->interactive) {
		write_index_patch(state);
		if (do_interactive(state))
			goto next;
	}

1855
	repo_rerere(the_repository, 0);
P
Paul Tan 已提交
1856

1857 1858
	do_commit(state);

1859
next:
1860
	am_next(state);
1861
	am_load(state);
1862
	am_run(state, 0);
1863 1864
}

P
Paul Tan 已提交
1865 1866 1867 1868 1869 1870 1871
/**
 * Performs a checkout fast-forward from `head` to `remote`. If `reset` is
 * true, any unmerged entries will be discarded. Returns 0 on success, -1 on
 * failure.
 */
static int fast_forward_to(struct tree *head, struct tree *remote, int reset)
{
1872
	struct lock_file lock_file = LOCK_INIT;
P
Paul Tan 已提交
1873 1874 1875 1876 1877 1878
	struct unpack_trees_options opts;
	struct tree_desc t[2];

	if (parse_tree(head) || parse_tree(remote))
		return -1;

1879
	hold_locked_index(&lock_file, LOCK_DIE_ON_ERROR);
P
Paul Tan 已提交
1880 1881 1882 1883 1884 1885 1886 1887 1888 1889 1890 1891 1892 1893 1894

	refresh_cache(REFRESH_QUIET);

	memset(&opts, 0, sizeof(opts));
	opts.head_idx = 1;
	opts.src_index = &the_index;
	opts.dst_index = &the_index;
	opts.update = 1;
	opts.merge = 1;
	opts.reset = reset;
	opts.fn = twoway_merge;
	init_tree_desc(&t[0], head->buffer, head->size);
	init_tree_desc(&t[1], remote->buffer, remote->size);

	if (unpack_trees(2, t, &opts)) {
1895
		rollback_lock_file(&lock_file);
P
Paul Tan 已提交
1896 1897 1898
		return -1;
	}

1899
	if (write_locked_index(&the_index, &lock_file, COMMIT_LOCK))
P
Paul Tan 已提交
1900 1901 1902 1903 1904
		die(_("unable to write new index file"));

	return 0;
}

1905 1906 1907 1908 1909 1910
/**
 * Merges a tree into the index. The index's stat info will take precedence
 * over the merged tree's. Returns 0 on success, -1 on failure.
 */
static int merge_tree(struct tree *tree)
{
1911
	struct lock_file lock_file = LOCK_INIT;
1912 1913 1914 1915 1916 1917
	struct unpack_trees_options opts;
	struct tree_desc t[1];

	if (parse_tree(tree))
		return -1;

1918
	hold_locked_index(&lock_file, LOCK_DIE_ON_ERROR);
1919 1920 1921 1922 1923 1924 1925 1926 1927 1928

	memset(&opts, 0, sizeof(opts));
	opts.head_idx = 1;
	opts.src_index = &the_index;
	opts.dst_index = &the_index;
	opts.merge = 1;
	opts.fn = oneway_merge;
	init_tree_desc(&t[0], tree->buffer, tree->size);

	if (unpack_trees(1, t, &opts)) {
1929
		rollback_lock_file(&lock_file);
1930 1931 1932
		return -1;
	}

1933
	if (write_locked_index(&the_index, &lock_file, COMMIT_LOCK))
1934 1935 1936 1937 1938
		die(_("unable to write new index file"));

	return 0;
}

P
Paul Tan 已提交
1939 1940 1941 1942
/**
 * Clean the index without touching entries that are not modified between
 * `head` and `remote`.
 */
1943
static int clean_index(const struct object_id *head, const struct object_id *remote)
P
Paul Tan 已提交
1944 1945
{
	struct tree *head_tree, *remote_tree, *index_tree;
1946
	struct object_id index;
P
Paul Tan 已提交
1947

1948
	head_tree = parse_tree_indirect(head);
P
Paul Tan 已提交
1949
	if (!head_tree)
1950
		return error(_("Could not parse object '%s'."), oid_to_hex(head));
P
Paul Tan 已提交
1951

1952
	remote_tree = parse_tree_indirect(remote);
P
Paul Tan 已提交
1953
	if (!remote_tree)
1954
		return error(_("Could not parse object '%s'."), oid_to_hex(remote));
P
Paul Tan 已提交
1955 1956 1957 1958 1959 1960

	read_cache_unmerged();

	if (fast_forward_to(head_tree, head_tree, 1))
		return -1;

1961
	if (write_cache_as_tree(&index, 0, NULL))
P
Paul Tan 已提交
1962 1963
		return -1;

1964
	index_tree = parse_tree_indirect(&index);
P
Paul Tan 已提交
1965
	if (!index_tree)
1966
		return error(_("Could not parse object '%s'."), oid_to_hex(&index));
P
Paul Tan 已提交
1967 1968 1969 1970

	if (fast_forward_to(index_tree, remote_tree, 0))
		return -1;

1971
	if (merge_tree(remote_tree))
P
Paul Tan 已提交
1972 1973
		return -1;

1974
	remove_branch_state(the_repository);
P
Paul Tan 已提交
1975 1976 1977 1978

	return 0;
}

P
Paul Tan 已提交
1979 1980 1981 1982 1983 1984
/**
 * Resets rerere's merge resolution metadata.
 */
static void am_rerere_clear(void)
{
	struct string_list merge_rr = STRING_LIST_INIT_DUP;
1985
	rerere_clear(the_repository, &merge_rr);
P
Paul Tan 已提交
1986 1987 1988
	string_list_clear(&merge_rr, 1);
}

P
Paul Tan 已提交
1989 1990 1991 1992 1993
/**
 * Resume the current am session by skipping the current patch.
 */
static void am_skip(struct am_state *state)
{
1994
	struct object_id head;
P
Paul Tan 已提交
1995

P
Paul Tan 已提交
1996 1997
	am_rerere_clear();

1998
	if (get_oid("HEAD", &head))
1999
		oidcpy(&head, the_hash_algo->empty_tree);
P
Paul Tan 已提交
2000

2001
	if (clean_index(&head, &head))
P
Paul Tan 已提交
2002 2003 2004
		die(_("failed to clean index"));

	am_next(state);
2005
	am_load(state);
P
Paul Tan 已提交
2006 2007 2008
	am_run(state, 0);
}

P
Paul Tan 已提交
2009 2010 2011 2012 2013 2014 2015 2016 2017 2018
/**
 * Returns true if it is safe to reset HEAD to the ORIG_HEAD, false otherwise.
 *
 * It is not safe to reset HEAD when:
 * 1. git-am previously failed because the index was dirty.
 * 2. HEAD has moved since git-am previously failed.
 */
static int safe_to_abort(const struct am_state *state)
{
	struct strbuf sb = STRBUF_INIT;
2019
	struct object_id abort_safety, head;
P
Paul Tan 已提交
2020 2021 2022 2023 2024

	if (file_exists(am_path(state, "dirtyindex")))
		return 0;

	if (read_state_file(&sb, state, "abort-safety", 1) > 0) {
2025
		if (get_oid_hex(sb.buf, &abort_safety))
2026
			die(_("could not parse %s"), am_path(state, "abort-safety"));
P
Paul Tan 已提交
2027
	} else
2028
		oidclr(&abort_safety);
2029
	strbuf_release(&sb);
P
Paul Tan 已提交
2030

2031 2032
	if (get_oid("HEAD", &head))
		oidclr(&head);
P
Paul Tan 已提交
2033

J
Jeff King 已提交
2034
	if (oideq(&head, &abort_safety))
P
Paul Tan 已提交
2035 2036
		return 1;

2037
	warning(_("You seem to have moved HEAD since the last 'am' failure.\n"
P
Paul Tan 已提交
2038 2039 2040 2041 2042 2043 2044 2045 2046 2047
		"Not rewinding to ORIG_HEAD"));

	return 0;
}

/**
 * Aborts the current am session if it is safe to do so.
 */
static void am_abort(struct am_state *state)
{
2048
	struct object_id curr_head, orig_head;
P
Paul Tan 已提交
2049 2050 2051 2052 2053 2054 2055 2056
	int has_curr_head, has_orig_head;
	char *curr_branch;

	if (!safe_to_abort(state)) {
		am_destroy(state);
		return;
	}

P
Paul Tan 已提交
2057 2058
	am_rerere_clear();

2059
	curr_branch = resolve_refdup("HEAD", 0, &curr_head, NULL);
2060
	has_curr_head = curr_branch && !is_null_oid(&curr_head);
P
Paul Tan 已提交
2061
	if (!has_curr_head)
2062
		oidcpy(&curr_head, the_hash_algo->empty_tree);
P
Paul Tan 已提交
2063

2064
	has_orig_head = !get_oid("ORIG_HEAD", &orig_head);
P
Paul Tan 已提交
2065
	if (!has_orig_head)
2066
		oidcpy(&orig_head, the_hash_algo->empty_tree);
P
Paul Tan 已提交
2067

2068
	clean_index(&curr_head, &orig_head);
P
Paul Tan 已提交
2069 2070

	if (has_orig_head)
2071 2072 2073
		update_ref("am --abort", "HEAD", &orig_head,
			   has_curr_head ? &curr_head : NULL, 0,
			   UPDATE_REFS_DIE_ON_ERR);
P
Paul Tan 已提交
2074
	else if (curr_branch)
2075
		delete_ref(NULL, curr_branch, NULL, REF_NO_DEREF);
P
Paul Tan 已提交
2076 2077 2078 2079 2080

	free(curr_branch);
	am_destroy(state);
}

2081 2082 2083 2084 2085 2086
static int show_patch(struct am_state *state)
{
	struct strbuf sb = STRBUF_INIT;
	const char *patch_path;
	int len;

2087 2088 2089 2090 2091 2092 2093 2094 2095 2096 2097
	if (!is_null_oid(&state->orig_commit)) {
		const char *av[4] = { "show", NULL, "--", NULL };
		char *new_oid_str;
		int ret;

		av[1] = new_oid_str = xstrdup(oid_to_hex(&state->orig_commit));
		ret = run_command_v_opt(av, RUN_GIT_CMD);
		free(new_oid_str);
		return ret;
	}

2098 2099 2100 2101 2102 2103 2104 2105 2106 2107 2108
	patch_path = am_path(state, msgnum(state));
	len = strbuf_read_file(&sb, patch_path, 0);
	if (len < 0)
		die_errno(_("failed to read '%s'"), patch_path);

	setup_pager();
	write_in_full(1, sb.buf, sb.len);
	strbuf_release(&sb);
	return 0;
}

2109 2110 2111 2112 2113 2114 2115 2116
/**
 * parse_options() callback that validates and sets opt->value to the
 * PATCH_FORMAT_* enum value corresponding to `arg`.
 */
static int parse_opt_patchformat(const struct option *opt, const char *arg, int unset)
{
	int *opt_value = opt->value;

J
Jeff King 已提交
2117 2118 2119
	if (unset)
		*opt_value = PATCH_FORMAT_UNKNOWN;
	else if (!strcmp(arg, "mbox"))
2120
		*opt_value = PATCH_FORMAT_MBOX;
2121 2122
	else if (!strcmp(arg, "stgit"))
		*opt_value = PATCH_FORMAT_STGIT;
2123 2124
	else if (!strcmp(arg, "stgit-series"))
		*opt_value = PATCH_FORMAT_STGIT_SERIES;
2125 2126
	else if (!strcmp(arg, "hg"))
		*opt_value = PATCH_FORMAT_HG;
E
Eric Wong 已提交
2127 2128
	else if (!strcmp(arg, "mboxrd"))
		*opt_value = PATCH_FORMAT_MBOXRD;
2129 2130 2131 2132 2133
	else
		return error(_("Invalid value for --patch-format: %s"), arg);
	return 0;
}

2134 2135
enum resume_mode {
	RESUME_FALSE = 0,
2136
	RESUME_APPLY,
P
Paul Tan 已提交
2137
	RESUME_RESOLVED,
P
Paul Tan 已提交
2138
	RESUME_SKIP,
N
Nguyễn Thái Ngọc Duy 已提交
2139
	RESUME_ABORT,
2140
	RESUME_QUIT,
2141
	RESUME_SHOW_PATCH
2142 2143
};

2144 2145 2146 2147 2148 2149 2150 2151 2152 2153 2154
static int git_am_config(const char *k, const char *v, void *cb)
{
	int status;

	status = git_gpg_config(k, v, NULL);
	if (status)
		return status;

	return git_default_config(k, v, NULL);
}

2155 2156
int cmd_am(int argc, const char **argv, const char *prefix)
{
2157
	struct am_state state;
2158
	int binary = -1;
2159
	int keep_cr = -1;
2160
	int patch_format = PATCH_FORMAT_UNKNOWN;
2161
	enum resume_mode resume = RESUME_FALSE;
2162
	int in_progress;
2163
	int ret = 0;
2164 2165

	const char * const usage[] = {
2166
		N_("git am [<options>] [(<mbox> | <Maildir>)...]"),
2167
		N_("git am [<options>] (--continue | --skip | --abort)"),
2168 2169 2170 2171
		NULL
	};

	struct option options[] = {
2172 2173
		OPT_BOOL('i', "interactive", &state.interactive,
			N_("run interactively")),
2174
		OPT_HIDDEN_BOOL('b', "binary", &binary,
2175
			N_("historical option -- no-op")),
P
Paul Tan 已提交
2176 2177
		OPT_BOOL('3', "3way", &state.threeway,
			N_("allow fall back on 3way merging if needed")),
P
Paul Tan 已提交
2178
		OPT__QUIET(&state.quiet, N_("be quiet")),
2179 2180 2181
		OPT_SET_INT('s', "signoff", &state.signoff,
			N_("add a Signed-off-by line to the commit message"),
			SIGNOFF_EXPLICIT),
P
Paul Tan 已提交
2182 2183
		OPT_BOOL('u', "utf8", &state.utf8,
			N_("recode into utf8 (default)")),
2184 2185 2186 2187
		OPT_SET_INT('k', "keep", &state.keep,
			N_("pass -k flag to git-mailinfo"), KEEP_TRUE),
		OPT_SET_INT(0, "keep-non-patch", &state.keep,
			N_("pass -b flag to git-mailinfo"), KEEP_NON_PATCH),
2188 2189
		OPT_BOOL('m', "message-id", &state.message_id,
			N_("pass -m flag to git-mailinfo")),
2190 2191 2192 2193 2194 2195
		OPT_SET_INT_F(0, "keep-cr", &keep_cr,
			N_("pass --keep-cr flag to git-mailsplit for mbox format"),
			1, PARSE_OPT_NONEG),
		OPT_SET_INT_F(0, "no-keep-cr", &keep_cr,
			N_("do not pass --keep-cr flag to git-mailsplit independent of am.keepcr"),
			0, PARSE_OPT_NONEG),
P
Paul Tan 已提交
2196 2197
		OPT_BOOL('c', "scissors", &state.scissors,
			N_("strip everything before a scissors line")),
2198 2199 2200 2201 2202 2203 2204 2205 2206 2207 2208 2209 2210 2211 2212 2213 2214 2215 2216 2217 2218 2219 2220 2221
		OPT_PASSTHRU_ARGV(0, "whitespace", &state.git_apply_opts, N_("action"),
			N_("pass it through git-apply"),
			0),
		OPT_PASSTHRU_ARGV(0, "ignore-space-change", &state.git_apply_opts, NULL,
			N_("pass it through git-apply"),
			PARSE_OPT_NOARG),
		OPT_PASSTHRU_ARGV(0, "ignore-whitespace", &state.git_apply_opts, NULL,
			N_("pass it through git-apply"),
			PARSE_OPT_NOARG),
		OPT_PASSTHRU_ARGV(0, "directory", &state.git_apply_opts, N_("root"),
			N_("pass it through git-apply"),
			0),
		OPT_PASSTHRU_ARGV(0, "exclude", &state.git_apply_opts, N_("path"),
			N_("pass it through git-apply"),
			0),
		OPT_PASSTHRU_ARGV(0, "include", &state.git_apply_opts, N_("path"),
			N_("pass it through git-apply"),
			0),
		OPT_PASSTHRU_ARGV('C', NULL, &state.git_apply_opts, N_("n"),
			N_("pass it through git-apply"),
			0),
		OPT_PASSTHRU_ARGV('p', NULL, &state.git_apply_opts, N_("num"),
			N_("pass it through git-apply"),
			0),
2222 2223 2224
		OPT_CALLBACK(0, "patch-format", &patch_format, N_("format"),
			N_("format the patch(es) are in"),
			parse_opt_patchformat),
2225 2226 2227
		OPT_PASSTHRU_ARGV(0, "reject", &state.git_apply_opts, NULL,
			N_("pass it through git-apply"),
			PARSE_OPT_NOARG),
2228 2229
		OPT_STRING(0, "resolvemsg", &state.resolvemsg, NULL,
			N_("override error message when patch failure occurs")),
2230 2231 2232 2233 2234 2235
		OPT_CMDMODE(0, "continue", &resume,
			N_("continue applying patches after resolving a conflict"),
			RESUME_RESOLVED),
		OPT_CMDMODE('r', "resolved", &resume,
			N_("synonyms for --continue"),
			RESUME_RESOLVED),
P
Paul Tan 已提交
2236 2237 2238
		OPT_CMDMODE(0, "skip", &resume,
			N_("skip the current patch"),
			RESUME_SKIP),
P
Paul Tan 已提交
2239 2240 2241
		OPT_CMDMODE(0, "abort", &resume,
			N_("restore the original branch and abort the patching operation."),
			RESUME_ABORT),
N
Nguyễn Thái Ngọc Duy 已提交
2242 2243 2244
		OPT_CMDMODE(0, "quit", &resume,
			N_("abort the patching operation but keep HEAD where it is."),
			RESUME_QUIT),
2245 2246 2247
		OPT_CMDMODE(0, "show-current-patch", &resume,
			N_("show the patch being applied."),
			RESUME_SHOW_PATCH),
2248 2249 2250
		OPT_BOOL(0, "committer-date-is-author-date",
			&state.committer_date_is_author_date,
			N_("lie about committer date")),
P
Paul Tan 已提交
2251 2252
		OPT_BOOL(0, "ignore-date", &state.ignore_date,
			N_("use current timestamp for author date")),
P
Paul Tan 已提交
2253
		OPT_RERERE_AUTOUPDATE(&state.allow_rerere_autoupdate),
2254 2255 2256
		{ OPTION_STRING, 'S', "gpg-sign", &state.sign_commit, N_("key-id"),
		  N_("GPG-sign commits"),
		  PARSE_OPT_OPTARG, NULL, (intptr_t) "" },
P
Paul Tan 已提交
2257 2258
		OPT_HIDDEN_BOOL(0, "rebasing", &state.rebasing,
			N_("(internal use for git-rebase)")),
2259 2260
		OPT_END()
	};
2261

J
Jeff King 已提交
2262 2263 2264
	if (argc == 2 && !strcmp(argv[1], "-h"))
		usage_with_options(usage, options);

2265
	git_config(git_am_config, NULL);
2266

2267
	am_state_init(&state);
2268

2269 2270 2271 2272
	in_progress = am_in_progress(&state);
	if (in_progress)
		am_load(&state);

2273 2274
	argc = parse_options(argc, argv, prefix, options, usage, 0);

2275 2276 2277 2278
	if (binary >= 0)
		fprintf_ln(stderr, _("The -b/--binary option has been a no-op for long time, and\n"
				"it will be removed. Please do not use it anymore."));

2279 2280 2281
	/* Ensure a valid committer ident can be constructed */
	git_committer_info(IDENT_STRICT);

2282
	if (repo_read_index_preload(the_repository, NULL, 0) < 0)
2283 2284
		die(_("failed to read the index"));

2285
	if (in_progress) {
2286 2287 2288 2289 2290 2291 2292 2293 2294 2295 2296 2297 2298 2299 2300
		/*
		 * Catch user error to feed us patches when there is a session
		 * in progress:
		 *
		 * 1. mbox path(s) are provided on the command-line.
		 * 2. stdin is not a tty: the user is trying to feed us a patch
		 *    from standard input. This is somewhat unreliable -- stdin
		 *    could be /dev/null for example and the caller did not
		 *    intend to feed us a patch but wanted to continue
		 *    unattended.
		 */
		if (argc || (resume == RESUME_FALSE && !isatty(0)))
			die(_("previous rebase directory %s still exists but mbox given."),
				state.dir);

2301 2302
		if (resume == RESUME_FALSE)
			resume = RESUME_APPLY;
2303 2304 2305

		if (state.signoff == SIGNOFF_EXPLICIT)
			am_append_signoff(&state);
2306
	} else {
2307 2308 2309
		struct argv_array paths = ARGV_ARRAY_INIT;
		int i;

2310 2311 2312 2313 2314 2315
		/*
		 * Handle stray state directory in the independent-run case. In
		 * the --rebasing case, it is up to the caller to take care of
		 * stray directories.
		 */
		if (file_exists(state.dir) && !state.rebasing) {
N
Nguyễn Thái Ngọc Duy 已提交
2316
			if (resume == RESUME_ABORT || resume == RESUME_QUIT) {
2317 2318 2319 2320 2321 2322 2323 2324 2325 2326
				am_destroy(&state);
				am_state_release(&state);
				return 0;
			}

			die(_("Stray %s directory found.\n"
				"Use \"git am --abort\" to remove it."),
				state.dir);
		}

2327 2328 2329
		if (resume)
			die(_("Resolve operation not in progress, we are not resuming."));

2330 2331 2332 2333 2334 2335 2336
		for (i = 0; i < argc; i++) {
			if (is_absolute_path(argv[i]) || !prefix)
				argv_array_push(&paths, argv[i]);
			else
				argv_array_push(&paths, mkpath("%s/%s", prefix, argv[i]));
		}

2337
		am_setup(&state, patch_format, paths.argv, keep_cr);
2338 2339 2340

		argv_array_clear(&paths);
	}
2341

2342 2343
	switch (resume) {
	case RESUME_FALSE:
2344 2345 2346 2347
		am_run(&state, 0);
		break;
	case RESUME_APPLY:
		am_run(&state, 1);
2348 2349 2350 2351
		break;
	case RESUME_RESOLVED:
		am_resolve(&state);
		break;
P
Paul Tan 已提交
2352 2353 2354
	case RESUME_SKIP:
		am_skip(&state);
		break;
P
Paul Tan 已提交
2355 2356 2357
	case RESUME_ABORT:
		am_abort(&state);
		break;
N
Nguyễn Thái Ngọc Duy 已提交
2358 2359 2360 2361
	case RESUME_QUIT:
		am_rerere_clear();
		am_destroy(&state);
		break;
2362 2363 2364
	case RESUME_SHOW_PATCH:
		ret = show_patch(&state);
		break;
2365
	default:
2366
		BUG("invalid resume value");
2367
	}
2368 2369 2370

	am_state_release(&state);

2371
	return ret;
2372
}