Paul Martin
2016-04-06 b00f45e9492d1cc17bc214daab9988c640741621
src/main/java/com/gitblit/wicket/pages/CommitDiffPage.java
@@ -15,66 +15,58 @@
 */
package com.gitblit.wicket.pages;
import java.text.MessageFormat;
import java.io.OutputStream;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import org.apache.wicket.PageParameters;
import org.apache.wicket.markup.html.basic.Label;
import org.apache.wicket.markup.html.link.BookmarkablePageLink;
import org.apache.wicket.markup.html.link.ExternalLink;
import org.apache.wicket.markup.html.link.Link;
import org.apache.wicket.markup.repeater.Item;
import org.apache.wicket.markup.repeater.data.DataView;
import org.apache.wicket.markup.repeater.data.ListDataProvider;
import org.apache.wicket.request.target.resource.ResourceStreamRequestTarget;
import org.apache.wicket.util.resource.AbstractResourceStreamWriter;
import org.apache.wicket.util.resource.IResourceStream;
import org.eclipse.jgit.diff.DiffEntry.ChangeType;
import org.eclipse.jgit.lib.Repository;
import org.eclipse.jgit.revwalk.RevCommit;
import com.gitblit.GitBlit;
import com.gitblit.Constants;
import com.gitblit.Keys;
import com.gitblit.models.GitNote;
import com.gitblit.models.PathModel.PathChangeModel;
import com.gitblit.models.SubmoduleModel;
import com.gitblit.models.UserModel;
import com.gitblit.servlet.RawServlet;
import com.gitblit.utils.DiffUtils;
import com.gitblit.utils.DiffUtils.DiffComparator;
import com.gitblit.utils.DiffUtils.DiffOutput;
import com.gitblit.utils.DiffUtils.DiffOutputType;
import com.gitblit.utils.JGitUtils;
import com.gitblit.wicket.CacheControl;
import com.gitblit.wicket.GitBlitWebSession;
import com.gitblit.wicket.CacheControl.LastModified;
import com.gitblit.wicket.WicketUtils;
import com.gitblit.wicket.panels.CommitHeaderPanel;
import com.gitblit.wicket.panels.CommitLegendPanel;
import com.gitblit.wicket.panels.DiffStatPanel;
import com.gitblit.wicket.panels.AvatarImage;
import com.gitblit.wicket.panels.LinkPanel;
import com.gitblit.wicket.panels.RefsPanel;
@CacheControl(LastModified.BOOT)
public class CommitDiffPage extends RepositoryPage {
   public CommitDiffPage(PageParameters params) {
      super(params);
      Repository r = getRepository();
      DiffOutputType diffType = DiffOutputType.forName(GitBlit.getString(Keys.web.diffStyle,
            DiffOutputType.GITBLIT.name()));
      RevCommit commit = null, otherCommit = null;
      if( objectId.contains("..") )
      {
         String[] parts = objectId.split("\\.\\.");
         commit = getCommit(r, parts[0]);
         otherCommit = getCommit(r, parts[1]);
      }
      else
      {
         commit = getCommit();
      }
      String diff;
      if(otherCommit == null)
      {
         diff = DiffUtils.getCommitDiff(r, commit, diffType);
      }
      else
      {
         diff = DiffUtils.getDiff(r, commit, otherCommit, diffType);
      }
      final Repository r = getRepository();
      final RevCommit commit = getCommit();
      final DiffComparator diffComparator = WicketUtils.getDiffComparator(params);
      List<String> parents = new ArrayList<String>();
      if (commit.getParentCount() > 0) {
@@ -94,36 +86,74 @@
            WicketUtils.newObjectParameter(repositoryName, objectId)));
      add(new BookmarkablePageLink<Void>("commitLink", CommitPage.class,
            WicketUtils.newObjectParameter(repositoryName, objectId)));
      add(new LinkPanel("whitespaceLink", null, getString(diffComparator.getOpposite().getTranslationKey()),
            CommitDiffPage.class, WicketUtils.newDiffParameter(repositoryName, objectId, diffComparator.getOpposite())));
      add(new CommitHeaderPanel("commitHeader", repositoryName, commit));
      final List<String> imageExtensions = app().settings().getStrings(Keys.web.imageExtensions);
      final ImageDiffHandler handler = new ImageDiffHandler(this, repositoryName,
            parents.isEmpty() ? null : parents.get(0), commit.getName(), imageExtensions);
      final int tabLength = app().settings().getInteger(Keys.web.tabLength, 4);
      final DiffOutput diff = DiffUtils.getCommitDiff(r, commit, diffComparator, DiffOutputType.HTML, handler, tabLength);
      if (handler.getImgDiffCount() > 0) {
         addBottomScript("scripts/imgdiff.js"); // Tiny support script for image diffs
      }
      // add commit diffstat
      int insertions = 0;
      int deletions = 0;
      for (PathChangeModel pcm : diff.stat.paths) {
         insertions += pcm.insertions;
         deletions += pcm.deletions;
      }
      add(new DiffStatPanel("diffStat", insertions, deletions));
      addFullText("fullMessage", commit.getFullMessage());
      // git notes
      List<GitNote> notes = JGitUtils.getNotesOnCommit(r, commit);
      ListDataProvider<GitNote> notesDp = new ListDataProvider<GitNote>(notes);
      DataView<GitNote> notesView = new DataView<GitNote>("notes", notesDp) {
         private static final long serialVersionUID = 1L;
         @Override
         public void populateItem(final Item<GitNote> item) {
            GitNote entry = item.getModelObject();
            item.add(new RefsPanel("refName", repositoryName, Arrays.asList(entry.notesRef)));
            item.add(createPersonPanel("authorName", entry.notesRef.getAuthorIdent(),
                  Constants.SearchType.AUTHOR));
            item.add(new AvatarImage("noteAuthorAvatar", entry.notesRef.getAuthorIdent()));
            item.add(WicketUtils.createTimestampLabel("authorDate", entry.notesRef
                  .getAuthorIdent().getWhen(), getTimeZone(), getTimeUtils()));
            item.add(new Label("noteContent", bugtraqProcessor().processPlainCommitMessage(getRepository(), repositoryName,
                  entry.content)).setEscapeModelStrings(false));
         }
      };
      add(notesView.setVisible(notes.size() > 0));
      // changed paths list
      List<PathChangeModel> paths;
      if( otherCommit == null )
      {
         paths = JGitUtils.getFilesInCommit(r, commit);
      }
      else
      {
         paths = JGitUtils.getFilesInCommit(r, otherCommit);
      }
      add(new CommitLegendPanel("commitLegend", paths));
      ListDataProvider<PathChangeModel> pathsDp = new ListDataProvider<PathChangeModel>(paths);
      add(new CommitLegendPanel("commitLegend", diff.stat.paths));
      ListDataProvider<PathChangeModel> pathsDp = new ListDataProvider<PathChangeModel>(diff.stat.paths);
      DataView<PathChangeModel> pathsView = new DataView<PathChangeModel>("changedPath", pathsDp) {
         private static final long serialVersionUID = 1L;
         int counter;
         @Override
         public void populateItem(final Item<PathChangeModel> item) {
            final PathChangeModel entry = item.getModelObject();
            Label changeType = new Label("changeType", "");
            WicketUtils.setChangeTypeCssClass(changeType, entry.changeType);
            setChangeTypeTooltip(changeType, entry.changeType);
            item.add(changeType);
            item.add(new DiffStatPanel("diffStat", entry.insertions, entry.deletions, true));
            item.add(WicketUtils.setHtmlTooltip(new Label("filestore", ""), getString("gb.filestore"))
                           .setVisible(entry.isFilestoreItem()));
            boolean hasSubmodule = false;
            String submodulePath = null;
            if (entry.isTree()) {
               // tree
               item.add(new LinkPanel("pathName", null, entry.path, TreePage.class,
@@ -136,62 +166,144 @@
               submodulePath = submodule.gitblitPath;
               hasSubmodule = submodule.hasSubmodule;
               item.add(new LinkPanel("pathName", "list", entry.path + " @ " +
                     getShortObjectId(submoduleId), TreePage.class,
                     WicketUtils
                           .newPathParameter(submodulePath, submoduleId, "")).setEnabled(hasSubmodule));
               // add relative link
               item.add(new LinkPanel("pathName", "list", entry.path + " @ " + getShortObjectId(submoduleId), "#n" + entry.objectId));
            } else {
               // blob
               item.add(new LinkPanel("pathName", "list", entry.path, BlobPage.class,
                     WicketUtils
                           .newPathParameter(repositoryName, entry.commitId, entry.path)));
               // add relative link
               if (entry.isFilestoreItem()) {
                  item.add(new LinkPanel("pathName", "list", entry.path, new Link<Object>("link", null) {
                     private static final long serialVersionUID = 1L;
                     @Override
                      public void onClick() {
                        IResourceStream resourceStream = new AbstractResourceStreamWriter() {
                           private static final long serialVersionUID = 1L;
                           @Override
                             public void write(OutputStream output) {
                              UserModel user =  GitBlitWebSession.get().getUser();
                               user = user == null ? UserModel.ANONYMOUS : user;
                                 app().filestore().downloadBlob(entry.getFilestoreOid(), user, getRepositoryModel(), output);
                           }
                        };
                         getRequestCycle().setRequestTarget(new ResourceStreamRequestTarget(resourceStream, entry.path));
                      }}));
               }
               else
               {
                  item.add(new LinkPanel("pathName", "list", entry.path, "#n" + entry.objectId));
               }
            }
            // quick links
            if (entry.isSubmodule()) {
               item.add(new ExternalLink("raw", "").setEnabled(false));
               // submodule
               item.add(new ExternalLink("patch", "").setEnabled(false));
               item.add(new BookmarkablePageLink<Void>("view", CommitPage.class, WicketUtils
                     .newObjectParameter(submodulePath, entry.objectId)).setEnabled(hasSubmodule));
               item.add(new ExternalLink("blame", "").setEnabled(false));
               item.add(new BookmarkablePageLink<Void>("history", HistoryPage.class, WicketUtils
                     .newPathParameter(repositoryName, entry.commitId, entry.path)));
                     .newPathParameter(repositoryName, entry.commitId, entry.path))
                     .setEnabled(!entry.changeType.equals(ChangeType.ADD)));
            } else {
               // tree or blob
               item.add(new BookmarkablePageLink<Void>("patch", PatchPage.class, WicketUtils
                     .newPathParameter(repositoryName, entry.commitId, entry.path)));
               item.add(new BookmarkablePageLink<Void>("view", BlobPage.class, WicketUtils
                     .newPathParameter(repositoryName, entry.commitId, entry.path)));
                     .newPathParameter(repositoryName, entry.commitId, entry.path))
                     .setEnabled(!entry.changeType.equals(ChangeType.ADD)
                           && !entry.changeType.equals(ChangeType.DELETE)));
               if (entry.isFilestoreItem()) {
                  item.add(new Link<Object>("view", null) {
                     private static final long serialVersionUID = 1L;
                     @Override
                      public void onClick() {
                          IResourceStream resourceStream = new AbstractResourceStreamWriter() {
                           private static final long serialVersionUID = 1L;
                           @Override
                               public void write(OutputStream output) {
                                   UserModel user =  GitBlitWebSession.get().getUser();
                                user = user == null ? UserModel.ANONYMOUS : user;
                                 app().filestore().downloadBlob(entry.getFilestoreOid(), user, getRepositoryModel(), output);
                               }
                           };
                         getRequestCycle().setRequestTarget(new ResourceStreamRequestTarget(resourceStream, entry.path));
                      }});
                  item.add(new Link<Object>("raw", null) {
                     private static final long serialVersionUID = 1L;
                     @Override
                      public void onClick() {
                          IResourceStream resourceStream = new AbstractResourceStreamWriter() {
                           private static final long serialVersionUID = 1L;
                           @Override
                               public void write(OutputStream output) {
                                   UserModel user =  GitBlitWebSession.get().getUser();
                                user = user == null ? UserModel.ANONYMOUS : user;
                                 app().filestore().downloadBlob(entry.getFilestoreOid(), user, getRepositoryModel(), output);
                               }
                           };
                         getRequestCycle().setRequestTarget(new ResourceStreamRequestTarget(resourceStream, entry.path));
                      }});
               } else {
                  item.add(new BookmarkablePageLink<Void>("view", BlobPage.class, WicketUtils
                        .newPathParameter(repositoryName, entry.commitId, entry.path))
                        .setEnabled(!entry.changeType.equals(ChangeType.DELETE)));
                  item.add(new ExternalLink("raw", RawServlet.asLink(getContextUrl(), repositoryName, entry.commitId, entry.path))
                        .setEnabled(!entry.changeType.equals(ChangeType.DELETE)));
               }
               item.add(new BookmarkablePageLink<Void>("blame", BlamePage.class, WicketUtils
                     .newPathParameter(repositoryName, entry.commitId, entry.path)));
                     .newPathParameter(repositoryName, entry.commitId, entry.path))
                     .setEnabled(!entry.changeType.equals(ChangeType.ADD)
                           && !entry.changeType.equals(ChangeType.DELETE)));
               item.add(new BookmarkablePageLink<Void>("history", HistoryPage.class, WicketUtils
                     .newPathParameter(repositoryName, entry.commitId, entry.path))
                     .setEnabled(!entry.changeType.equals(ChangeType.ADD)));
            }
            WicketUtils.setAlternatingBackground(item, counter);
            counter++;
         }
      };
      add(pathsView);
      add(new Label("diffText", diff).setEscapeModelStrings(false));
      add(new Label("diffText", diff.content).setEscapeModelStrings(false));
   }
   @Override
   protected String getPageName() {
      return getString("gb.commitdiff");
   }
   @Override
   protected boolean isCommitPage() {
      return true;
   }
   @Override
   protected Class<? extends BasePage> getRepoNavPageClass() {
      return LogPage.class;
   }
   private RevCommit getCommit(Repository r, String rev)
   {
      RevCommit otherCommit = JGitUtils.getCommit(r, rev);
      if (otherCommit == null) {
         error(MessageFormat.format(getString("gb.failedToFindCommit"), rev, repositoryName, getPageName()), true);
      }
      return otherCommit;
   }
}