Add GitAddTask

Change-Id: Ia9a3c9f4728e13d1e62f530b1d843d09afb4eb42
Signed-off-by: Chris Aniszczyk <caniszczyk@gmail.com>

1 files changed, 152 insertions, 0 deletions

diff --git a/org.eclipse.jgit.ant/src/org/eclipse/jgit/ant/tasks/GitAddTask.java b/org.eclipse.jgit.ant/src/org/eclipse/jgit/ant/tasks/GitAddTask.java
new file mode 100644
index 0000000000..ebc48ba160
--- /dev/null
+++ b/org.eclipse.jgit.ant/src/org/eclipse/jgit/ant/tasks/GitAddTask.java
@@ -0,0 +1,152 @@
+/*
+ * Copyright (C) 2011, Ketan Padegaonkar <KetanPadegaonkar@gmail.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.ant.tasks;
+
+import java.io.File;
+import java.io.IOException;
+
+import org.apache.tools.ant.BuildException;
+import org.apache.tools.ant.Project;
+import org.apache.tools.ant.Task;
+import org.apache.tools.ant.types.DirSet;
+import org.apache.tools.ant.types.FileSet;
+import org.apache.tools.ant.types.resources.Union;
+import org.eclipse.jgit.api.AddCommand;
+import org.eclipse.jgit.api.Git;
+import org.eclipse.jgit.lib.Repository;
+import org.eclipse.jgit.lib.RepositoryCache;
+import org.eclipse.jgit.storage.file.FileRepositoryBuilder;
+import org.eclipse.jgit.util.FS;
+
+/**
+ * Adds a file to the git index.
+ *
+ * @see <a href="http://www.kernel.org/pub/software/scm/git/docs/git-add.html"
+ *      >git-add(1)</a>
+ */
+public class GitAddTask extends Task {
+
+	private File src;
+	private Union path;
+
+	/**
+	 * @param src
+	 *            the src to set
+	 */
+	public void setSrc(File src) {
+		this.src = src;
+	}
+
+	/**
+	 * Add a set of files to add.
+	 *
+	 * @param set
+	 *            a set of files to add.
+	 */
+	public void addFileset(FileSet set) {
+		getPath().add(set);
+	}
+
+	/**
+	 * Add a set of files to add.
+	 *
+	 * @param set
+	 *            a set of files to add.
+	 */
+	public void addDirset(DirSet set) {
+		getPath().add(set);
+	}
+
+	private synchronized Union getPath() {
+		if (path == null) {
+			path = new Union();
+			path.setProject(getProject());
+		}
+		return path;
+	}
+
+	@Override
+	public void execute() throws BuildException {
+		if (src == null) {
+			throw new BuildException("Repository path not specified.");
+		}
+		if (!RepositoryCache.FileKey.isGitRepository(new File(src, ".git"),
+				FS.DETECTED)) {
+			throw new BuildException("Specified path (" + src
+					+ ") is not a git repository.");
+		}
+
+		AddCommand gitAdd;
+		try {
+			Repository repo = new FileRepositoryBuilder().readEnvironment()
+					.findGitDir(src).build();
+			gitAdd = new Git(repo).add();
+		} catch (IOException e) {
+			throw new BuildException("Could not access repository " + src, e);
+		}
+
+		try {
+			String prefix = src.getCanonicalPath();
+			String[] allFiles = getPath().list();
+
+			for (String file : allFiles) {
+				String toAdd = translateFilePathUsingPrefix(file, prefix);
+				log("Adding " + toAdd, Project.MSG_VERBOSE);
+				gitAdd.addFilepattern(toAdd);
+			}
+			gitAdd.call();
+		} catch (Exception e) {
+			throw new BuildException("Could not add files to index." + src, e);
+		}
+
+	}
+
+	private String translateFilePathUsingPrefix(String file, String prefix)
+			throws IOException {
+		if (file.equals(prefix)) {
+			return ".";
+		}
+		return new File(file).getCanonicalPath().substring(prefix.length() + 1);
+	}
+
+}