From fc948cacef9c9b8c0a1e84cbc082ca67cd5f68d9 Mon Sep 17 00:00:00 2001
From: James Moger <james.moger@gitblit.com>
Date: Thu, 14 Apr 2011 18:29:10 -0400
Subject: [PATCH] Authenticate the webapp against the same realm as the git servlet.

---
 src/com/gitblit/utils/JGitUtils.java |  122 +++++++++++++++++++++++++++++++++-------
 1 files changed, 100 insertions(+), 22 deletions(-)

diff --git a/src/com/gitblit/utils/JGitUtils.java b/src/com/gitblit/utils/JGitUtils.java
index 1b418b0..69c46da 100644
--- a/src/com/gitblit/utils/JGitUtils.java
+++ b/src/com/gitblit/utils/JGitUtils.java
@@ -16,7 +16,6 @@
 import java.util.Map;
 import java.util.Set;
 
-import org.eclipse.jgit.api.Git;
 import org.eclipse.jgit.diff.DiffEntry;
 import org.eclipse.jgit.diff.DiffFormatter;
 import org.eclipse.jgit.diff.RawTextComparator;
@@ -43,10 +42,11 @@
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
-import com.gitblit.utils.TicGitTicket.Comment;
 import com.gitblit.wicket.models.Metric;
 import com.gitblit.wicket.models.PathModel;
 import com.gitblit.wicket.models.RefModel;
+import com.gitblit.wicket.models.TicGitTicket;
+import com.gitblit.wicket.models.TicGitTicket.Comment;
 
 public class JGitUtils {
 
@@ -99,12 +99,15 @@
 		return getCommitDate(commit);
 	}
 
-	public static RevCommit getCommit(Repository r, String commitId) {
+	public static RevCommit getCommit(Repository r, String objectId) {
 		RevCommit commit = null;
 		try {
-			ObjectId objectId = r.resolve(commitId);
+			if (objectId == null || objectId.trim().length() == 0) {
+				objectId = Constants.HEAD;
+			}
+			ObjectId object = r.resolve(objectId);
 			RevWalk walk = new RevWalk(r);
-			RevCommit rev = walk.parseCommit(objectId);
+			RevCommit rev = walk.parseCommit(object);
 			commit = rev;
 			walk.dispose();
 		} catch (Throwable t) {
@@ -206,8 +209,8 @@
 		return new String(getRawContent(r, (RevBlob) obj));
 	}
 
-	public static List<PathModel> getFilesInPath(Repository r, String basePath, String commitId) {
-		RevCommit commit = getCommit(r, commitId);
+	public static List<PathModel> getFilesInPath(Repository r, String basePath, String objectId) {
+		RevCommit commit = getCommit(r, objectId);
 		return getFilesInPath(r, basePath, commit);
 	}
 
@@ -286,7 +289,7 @@
 	public static String getCommitDiff(Repository r, RevCommit commit, boolean outputHtml) {
 		return getCommitDiff(r, commit, null, outputHtml);
 	}
-	
+
 	public static String getCommitDiff(Repository r, RevCommit commit, String path, boolean outputHtml) {
 		try {
 			final RevWalk rw = new RevWalk(r);
@@ -299,11 +302,7 @@
 			walk.setRecursive(true);
 			walk.addTree(parentTree);
 			walk.addTree(commitTree);
-			if (path != null && path.trim().length() > 0) {
-				walk.setFilter(PathFilter.create(path));
-			} else {
-				walk.setFilter(TreeFilter.ANY_DIFF);
-			}
+			walk.setFilter(TreeFilter.ANY_DIFF);
 
 			final ByteArrayOutputStream os = new ByteArrayOutputStream();
 			RawTextComparator cmp = RawTextComparator.DEFAULT;
@@ -317,7 +316,16 @@
 			df.setDiffComparator(cmp);
 			df.setDetectRenames(true);
 			List<DiffEntry> diffs = df.scan(parentTree, commitTree);
-			df.format(diffs);
+			if (path != null && path.length() > 0) {
+				for (DiffEntry diff : diffs) {
+					if (diff.getNewPath().equalsIgnoreCase(path)) {
+						df.format(diff);
+						break;
+					}
+				}
+			} else {
+				df.format(diffs);
+			}
 			String diff;
 			if (outputHtml) {
 				// workaround for complex private methods in DiffFormatter
@@ -325,6 +333,50 @@
 			} else {
 				diff = os.toString();
 			}
+			df.flush();
+			return diff;
+		} catch (Throwable t) {
+			LOGGER.error("failed to generate commit diff!", t);
+		}
+		return null;
+	}
+	
+	public static String getCommitPatch(Repository r, RevCommit commit) {
+		return getCommitPatch(r, commit);
+	}
+	
+	public static String getCommitPatch(Repository r, RevCommit commit, String path) {
+		try {
+			final RevWalk rw = new RevWalk(r);
+			RevCommit parent = rw.parseCommit(commit.getParent(0).getId());
+			RevTree parentTree = parent.getTree();
+			RevTree commitTree = commit.getTree();
+
+			final TreeWalk walk = new TreeWalk(r);
+			walk.reset();
+			walk.setRecursive(true);
+			walk.addTree(parentTree);
+			walk.addTree(commitTree);
+			walk.setFilter(TreeFilter.ANY_DIFF);
+
+			final ByteArrayOutputStream os = new ByteArrayOutputStream();
+			RawTextComparator cmp = RawTextComparator.DEFAULT;
+			PatchFormatter df = new PatchFormatter(os);
+			df.setRepository(r);
+			df.setDiffComparator(cmp);
+			df.setDetectRenames(true);
+			List<DiffEntry> diffs = df.scan(parentTree, commitTree);
+			if (path != null && path.length() > 0) {
+				for (DiffEntry diff : diffs) {
+					if (diff.getNewPath().equalsIgnoreCase(path)) {
+						df.format(diff);
+						break;
+					}
+				}
+			} else {
+				df.format(diffs);
+			}
+			String diff = df.getPatch(commit);
 			df.flush();
 			return diff;
 		} catch (Throwable t) {
@@ -380,16 +432,39 @@
 	}
 
 	public static List<RevCommit> getRevLog(Repository r, int maxCount) {
+		return getRevLog(r, Constants.HEAD, 0, maxCount);
+	}
+
+	public static List<RevCommit> getRevLog(Repository r, String objectId, int offset, int maxCount) {
 		List<RevCommit> list = new ArrayList<RevCommit>();
 		try {
-			Git git = new Git(r);
-			Iterable<RevCommit> revlog = git.log().call();
-			for (RevCommit rev : revlog) {
-				list.add(rev);
-				if (maxCount > 0 && list.size() == maxCount) {
-					break;
+			if (objectId == null || objectId.trim().length() == 0) {
+				objectId = Constants.HEAD;
+			}
+			RevWalk walk = new RevWalk(r);
+			ObjectId object = r.resolve(objectId);
+			walk.markStart(walk.parseCommit(object));
+			Iterable<RevCommit> revlog = walk;
+			if (offset > 0) {
+				int count = 0;
+				for (RevCommit rev : revlog) {
+					count++;
+					if (count > offset) {
+						list.add(rev);
+						if (maxCount > 0 && list.size() == maxCount) {
+							break;
+						}
+					}
+				}
+			} else {
+				for (RevCommit rev : revlog) {
+					list.add(rev);
+					if (maxCount > 0 && list.size() == maxCount) {
+						break;
+					}
 				}
 			}
+			walk.dispose();
 		} catch (Throwable t) {
 			LOGGER.error("Failed to determine last change", t);
 		}
@@ -501,11 +576,14 @@
 	}
 
 	public static List<Metric> getDateMetrics(Repository r) {
+		final List<RefModel> tags = getTags(r, -1);
 		final Map<String, Metric> map = new HashMap<String, Metric>();
 		try {
 			DateFormat df = new SimpleDateFormat("yyyy-MM");
-			Git git = new Git(r);
-			Iterable<RevCommit> revlog = git.log().call();
+			RevWalk walk = new RevWalk(r);
+			ObjectId object = r.resolve(Constants.HEAD);
+			walk.markStart(walk.parseCommit(object));
+			Iterable<RevCommit> revlog = walk;
 			for (RevCommit rev : revlog) {
 				Date d = getCommitDate(rev);
 				String p = df.format(d);

--
Gitblit v1.9.1