Merge "Make PullCommand work with Rebase"

4 files changed, 359 insertions, 51 deletions

diff --git a/org.eclipse.jgit.test/tst/org/eclipse/jgit/api/PullCommandTest.java b/org.eclipse.jgit.test/tst/org/eclipse/jgit/api/PullCommandTest.java
index 04f666cc7b..a25afc7c22 100644
--- a/org.eclipse.jgit.test/tst/org/eclipse/jgit/api/PullCommandTest.java
+++ b/org.eclipse.jgit.test/tst/org/eclipse/jgit/api/PullCommandTest.java
@@ -56,6 +56,7 @@ import java.io.IOException;
 import org.eclipse.jgit.api.CreateBranchCommand.SetupUpstreamMode;
 import org.eclipse.jgit.api.MergeResult.MergeStatus;
 import org.eclipse.jgit.lib.Constants;
+import org.eclipse.jgit.lib.RepositoryState;
 import org.eclipse.jgit.lib.RepositoryTestCase;
 import org.eclipse.jgit.lib.StoredConfig;
 import org.eclipse.jgit.storage.file.FileRepository;
@@ -98,6 +99,12 @@ public class PullCommandTest extends RepositoryTestCase {
 		assertEquals(res.getMergeResult().getMergeStatus(),
 				MergeStatus.FAST_FORWARD);
 		assertFileContentsEqual(targetFile, "Another change");
+		assertEquals(RepositoryState.SAFE, target.getRepository()
+				.getRepositoryState());
+
+		res = target.pull().call();
+		assertEquals(res.getMergeResult().getMergeStatus(),
+				MergeStatus.ALREADY_UP_TO_DATE);
 	}
 
 	@Test
@@ -132,6 +139,8 @@ public class PullCommandTest extends RepositoryTestCase {
 		String result = "<<<<<<< HEAD\nTarget change\n=======\n"
 				+ sourceChangeString + "\n";
 		assertFileContentsEqual(targetFile, result);
+		assertEquals(RepositoryState.MERGING, target.getRepository()
+				.getRepositoryState());
 	}
 
 	@Test
@@ -173,6 +182,8 @@ public class PullCommandTest extends RepositoryTestCase {
 		String result = "<<<<<<< HEAD\nSlave change\n=======\n"
 				+ sourceChangeString + "\n";
 		assertFileContentsEqual(targetFile, result);
+		assertEquals(RepositoryState.MERGING, target.getRepository()
+				.getRepositoryState());
 	}
 
 	@Override
diff --git a/org.eclipse.jgit.test/tst/org/eclipse/jgit/api/PullCommandWithRebaseTest.java b/org.eclipse.jgit.test/tst/org/eclipse/jgit/api/PullCommandWithRebaseTest.java
new file mode 100644
index 0000000000..2847bbe1f0
--- /dev/null
+++ b/org.eclipse.jgit.test/tst/org/eclipse/jgit/api/PullCommandWithRebaseTest.java
@@ -0,0 +1,254 @@
+/*
+ * Copyright (C) 2011, Mathias Kinzler <mathias.kinzler@sap.com>
+ * and other copyright owners as documented in the project's IP log.
+ *
+ * This program and the accompanying materials are made available
+ * under the terms of the Eclipse Distribution License v1.0 which
+ * accompanies this distribution, is reproduced below, and is
+ * available at http://www.eclipse.org/org/documents/edl-v10.php
+ *
+ * All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or
+ * without modification, are permitted provided that the following
+ * conditions are met:
+ *
+ * - Redistributions of source code must retain the above copyright
+ *   notice, this list of conditions and the following disclaimer.
+ *
+ * - Redistributions in binary form must reproduce the above
+ *   copyright notice, this list of conditions and the following
+ *   disclaimer in the documentation and/or other materials provided
+ *   with the distribution.
+ *
+ * - Neither the name of the Eclipse Foundation, Inc. nor the
+ *   names of its contributors may be used to endorse or promote
+ *   products derived from this software without specific prior
+ *   written permission.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND
+ * CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES,
+ * INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
+ * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
+ * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR
+ * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
+ * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
+ * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
+ * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
+ * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
+ * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
+ * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
+ * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+ */
+package org.eclipse.jgit.api;
+
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertFalse;
+import static org.junit.Assert.assertNull;
+import static org.junit.Assert.assertTrue;
+
+import java.io.ByteArrayOutputStream;
+import java.io.File;
+import java.io.FileInputStream;
+import java.io.FileOutputStream;
+import java.io.IOException;
+
+import org.eclipse.jgit.api.CreateBranchCommand.SetupUpstreamMode;
+import org.eclipse.jgit.api.RebaseResult.Status;
+import org.eclipse.jgit.lib.Constants;
+import org.eclipse.jgit.lib.RepositoryState;
+import org.eclipse.jgit.lib.RepositoryTestCase;
+import org.eclipse.jgit.lib.StoredConfig;
+import org.eclipse.jgit.storage.file.FileRepository;
+import org.eclipse.jgit.transport.RefSpec;
+import org.eclipse.jgit.transport.RemoteConfig;
+import org.eclipse.jgit.transport.URIish;
+import org.junit.Before;
+import org.junit.Test;
+
+public class PullCommandWithRebaseTest extends RepositoryTestCase {
+	/** Second Test repository */
+	protected FileRepository dbTarget;
+
+	private Git source;
+
+	private Git target;
+
+	private File sourceFile;
+
+	private File targetFile;
+
+	@Test
+	public void testPullFastForward() throws Exception {
+		PullResult res = target.pull().call();
+		// nothing to update since we don't have different data yet
+		assertTrue(res.getFetchResult().getTrackingRefUpdates().isEmpty());
+		assertEquals(Status.UP_TO_DATE, res.getRebaseResult().getStatus());
+
+		assertFileContentsEqual(targetFile, "Hello world");
+
+		// change the source file
+		writeToFile(sourceFile, "Another change");
+		source.add().addFilepattern("SomeFile.txt").call();
+		source.commit().setMessage("Some change in remote").call();
+
+		res = target.pull().call();
+
+		assertFalse(res.getFetchResult().getTrackingRefUpdates().isEmpty());
+		assertEquals(Status.FAST_FORWARD, res.getRebaseResult().getStatus());
+		assertFileContentsEqual(targetFile, "Another change");
+		assertEquals(RepositoryState.SAFE, target.getRepository()
+				.getRepositoryState());
+
+		res = target.pull().call();
+		assertEquals(Status.UP_TO_DATE, res.getRebaseResult().getStatus());
+	}
+
+	@Test
+	public void testPullConflict() throws Exception {
+		PullResult res = target.pull().call();
+		// nothing to update since we don't have different data yet
+		assertTrue(res.getFetchResult().getTrackingRefUpdates().isEmpty());
+		assertTrue(res.getRebaseResult().getStatus().equals(Status.UP_TO_DATE));
+
+		assertFileContentsEqual(targetFile, "Hello world");
+
+		// change the source file
+		writeToFile(sourceFile, "Source change");
+		source.add().addFilepattern("SomeFile.txt").call();
+		source.commit().setMessage("Source change in remote").call();
+
+		// change the target file
+		writeToFile(targetFile, "Target change");
+		target.add().addFilepattern("SomeFile.txt").call();
+		target.commit().setMessage("Target change in local").call();
+
+		res = target.pull().call();
+
+		assertFalse(res.getFetchResult().getTrackingRefUpdates().isEmpty());
+		assertTrue(res.getRebaseResult().getStatus().equals(Status.STOPPED));
+		String result = "<<<<<<< OURS\nSource change\n=======\nTarget change\n>>>>>>> THEIRS\n";
+		assertFileContentsEqual(targetFile, result);
+		assertEquals(RepositoryState.REBASING_INTERACTIVE, target
+				.getRepository().getRepositoryState());
+	}
+
+	@Test
+	public void testPullLocalConflict() throws Exception {
+		target.branchCreate().setName("basedOnMaster").setStartPoint(
+				"refs/heads/master").setUpstreamMode(SetupUpstreamMode.NOTRACK)
+				.call();
+		StoredConfig config = target.getRepository().getConfig();
+		config.setString("branch", "basedOnMaster", "remote", ".");
+		config.setString("branch", "basedOnMaster", "rebase",
+				"refs/heads/master");
+		config.save();
+		target.getRepository().updateRef(Constants.HEAD).link(
+				"refs/heads/basedOnMaster");
+		PullResult res = target.pull().call();
+		// nothing to update since we don't have different data yet
+		assertNull(res.getFetchResult());
+		assertEquals(Status.UP_TO_DATE, res.getRebaseResult().getStatus());
+
+		assertFileContentsEqual(targetFile, "Hello world");
+
+		// change the file in master
+		target.getRepository().updateRef(Constants.HEAD).link(
+				"refs/heads/master");
+		writeToFile(targetFile, "Master change");
+		target.add().addFilepattern("SomeFile.txt").call();
+		target.commit().setMessage("Source change in master").call();
+
+		// change the file in slave
+		target.getRepository().updateRef(Constants.HEAD).link(
+				"refs/heads/basedOnMaster");
+		writeToFile(targetFile, "Slave change");
+		target.add().addFilepattern("SomeFile.txt").call();
+		target.commit().setMessage("Source change in based on master").call();
+
+		res = target.pull().call();
+
+		assertNull(res.getFetchResult());
+		assertEquals(Status.STOPPED, res.getRebaseResult().getStatus());
+		String result = "<<<<<<< OURS\nMaster change\n=======\nSlave change\n>>>>>>> THEIRS\n";
+		assertFileContentsEqual(targetFile, result);
+		assertEquals(RepositoryState.REBASING_INTERACTIVE, target
+				.getRepository().getRepositoryState());
+	}
+
+	@Override
+	@Before
+	public void setUp() throws Exception {
+		super.setUp();
+		dbTarget = createWorkRepository();
+		source = new Git(db);
+		target = new Git(dbTarget);
+
+		// put some file in the source repo
+		sourceFile = new File(db.getWorkTree(), "SomeFile.txt");
+		writeToFile(sourceFile, "Hello world");
+		// and commit it
+		source.add().addFilepattern("SomeFile.txt").call();
+		source.commit().setMessage("Initial commit for source").call();
+
+		// configure the target repo to connect to the source via "origin"
+		StoredConfig targetConfig = dbTarget.getConfig();
+		targetConfig.setString("branch", "master", "remote", "origin");
+		targetConfig
+				.setString("branch", "master", "merge", "refs/heads/master");
+		RemoteConfig config = new RemoteConfig(targetConfig, "origin");
+
+		config
+				.addURI(new URIish(source.getRepository().getWorkTree()
+						.getPath()));
+		config.addFetchRefSpec(new RefSpec(
+				"+refs/heads/*:refs/remotes/origin/*"));
+		config.update(targetConfig);
+		targetConfig.save();
+
+		targetFile = new File(dbTarget.getWorkTree(), "SomeFile.txt");
+		// make sure we have the same content
+		target.pull().call();
+		target.checkout().setStartPoint("refs/remotes/origin/master").setName(
+				"master").call();
+
+		targetConfig.setString("branch", "master", "rebase",
+				"refs/remotes/origin/master");
+		targetConfig.unset("branch", "master", "merge");
+		targetConfig.save();
+
+		assertFileContentsEqual(targetFile, "Hello world");
+	}
+
+	private void writeToFile(File actFile, String string) throws IOException {
+		FileOutputStream fos = null;
+		try {
+			fos = new FileOutputStream(actFile);
+			fos.write(string.getBytes("UTF-8"));
+			fos.close();
+		} finally {
+			if (fos != null)
+				fos.close();
+		}
+	}
+
+	private void assertFileContentsEqual(File actFile, String string)
+			throws IOException {
+		ByteArrayOutputStream bos = new ByteArrayOutputStream();
+		FileInputStream fis = null;
+		byte[] buffer = new byte[100];
+		try {
+			fis = new FileInputStream(actFile);
+			int read = fis.read(buffer);
+			while (read > 0) {
+				bos.write(buffer, 0, read);
+				read = fis.read(buffer);
+			}
+			String content = new String(bos.toByteArray(), "UTF-8");
+			assertEquals(string, content);
+		} finally {
+			if (fis != null)
+				fis.close();
+		}
+	}
+}
diff --git a/org.eclipse.jgit/src/org/eclipse/jgit/api/PullCommand.java b/org.eclipse.jgit/src/org/eclipse/jgit/api/PullCommand.java
index adcea90f56..73dc452ae2 100644
--- a/org.eclipse.jgit/src/org/eclipse/jgit/api/PullCommand.java
+++ b/org.eclipse.jgit/src/org/eclipse/jgit/api/PullCommand.java
@@ -47,16 +47,19 @@ import java.io.IOException;
 import java.text.MessageFormat;
 
 import org.eclipse.jgit.JGitText;
+import org.eclipse.jgit.api.RebaseCommand.Operation;
 import org.eclipse.jgit.api.errors.CanceledException;
 import org.eclipse.jgit.api.errors.CheckoutConflictException;
 import org.eclipse.jgit.api.errors.ConcurrentRefUpdateException;
 import org.eclipse.jgit.api.errors.DetachedHeadException;
+import org.eclipse.jgit.api.errors.GitAPIException;
 import org.eclipse.jgit.api.errors.InvalidConfigurationException;
 import org.eclipse.jgit.api.errors.InvalidMergeHeadsException;
 import org.eclipse.jgit.api.errors.InvalidRemoteException;
 import org.eclipse.jgit.api.errors.JGitInternalException;
 import org.eclipse.jgit.api.errors.NoHeadException;
 import org.eclipse.jgit.api.errors.NoMessageException;
+import org.eclipse.jgit.api.errors.RefNotFoundException;
 import org.eclipse.jgit.api.errors.WrongRepositoryStateException;
 import org.eclipse.jgit.lib.AnyObjectId;
 import org.eclipse.jgit.lib.Config;
@@ -163,15 +166,14 @@ public class PullCommand extends GitCommand<PullResult> {
 		String remoteBranchName = repoConfig.getString(
 				ConfigConstants.CONFIG_BRANCH_SECTION, branchName,
 				ConfigConstants.CONFIG_KEY_MERGE);
+		boolean doRebase = false;
 		if (remoteBranchName == null) {
 			// check if the branch is configured for pull-rebase
 			remoteBranchName = repoConfig.getString(
 					ConfigConstants.CONFIG_BRANCH_SECTION, branchName,
 					ConfigConstants.CONFIG_KEY_REBASE);
 			if (remoteBranchName != null) {
-				// TODO implement pull-rebase
-				throw new JGitInternalException(
-						"Pull with rebase is not yet supported");
+				doRebase = true;
 			}
 		}
 
@@ -214,60 +216,81 @@ public class PullCommand extends GitCommand<PullResult> {
 
 		monitor.update(1);
 
-		// we check the updates to see which of the updated branches corresponds
-		// to the remote branch name
+		if (monitor.isCancelled())
+			throw new CanceledException(MessageFormat.format(
+					JGitText.get().operationCanceled,
+					JGitText.get().pullTaskName));
 
-		AnyObjectId commitToMerge;
+		PullResult result;
+		if (doRebase) {
+			RebaseCommand rebase = new RebaseCommand(repo);
+			try {
+				RebaseResult rebaseRes = rebase.setUpstream(remoteBranchName)
+						.setProgressMonitor(monitor).setOperation(
+								Operation.BEGIN).call();
+				result = new PullResult(fetchRes, remote, rebaseRes);
+			} catch (NoHeadException e) {
+				throw new JGitInternalException(e.getMessage(), e);
+			} catch (RefNotFoundException e) {
+				throw new JGitInternalException(e.getMessage(), e);
+			} catch (JGitInternalException e) {
+				throw new JGitInternalException(e.getMessage(), e);
+			} catch (GitAPIException e) {
+				throw new JGitInternalException(e.getMessage(), e);
+			}
+		} else {
+			// we check the updates to see which of the updated branches
+			// corresponds
+			// to the remote branch name
+			AnyObjectId commitToMerge;
 
-		if (isRemote) {
-			Ref r = null;
-			if (fetchRes != null) {
-				r = fetchRes.getAdvertisedRef(remoteBranchName);
+			if (isRemote) {
+				Ref r = null;
+				if (fetchRes != null) {
+					r = fetchRes.getAdvertisedRef(remoteBranchName);
+					if (r == null)
+						r = fetchRes.getAdvertisedRef(Constants.R_HEADS
+								+ remoteBranchName);
+				}
 				if (r == null)
-					r = fetchRes.getAdvertisedRef(Constants.R_HEADS
-							+ remoteBranchName);
+					throw new JGitInternalException(MessageFormat.format(
+							JGitText.get().couldNotGetAdvertisedRef,
+							remoteBranchName));
+				else
+					commitToMerge = r.getObjectId();
+			} else {
+				try {
+					commitToMerge = repo.resolve(remoteBranchName);
+				} catch (IOException e) {
+					throw new JGitInternalException(
+							JGitText.get().exceptionCaughtDuringExecutionOfPullCommand,
+							e);
+				}
 			}
-			if (r == null)
-				throw new JGitInternalException(MessageFormat.format(JGitText
-						.get().couldNotGetAdvertisedRef, remoteBranchName));
-			else
-				commitToMerge = r.getObjectId();
-		} else {
+			MergeCommand merge = new MergeCommand(repo);
+			merge.include(
+					"branch \'" + remoteBranchName + "\' of " + remoteUri,
+					commitToMerge);
+			MergeResult mergeRes;
 			try {
-				commitToMerge = repo.resolve(remoteBranchName);
-			} catch (IOException e) {
-				throw new JGitInternalException(
-						JGitText.get().exceptionCaughtDuringExecutionOfPullCommand,
-						e);
+				mergeRes = merge.call();
+				monitor.update(1);
+				result = new PullResult(fetchRes, remote, mergeRes);
+			} catch (NoHeadException e) {
+				throw new JGitInternalException(e.getMessage(), e);
+			} catch (ConcurrentRefUpdateException e) {
+				throw new JGitInternalException(e.getMessage(), e);
+			} catch (CheckoutConflictException e) {
+				throw new JGitInternalException(e.getMessage(), e);
+			} catch (InvalidMergeHeadsException e) {
+				throw new JGitInternalException(e.getMessage(), e);
+			} catch (WrongRepositoryStateException e) {
+				throw new JGitInternalException(e.getMessage(), e);
+			} catch (NoMessageException e) {
+				throw new JGitInternalException(e.getMessage(), e);
 			}
 		}
-
-		if (monitor.isCancelled())
-			throw new CanceledException(MessageFormat.format(
-					JGitText.get().operationCanceled,
-					JGitText.get().pullTaskName));
-
-		MergeCommand merge = new MergeCommand(repo);
-		merge.include("branch \'" + remoteBranchName + "\' of " + remoteUri,
-				commitToMerge);
-		MergeResult mergeRes;
-		try {
-			mergeRes = merge.call();
-			monitor.update(1);
-		} catch (NoHeadException e) {
-			throw new JGitInternalException(e.getMessage(), e);
-		} catch (ConcurrentRefUpdateException e) {
-			throw new JGitInternalException(e.getMessage(), e);
-		} catch (CheckoutConflictException e) {
-			throw new JGitInternalException(e.getMessage(), e);
-		} catch (InvalidMergeHeadsException e) {
-			throw new JGitInternalException(e.getMessage(), e);
-		} catch (WrongRepositoryStateException e) {
-			throw new JGitInternalException(e.getMessage(), e);
-		} catch (NoMessageException e) {
-			throw new JGitInternalException(e.getMessage(), e);
-		}
 		monitor.endTask();
-		return new PullResult(fetchRes, remote, mergeRes);
+		return result;
 	}
 }
diff --git a/org.eclipse.jgit/src/org/eclipse/jgit/api/PullResult.java b/org.eclipse.jgit/src/org/eclipse/jgit/api/PullResult.java
index 105b76f811..40ed137ee0 100644
--- a/org.eclipse.jgit/src/org/eclipse/jgit/api/PullResult.java
+++ b/org.eclipse.jgit/src/org/eclipse/jgit/api/PullResult.java
@@ -52,6 +52,8 @@ public class PullResult {
 
 	private final MergeResult mergeResult;
 
+	private final RebaseResult rebaseResult;
+
 	private final String fetchedFrom;
 
 	PullResult(FetchResult fetchResult, String fetchedFrom,
@@ -59,6 +61,15 @@ public class PullResult {
 		this.fetchResult = fetchResult;
 		this.fetchedFrom = fetchedFrom;
 		this.mergeResult = mergeResult;
+		this.rebaseResult = null;
+	}
+
+	PullResult(FetchResult fetchResult, String fetchedFrom,
+			RebaseResult rebaseResult) {
+		this.fetchResult = fetchResult;
+		this.fetchedFrom = fetchedFrom;
+		this.mergeResult = null;
+		this.rebaseResult = rebaseResult;
 	}
 
 	/**
@@ -76,6 +87,13 @@ public class PullResult {
 	}
 
 	/**
+	 * @return the rebase result, or <code>null</code>
+	 */
+	public RebaseResult getRebaseResult() {
+		return this.rebaseResult;
+	}
+
+	/**
 	 * @return the name of the remote configuration from which fetch was tried,
 	 *         or <code>null</code>
 	 */
@@ -93,8 +111,10 @@ public class PullResult {
 		sb.append("\n");
 		if (mergeResult != null)
 			sb.append(mergeResult.toString());
+		else if (rebaseResult != null)
+			sb.append(rebaseResult.toString());
 		else
-			sb.append("No merge result");
+			sb.append("No update result");
 		return sb.toString();
 	}
 }