From 98ce17280c49050213ad0ca628964eb174f89497 Mon Sep 17 00:00:00 2001 From: James Moger <james.moger@gitblit.com> Date: Mon, 18 Apr 2011 16:53:23 -0400 Subject: [PATCH] Author and Committer search integration. --- src/com/gitblit/utils/JGitUtils.java | 85 +++++++++++++++++++++++++++++++++++++++--- 1 files changed, 79 insertions(+), 6 deletions(-) diff --git a/src/com/gitblit/utils/JGitUtils.java b/src/com/gitblit/utils/JGitUtils.java index 5590ffe..3b69c22 100644 --- a/src/com/gitblit/utils/JGitUtils.java +++ b/src/com/gitblit/utils/JGitUtils.java @@ -20,6 +20,9 @@ import org.eclipse.jgit.diff.DiffFormatter; import org.eclipse.jgit.diff.RawTextComparator; import org.eclipse.jgit.errors.ConfigInvalidException; +import org.eclipse.jgit.errors.IncorrectObjectTypeException; +import org.eclipse.jgit.errors.MissingObjectException; +import org.eclipse.jgit.errors.StopWalkException; import org.eclipse.jgit.lib.AnyObjectId; import org.eclipse.jgit.lib.Constants; import org.eclipse.jgit.lib.FileMode; @@ -34,6 +37,7 @@ import org.eclipse.jgit.revwalk.RevObject; import org.eclipse.jgit.revwalk.RevTree; import org.eclipse.jgit.revwalk.RevWalk; +import org.eclipse.jgit.revwalk.filter.RevFilter; import org.eclipse.jgit.treewalk.TreeWalk; import org.eclipse.jgit.treewalk.filter.AndTreeFilter; import org.eclipse.jgit.treewalk.filter.PathFilter; @@ -306,7 +310,7 @@ final RevWalk rw = new RevWalk(r); RevCommit parent = rw.parseCommit(commit.getParent(0).getId()); rw.dispose(); - baseTree = parent.getTree(); + baseTree = parent.getTree(); } else { baseTree = baseCommit.getTree(); } @@ -364,7 +368,7 @@ public static String getCommitPatch(Repository r, RevCommit commit, String path) { return getCommitPatch(r, null, commit, path); } - + public static String getCommitPatch(Repository r, RevCommit baseCommit, RevCommit commit, String path) { try { RevTree baseTree; @@ -463,7 +467,7 @@ public static List<RevCommit> getRevLog(Repository r, String objectId, int offset, int maxCount) { return getRevLog(r, objectId, null, offset, maxCount); } - + public static List<RevCommit> getRevLog(Repository r, String objectId, String path, int offset, int maxCount) { List<RevCommit> list = new ArrayList<RevCommit>(); try { @@ -474,9 +478,7 @@ ObjectId object = r.resolve(objectId); walk.markStart(walk.parseCommit(object)); if (!StringUtils.isEmpty(path)) { - TreeFilter filter = AndTreeFilter.create(PathFilterGroup - .createFromStrings(Collections.singleton(path)), - TreeFilter.ANY_DIFF); + TreeFilter filter = AndTreeFilter.create(PathFilterGroup.createFromStrings(Collections.singleton(path)), TreeFilter.ANY_DIFF); walk.setTreeFilter(filter); } Iterable<RevCommit> revlog = walk; @@ -506,6 +508,77 @@ return list; } + public static enum SearchType { + AUTHOR, COMMITTER, COMMIT; + + public static SearchType forName(String name) { + for (SearchType type : values()) { + if (type.name().equalsIgnoreCase(name)) { + return type; + } + } + return null; + } + } + + public static List<RevCommit> searchRevlogs(Repository r, String objectId, String value, final SearchType type, int offset, int maxCount) { + final String lcValue = value.toLowerCase(); + List<RevCommit> list = new ArrayList<RevCommit>(); + try { + if (objectId == null || objectId.trim().length() == 0) { + objectId = Constants.HEAD; + } + RevWalk walk = new RevWalk(r); + walk.setRevFilter(new RevFilter() { + + @Override + public RevFilter clone() { + return this; + } + + @Override + public boolean include(RevWalk walker, RevCommit commit) throws StopWalkException, MissingObjectException, IncorrectObjectTypeException, IOException { + switch (type) { + case AUTHOR: + return (commit.getAuthorIdent().getName().toLowerCase().indexOf(lcValue) > -1) || (commit.getAuthorIdent().getEmailAddress().toLowerCase().indexOf(lcValue) > -1); + case COMMITTER: + return (commit.getCommitterIdent().getName().toLowerCase().indexOf(lcValue) > -1)|| (commit.getCommitterIdent().getEmailAddress().toLowerCase().indexOf(lcValue) > -1); + case COMMIT: + return commit.getFullMessage().toLowerCase().indexOf(lcValue) > -1; + } + return false; + } + + }); + 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); + } + return list; + } + public static List<RefModel> getTags(Repository r, int maxCount) { return getRefs(r, Constants.R_TAGS, maxCount); } -- Gitblit v1.9.1