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