Support checking out stage with DiscardChangesOperation

For resolving conflicts, we want to be able to do `git checkout --ours`
or `git checkout --theirs` for a path.

Change-Id: Ia6d003993ad1016602ff960203b3526cd884bad2
Signed-off-by: Robin Stocker <robin@nibor.org>
Signed-off-by: Matthias Sohn <matthias.sohn@sap.com>

1 files changed, 43 insertions, 1 deletions

diff --git a/org.eclipse.egit.core/src/org/eclipse/egit/core/op/DiscardChangesOperation.java b/org.eclipse.egit.core/src/org/eclipse/egit/core/op/DiscardChangesOperation.java
index f40e59c44e..b5eeac661b 100644
--- a/org.eclipse.egit.core/src/org/eclipse/egit/core/op/DiscardChangesOperation.java
+++ b/org.eclipse.egit.core/src/org/eclipse/egit/core/op/DiscardChangesOperation.java
@@ -50,6 +50,32 @@ public class DiscardChangesOperation implements IEGitOperation {
 
 	private String revision;
 
+	private Stage stage;
+
+	/**
+	 * The index stage to check out for conflicting files.
+	 */
+	public enum Stage {
+		/**
+		 * "base" stage
+		 */
+		BASE(CheckoutCommand.Stage.BASE),
+		/**
+		 * "ours" stage
+		 */
+		OURS(CheckoutCommand.Stage.OURS),
+		/**
+		 * "theirs" stage
+		 */
+		THEIRS(CheckoutCommand.Stage.THEIRS);
+
+		private CheckoutCommand.Stage checkoutStage;
+
+		private Stage(CheckoutCommand.Stage checkoutStage) {
+			this.checkoutStage = checkoutStage;
+		}
+	}
+
 	/**
 	 * Construct a {@link DiscardChangesOperation} object.
 	 *
@@ -86,6 +112,19 @@ public class DiscardChangesOperation implements IEGitOperation {
 				.keySet());
 	}
 
+	/**
+	 * Set the index stage to check out for conflicting files. Not compatible
+	 * with revision.
+	 *
+	 * @param stage
+	 */
+	public void setStage(Stage stage) {
+		if (revision != null)
+			throw new IllegalStateException(
+					"Either stage or revision can be set, but not both"); //$NON-NLS-1$
+		this.stage = stage;
+	}
+
 	/*
 	 * (non-Javadoc)
 	 *
@@ -151,7 +190,10 @@ public class DiscardChangesOperation implements IEGitOperation {
 			throws GitAPIException {
 		ResourceUtil.saveLocalHistory(repository);
 		CheckoutCommand checkoutCommand = new Git(repository).checkout();
-		checkoutCommand.setStartPoint(this.revision);
+		if (revision != null)
+			checkoutCommand.setStartPoint(revision);
+		if (stage != null)
+			checkoutCommand.setStage(stage.checkoutStage);
 		if (paths.isEmpty() || paths.contains("")) //$NON-NLS-1$
 			checkoutCommand.setAllPaths(true);
 		else {