Real raw page for text, images, and binaries (issue 6, issue 114)

James Moger

2012-07-25 e4f49a0b6956c3000915bc89f7158abd7969f7fe

 src/com/gitblit/wicket/pages/RawPage.java

@@ -15,15 +15,19 @@
 */

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;

@@ -33,13 +37,25 @@


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(getString("gb.repositoryNotSpecified"));

         redirectToInterceptPage(new RepositoriesPage());

      }



      getRequestCycle().setRequestTarget(new IRequestTarget() {

         @Override

         public void detach(RequestCycle requestCycle) {

         }



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

@@ -54,12 +70,22 @@


      if (StringUtils.isEmpty(blobPath)) {

         // objectid referenced raw view

         Label blobLabel = new Label("rawText", JGitUtils.getStringContent(r, objectId, encodings));

         WicketUtils.setCssClass(blobLabel, "plainprint");

         add(blobLabel);

               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) {

@@ -80,31 +106,56 @@
            if (map.containsKey(extension)) {

               type = map.get(extension);

            }

            Component c = null;

            switch (type) {

            case 2:

               // TODO image blobs

               c = new Label("rawText", "Image File");

                     // 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:

               // TODO binary blobs

               c = new Label("rawText", "Binary File");

                     // 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

               c = new Label("rawText", JGitUtils.getStringContent(r, commit.getTree(),

                     blobPath, encodings));

               WicketUtils.setCssClass(c, "plainprint");

                     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);

            }

            add(c);

                  }



         } else {

            // plain text

            Label blobLabel = new Label("rawText", JGitUtils.getStringContent(r,

                  commit.getTree(), blobPath, encodings));

            WicketUtils.setCssClass(blobLabel, "plainprint");

            add(blobLabel);

                  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();

   }

      });

   }

}