James Moger
2012-11-06 798581cab5817310a1b9991dac3b10cd7813f86a
src/com/gitblit/wicket/pages/RawPage.java
@@ -15,84 +15,147 @@
 */
package com.gitblit.wicket.pages;
import java.io.IOException;
import java.util.HashMap;
import java.util.Map;
import org.apache.wicket.Component;
import org.apache.wicket.IRequestTarget;
import org.apache.wicket.PageParameters;
import org.apache.wicket.RequestCycle;
import org.apache.wicket.markup.html.WebPage;
import org.apache.wicket.markup.html.basic.Label;
import org.apache.wicket.protocol.http.WebResponse;
import org.eclipse.jgit.lib.Repository;
import org.eclipse.jgit.revwalk.RevCommit;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import com.gitblit.GitBlit;
import com.gitblit.Keys;
import com.gitblit.utils.JGitUtils;
import com.gitblit.utils.StringUtils;
import com.gitblit.wicket.WicketUtils;
public class RawPage extends WebPage {
   public RawPage(PageParameters params) {
   private final Logger logger = LoggerFactory.getLogger(getClass().getSimpleName());
   public RawPage(final PageParameters params) {
      super(params);
      if (!params.containsKey("r")) {
         error("Repository not specified!");
         error(getString("gb.repositoryNotSpecified"));
         redirectToInterceptPage(new RepositoriesPage());
      }
      final String repositoryName = WicketUtils.getRepositoryName(params);
      final String objectId = WicketUtils.getObject(params);
      final String blobPath = WicketUtils.getPath(params);
      Repository r = GitBlit.self().getRepository(repositoryName);
      if (r == null) {
         error("Can not load repository " + repositoryName);
         redirectToInterceptPage(new RepositoriesPage());
         return;
      }
      RevCommit commit = JGitUtils.getCommit(r, objectId);
      String extension = null;
      if (blobPath.lastIndexOf('.') > -1) {
         extension = blobPath.substring(blobPath.lastIndexOf('.') + 1);
      }
      // Map the extensions to types
      Map<String, Integer> map = new HashMap<String, Integer>();
      for (String ext : GitBlit.self().settings().getStrings(Keys.web.imageExtensions)) {
         map.put(ext.toLowerCase(), 2);
      }
      for (String ext : GitBlit.self().settings().getStrings(Keys.web.binaryExtensions)) {
         map.put(ext.toLowerCase(), 3);
      }
      if (extension != null) {
         int type = 0;
         if (map.containsKey(extension)) {
            type = map.get(extension);
      getRequestCycle().setRequestTarget(new IRequestTarget() {
         @Override
         public void detach(RequestCycle requestCycle) {
         }
         Component c = null;
         switch (type) {
         case 2:
            // TODO image blobs
            c = new Label("rawText", "Image File");
            break;
         case 3:
            // TODO binary blobs
            c = new Label("rawText", "Binary File");
            break;
         default:
            // plain text
            c = new Label("rawText", JGitUtils.getRawContentAsString(r, commit, blobPath));
            WicketUtils.setCssClass(c, "plainprint");
         @Override
         public void respond(RequestCycle requestCycle) {
            WebResponse response = (WebResponse) requestCycle.getResponse();
            final String repositoryName = WicketUtils.getRepositoryName(params);
            final String objectId = WicketUtils.getObject(params);
            final String blobPath = WicketUtils.getPath(params);
            String[] encodings = GitBlit.getEncodings();
            Repository r = GitBlit.self().getRepository(repositoryName);
            if (r == null) {
               error(getString("gb.canNotLoadRepository") + " " + repositoryName);
               redirectToInterceptPage(new RepositoriesPage());
               return;
            }
            if (StringUtils.isEmpty(blobPath)) {
               // objectid referenced raw view
               byte [] binary = JGitUtils.getByteContent(r, objectId);
               response.setContentType("application/octet-stream");
               response.setContentLength(binary.length);
               try {
                  response.getOutputStream().write(binary);
               } catch (Exception e) {
                  logger.error("Failed to write binary response", e);
               }
            } else {
               // standard raw blob view
               RevCommit commit = JGitUtils.getCommit(r, objectId);
               String filename = blobPath;
               if (blobPath.indexOf('/') > -1) {
                  filename = blobPath.substring(blobPath.lastIndexOf('/') + 1);
               }
               String extension = null;
               if (blobPath.lastIndexOf('.') > -1) {
                  extension = blobPath.substring(blobPath.lastIndexOf('.') + 1);
               }
               // Map the extensions to types
               Map<String, Integer> map = new HashMap<String, Integer>();
               for (String ext : GitBlit.getStrings(Keys.web.imageExtensions)) {
                  map.put(ext.toLowerCase(), 2);
               }
               for (String ext : GitBlit.getStrings(Keys.web.binaryExtensions)) {
                  map.put(ext.toLowerCase(), 3);
               }
               if (extension != null) {
                  int type = 0;
                  if (map.containsKey(extension)) {
                     type = map.get(extension);
                  }
                  switch (type) {
                  case 2:
                     // image blobs
                     byte[] image = JGitUtils.getByteContent(r, commit.getTree(), blobPath);
                     response.setContentType("image/" + extension.toLowerCase());
                     response.setContentLength(image.length);
                     try {
                        response.getOutputStream().write(image);
                     } catch (IOException e) {
                        logger.error("Failed to write image response", e);
                     }
                     break;
                  case 3:
                     // binary blobs (download)
                     byte[] binary = JGitUtils.getByteContent(r, commit.getTree(), blobPath);
                     response.setContentLength(binary.length);
                     response.setContentType("application/octet-stream");
                     response.setHeader("Content-Disposition", "attachment; filename=\"" + filename + "\"");
                     try {
                        response.getOutputStream().write(binary);
                     } catch (IOException e) {
                        logger.error("Failed to write binary response", e);
                     }
                     break;
                  default:
                     // plain text
                     String content = JGitUtils.getStringContent(r, commit.getTree(),
                           blobPath, encodings);
                     response.setContentType("text/plain; charset=UTF-8");
                     try {
                        response.getOutputStream().write(content.getBytes("UTF-8"));
                     } catch (Exception e) {
                        logger.error("Failed to write text response", e);
                     }
                  }
               } else {
                  // plain text
                  String content = JGitUtils.getStringContent(r, commit.getTree(), blobPath,
                        encodings);
                  response.setContentType("text/plain; charset=UTF-8");
                  try {
                     response.getOutputStream().write(content.getBytes("UTF-8"));
                  } catch (Exception e) {
                     logger.error("Failed to write text response", e);
                  }
               }
            }
            r.close();
         }
         add(c);
      } else {
         // plain text
         Label blobLabel = new Label("rawText", JGitUtils.getRawContentAsString(r, commit, blobPath));
         WicketUtils.setCssClass(blobLabel, "plainprint");
         add(blobLabel);
      }
      r.close();
      });
   }
}