Merge pull request #1027 from gitblit/1018-Filestore-downloads-filename

Paul Martin

2016-03-23 caa660d97f95e4da66c9f7722267145ffe8e6d48

 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,72 @@
            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));

            boolean hasSubmodule = false;
            String submodulePath = null;
				
            if (entry.isTree()) {
               // tree
               item.add(new LinkPanel("pathName", null, entry.path, TreePage.class,
@@ -136,62 +164,151 @@
               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 Label("filestore", getString("gb.filestore")).setVisible(false));
					
               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 Label("filestore", getString("gb.filestore")).setVisible(true));
						
                  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 Label("filestore", getString("gb.filestore")).setVisible(false));
						
                  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;
   }
}