From be0fb213529d9d571078c59e534dbbd6b6cfc46c Mon Sep 17 00:00:00 2001
From: James Moger <james.moger@gitblit.com>
Date: Tue, 28 Jun 2011 22:12:35 -0400
Subject: [PATCH] Use Sitename on login page.

---
 src/com/gitblit/utils/DiffUtils.java |   74 ++++++++++++++++++++++++++++--------
 1 files changed, 57 insertions(+), 17 deletions(-)

diff --git a/src/com/gitblit/utils/DiffUtils.java b/src/com/gitblit/utils/DiffUtils.java
index 969cb91..c1401f9 100644
--- a/src/com/gitblit/utils/DiffUtils.java
+++ b/src/com/gitblit/utils/DiffUtils.java
@@ -16,11 +16,16 @@
 package com.gitblit.utils;
 
 import java.io.ByteArrayOutputStream;
+import java.util.ArrayList;
 import java.util.List;
 
+import org.eclipse.jgit.api.BlameCommand;
+import org.eclipse.jgit.blame.BlameResult;
 import org.eclipse.jgit.diff.DiffEntry;
 import org.eclipse.jgit.diff.DiffFormatter;
+import org.eclipse.jgit.diff.RawText;
 import org.eclipse.jgit.diff.RawTextComparator;
+import org.eclipse.jgit.lib.Constants;
 import org.eclipse.jgit.lib.Repository;
 import org.eclipse.jgit.revwalk.RevCommit;
 import org.eclipse.jgit.revwalk.RevTree;
@@ -30,11 +35,24 @@
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
-import com.gitblit.utils.JGitUtils.DiffOutputType;
+import com.gitblit.models.AnnotatedLine;
 
 public class DiffUtils {
 
 	private static final Logger LOGGER = LoggerFactory.getLogger(DiffUtils.class);
+
+	public static enum DiffOutputType {
+		PLAIN, GITWEB, GITBLIT;
+
+		public static DiffOutputType forName(String name) {
+			for (DiffOutputType type : values()) {
+				if (type.name().equalsIgnoreCase(name)) {
+					return type;
+				}
+			}
+			return null;
+		}
+	}
 
 	public static String getCommitDiff(Repository r, RevCommit commit, DiffOutputType outputType) {
 		return getDiff(r, null, commit, null, outputType);
@@ -52,6 +70,7 @@
 
 	public static String getDiff(Repository r, RevCommit baseCommit, RevCommit commit, String path,
 			DiffOutputType outputType) {
+		String diff = null;
 		try {
 			RevTree baseTree;
 			if (baseCommit == null) {
@@ -90,18 +109,17 @@
 			df.setRepository(r);
 			df.setDiffComparator(cmp);
 			df.setDetectRenames(true);
-			List<DiffEntry> diffs = df.scan(baseTree, commitTree);
+			List<DiffEntry> diffEntries = df.scan(baseTree, commitTree);
 			if (path != null && path.length() > 0) {
-				for (DiffEntry diff : diffs) {
-					if (diff.getNewPath().equalsIgnoreCase(path)) {
-						df.format(diff);
+				for (DiffEntry diffEntry : diffEntries) {
+					if (diffEntry.getNewPath().equalsIgnoreCase(path)) {
+						df.format(diffEntry);
 						break;
 					}
 				}
 			} else {
-				df.format(diffs);
+				df.format(diffEntries);
 			}
-			String diff;
 			if (df instanceof GitWebDiffFormatter) {
 				// workaround for complex private methods in DiffFormatter
 				diff = ((GitWebDiffFormatter) df).getHtml();
@@ -109,15 +127,15 @@
 				diff = os.toString();
 			}
 			df.flush();
-			return diff;
 		} catch (Throwable t) {
 			LOGGER.error("failed to generate commit diff!", t);
 		}
-		return null;
+		return diff;
 	}
 
 	public static String getCommitPatch(Repository r, RevCommit baseCommit, RevCommit commit,
 			String path) {
+		String diff = null;
 		try {
 			RevTree baseTree;
 			if (baseCommit == null) {
@@ -142,23 +160,45 @@
 			df.setRepository(r);
 			df.setDiffComparator(cmp);
 			df.setDetectRenames(true);
-			List<DiffEntry> diffs = df.scan(baseTree, commitTree);
+			List<DiffEntry> diffEntries = df.scan(baseTree, commitTree);
 			if (path != null && path.length() > 0) {
-				for (DiffEntry diff : diffs) {
-					if (diff.getNewPath().equalsIgnoreCase(path)) {
-						df.format(diff);
+				for (DiffEntry diffEntry : diffEntries) {
+					if (diffEntry.getNewPath().equalsIgnoreCase(path)) {
+						df.format(diffEntry);
 						break;
 					}
 				}
 			} else {
-				df.format(diffs);
+				df.format(diffEntries);
 			}
-			String diff = df.getPatch(commit);
+			diff = df.getPatch(commit);
 			df.flush();
-			return diff;
 		} catch (Throwable t) {
 			LOGGER.error("failed to generate commit diff!", t);
 		}
-		return null;
+		return diff;
+	}
+
+	public static List<AnnotatedLine> blame(Repository r, String blobPath, String objectId) {
+		List<AnnotatedLine> lines = new ArrayList<AnnotatedLine>();
+		try {
+			if (StringUtils.isEmpty(objectId)) {
+				objectId = Constants.HEAD;
+			}
+			BlameCommand blameCommand = new BlameCommand(r);
+			blameCommand.setFilePath(blobPath);
+			blameCommand.setStartCommit(r.resolve(objectId));
+			BlameResult blameResult = blameCommand.call();
+			RawText rawText = blameResult.getResultContents();
+			int length = rawText.size();
+			for (int i = 0; i < length; i++) {
+				RevCommit commit = blameResult.getSourceCommit(i);
+				AnnotatedLine line = new AnnotatedLine(commit, i + 1, rawText.getString(i));
+				lines.add(line);
+			}
+		} catch (Throwable t) {
+			LOGGER.error("failed to generate blame!", t);
+		}
+		return lines;
 	}
 }

--
Gitblit v1.9.1