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