| // Copyright 2012 Google Inc. All Rights Reserved. |
| // |
| // Licensed under the Apache License, Version 2.0 (the "License"); |
| // you may not use this file except in compliance with the License. |
| // You may obtain a copy of the License at |
| // |
| // http://www.apache.org/licenses/LICENSE-2.0 |
| // |
| // Unless required by applicable law or agreed to in writing, software |
| // distributed under the License is distributed on an "AS IS" BASIS, |
| // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| // See the License for the specific language governing permissions and |
| // limitations under the License. |
| |
| package com.google.gitiles; |
| |
| import static com.google.common.base.Preconditions.checkNotNull; |
| import static javax.servlet.http.HttpServletResponse.SC_NOT_FOUND; |
| |
| import com.google.common.io.BaseEncoding; |
| import com.google.gitiles.CommitData.Field; |
| import com.google.gitiles.DateFormatter.Format; |
| import java.io.IOException; |
| import java.io.OutputStream; |
| import java.io.Writer; |
| import java.util.Arrays; |
| import java.util.Map; |
| import java.util.Set; |
| import javax.servlet.http.HttpServletRequest; |
| import javax.servlet.http.HttpServletResponse; |
| import org.eclipse.jgit.diff.DiffFormatter; |
| import org.eclipse.jgit.errors.IncorrectObjectTypeException; |
| import org.eclipse.jgit.errors.MissingObjectException; |
| import org.eclipse.jgit.http.server.ServletUtils; |
| import org.eclipse.jgit.lib.FileMode; |
| import org.eclipse.jgit.lib.ObjectId; |
| import org.eclipse.jgit.lib.Repository; |
| import org.eclipse.jgit.revwalk.RevCommit; |
| import org.eclipse.jgit.revwalk.RevWalk; |
| import org.eclipse.jgit.treewalk.AbstractTreeIterator; |
| import org.eclipse.jgit.treewalk.CanonicalTreeParser; |
| import org.eclipse.jgit.treewalk.EmptyTreeIterator; |
| import org.eclipse.jgit.treewalk.TreeWalk; |
| import org.eclipse.jgit.treewalk.filter.PathFilter; |
| |
| /** Serves an HTML page with all the diffs for a commit. */ |
| public class DiffServlet extends BaseServlet { |
| private static final long serialVersionUID = 1L; |
| |
| private final Linkifier linkifier; |
| |
| public DiffServlet(GitilesAccess.Factory accessFactory, Renderer renderer, Linkifier linkifier) { |
| super(renderer, accessFactory); |
| this.linkifier = checkNotNull(linkifier, "linkifier"); |
| } |
| |
| @Override |
| protected void doGetHtml(HttpServletRequest req, HttpServletResponse res) throws IOException { |
| GitilesView view = ViewFilter.getView(req); |
| Repository repo = ServletUtils.getRepository(req); |
| |
| try (RevWalk walk = new RevWalk(repo); |
| TreeWalk tw = newTreeWalk(walk, view)) { |
| boolean showCommit; |
| boolean isFile; |
| AbstractTreeIterator oldTree; |
| AbstractTreeIterator newTree; |
| try { |
| if (tw == null && !view.getPathPart().isEmpty()) { |
| res.setStatus(SC_NOT_FOUND); |
| return; |
| } |
| isFile = tw != null && isFile(tw); |
| |
| // If we are viewing the diff between a commit and one of its parents, |
| // include the commit detail in the rendered page. |
| showCommit = isParentOf(walk, view.getOldRevision(), view.getRevision()); |
| oldTree = getTreeIterator(walk, view.getOldRevision().getId()); |
| newTree = getTreeIterator(walk, view.getRevision().getId()); |
| } catch (MissingObjectException | IncorrectObjectTypeException e) { |
| res.setStatus(SC_NOT_FOUND); |
| return; |
| } |
| |
| Map<String, Object> data = getData(req); |
| data.put("title", "Diff - " + view.getRevisionRange()); |
| if (showCommit) { |
| Set<Field> fs = CommitSoyData.DEFAULT_FIELDS; |
| if (isFile) { |
| fs = Field.setOf(fs, Field.PARENT_BLAME_URL); |
| } |
| GitilesAccess access = getAccess(req); |
| DateFormatter df = new DateFormatter(access, Format.DEFAULT); |
| data.put( |
| "commit", |
| new CommitSoyData() |
| .setLinkifier(linkifier) |
| .setArchiveFormat(getArchiveFormat(access)) |
| .toSoyData(req, walk.parseCommit(view.getRevision().getId()), fs, df)); |
| } |
| if (!data.containsKey("repositoryName") && (view.getRepositoryName() != null)) { |
| data.put("repositoryName", view.getRepositoryName()); |
| } |
| if (!data.containsKey("breadcrumbs")) { |
| data.put("breadcrumbs", view.getBreadcrumbs()); |
| } |
| |
| setCacheHeaders(req, res); |
| try (OutputStream out = startRenderStreamingHtml(req, res, "gitiles.diffDetail", data); |
| DiffFormatter diff = new HtmlDiffFormatter(renderer, view, out)) { |
| formatDiff(repo, oldTree, newTree, view.getPathPart(), diff); |
| } |
| } |
| } |
| |
| @Override |
| protected void doGetText(HttpServletRequest req, HttpServletResponse res) throws IOException { |
| GitilesView view = ViewFilter.getView(req); |
| Repository repo = ServletUtils.getRepository(req); |
| |
| try (RevWalk walk = new RevWalk(repo)) { |
| AbstractTreeIterator oldTree; |
| AbstractTreeIterator newTree; |
| try { |
| oldTree = getTreeIterator(walk, view.getOldRevision().getId()); |
| newTree = getTreeIterator(walk, view.getRevision().getId()); |
| } catch (MissingObjectException | IncorrectObjectTypeException e) { |
| res.setStatus(SC_NOT_FOUND); |
| return; |
| } |
| |
| try (Writer writer = startRenderText(req, res); |
| OutputStream out = BaseEncoding.base64().encodingStream(writer); |
| DiffFormatter diff = new DiffFormatter(out)) { |
| formatDiff(repo, oldTree, newTree, view.getPathPart(), diff); |
| } |
| } |
| } |
| |
| private static TreeWalk newTreeWalk(RevWalk walk, GitilesView view) throws IOException { |
| if (view.getPathPart().isEmpty()) { |
| return null; |
| } |
| return TreeWalk.forPath( |
| walk.getObjectReader(), view.getPathPart(), walk.parseTree(view.getRevision().getId())); |
| } |
| |
| private static boolean isParentOf(RevWalk walk, Revision oldRevision, Revision newRevision) |
| throws MissingObjectException, IncorrectObjectTypeException, IOException { |
| RevCommit newCommit = walk.parseCommit(newRevision.getId()); |
| if (newCommit.getParentCount() > 0) { |
| return Arrays.asList(newCommit.getParents()).contains(oldRevision.getId()); |
| } |
| return oldRevision == Revision.NULL; |
| } |
| |
| private static boolean isFile(TreeWalk tw) { |
| return (tw.getRawMode(0) & FileMode.TYPE_FILE) > 0; |
| } |
| |
| private static void formatDiff( |
| Repository repo, |
| AbstractTreeIterator oldTree, |
| AbstractTreeIterator newTree, |
| String path, |
| DiffFormatter diff) |
| throws IOException { |
| if (!path.isEmpty()) { |
| diff.setPathFilter(PathFilter.create(path)); |
| } |
| diff.setRepository(repo); |
| diff.setDetectRenames(true); |
| diff.format(oldTree, newTree); |
| } |
| |
| private static AbstractTreeIterator getTreeIterator(RevWalk walk, ObjectId id) |
| throws IOException { |
| if (!id.equals(ObjectId.zeroId())) { |
| CanonicalTreeParser p = new CanonicalTreeParser(); |
| p.reset(walk.getObjectReader(), walk.parseTree(id)); |
| return p; |
| } |
| return new EmptyTreeIterator(); |
| } |
| } |