Refine branch and pages servlets

James Moger

2014-05-04 1946fe76331b37c5a3be97268f0e3b0e58f3bb00

 src/main/java/com/gitblit/servlet/BranchServlet.java

@@ -35,14 +35,22 @@

import org.apache.tika.Tika;
import org.eclipse.jgit.lib.FileMode;
import org.eclipse.jgit.lib.MutableObjectId;
import org.eclipse.jgit.lib.ObjectLoader;
import org.eclipse.jgit.lib.ObjectReader;
import org.eclipse.jgit.lib.Repository;
import org.eclipse.jgit.revwalk.RevCommit;
import org.eclipse.jgit.revwalk.RevWalk;
import org.eclipse.jgit.treewalk.TreeWalk;
import org.eclipse.jgit.treewalk.filter.PathFilter;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.gitblit.Constants;
import com.gitblit.Keys;
import com.gitblit.dagger.DaggerServlet;
import com.gitblit.manager.IRepositoryManager;
import com.gitblit.manager.IRuntimeManager;
import com.gitblit.models.PathModel;
import com.gitblit.utils.ByteFormat;
import com.gitblit.utils.JGitUtils;
@@ -63,10 +71,13 @@

   private transient Logger logger = LoggerFactory.getLogger(BranchServlet.class);

   private IRuntimeManager runtimeManager;

   private IRepositoryManager repositoryManager;

   @Override
   protected void inject(ObjectGraph dagger) {
      this.runtimeManager = dagger.get(IRuntimeManager.class);
      this.repositoryManager = dagger.get(IRepositoryManager.class);
   }

@@ -83,7 +94,12 @@
      if (baseURL.length() > 0 && baseURL.charAt(baseURL.length() - 1) == '/') {
         baseURL = baseURL.substring(0, baseURL.length() - 1);
      }
      return baseURL + Constants.BRANCH + repository + "/" + (branch == null ? "" : (branch + "/" + (path == null ? "" : (path + "/"))));
      String encodedPath = path.replace(' ', '-');
      try {
         encodedPath = URLEncoder.encode(encodedPath, "UTF-8");
      } catch (UnsupportedEncodingException e) {
      }
      return baseURL + Constants.BRANCH + repository + "/" + (branch == null ? "" : (branch + "/" + (path == null ? "" : (encodedPath + "/"))));
   }

   protected String getBranch(String repository, HttpServletRequest request) {
@@ -204,9 +220,8 @@
         List<PathModel> pathEntries = JGitUtils.getFilesInPath(r, requestedPath, commit);
         if (pathEntries.isEmpty()) {
            // requested a specific resource
            String file = StringUtils.getLastPathElement(requestedPath);
            try {
               String file = StringUtils.getLastPathElement(requestedPath);

               // query Tika for the content type
               Tika tika = new Tika();
               String contentType = tika.detect(file);
@@ -220,16 +235,20 @@
                     contentType = "application/octet-stream";
                  }
               }
               response.setContentType(contentType);

               setContentType(response, contentType);

               if (contentType.startsWith("text/")
                     || "application/json".equals(contentType)
                     || "application/xml".equals(contentType)) {
               if (isTextType(contentType)) {

                  // serve text content
                  String encoding = commit.getEncoding().name();
                  response.setCharacterEncoding(encoding);
                  // load, interpret, and serve text content as UTF-8
                  String [] encodings = runtimeManager.getSettings().getStrings(Keys.web.blobEncodings).toArray(new String[0]);
                  String content = JGitUtils.getStringContent(r, commit.getTree(), requestedPath, encodings);

                  byte [] bytes = content.getBytes(Constants.ENCODING);
                  response.setContentLength(bytes.length);
                  ByteArrayInputStream is = new ByteArrayInputStream(bytes);
                  sendContent(response, JGitUtils.getCommitDate(commit), is);

               } else {
                  // serve binary content
                  String filename = StringUtils.getLastPathElement(requestedPath);
@@ -249,13 +268,10 @@
                  catch (UnsupportedEncodingException e) {
                     response.setHeader("Content-Disposition", "attachment; filename=\"" + filename + "\"");
                  }

                  // stream binary content directly from the repository
                  streamFromRepo(response, r, commit, requestedPath);
               }

               // send content
               byte [] content = JGitUtils.getByteContent(r, commit.getTree(), requestedPath, false);
               InputStream is = new ByteArrayInputStream(content);
               sendContent(response, JGitUtils.getCommitDate(commit), is);

               return;
            } catch (Exception e) {
               logger.error(null, e);
@@ -290,8 +306,8 @@
                        fullPath = requestedPath + "/" + fileName;
                     }

                     String encoding = commit.getEncoding().name();
                     String stringContent = JGitUtils.getStringContent(r, commit.getTree(), fullPath, encoding);
                     String [] encodings = runtimeManager.getSettings().getStrings(Keys.web.blobEncodings).toArray(new String[0]);
                     String stringContent = JGitUtils.getStringContent(r, commit.getTree(), fullPath, encodings);
                     if (stringContent == null) {
                        continue;
                     }
@@ -303,6 +319,7 @@

               response.setContentType("text/html; charset=" + Constants.ENCODING);
               byte [] bytes = content.getBytes(Constants.ENCODING);
               response.setContentLength(bytes.length);

               ByteArrayInputStream is = new ByteArrayInputStream(bytes);
               sendContent(response, JGitUtils.getCommitDate(commit), is);
@@ -368,6 +385,65 @@
      }
   }

   protected boolean isTextType(String contentType) {
      if (contentType.startsWith("text/")
            || "application/json".equals(contentType)
            || "application/xml".equals(contentType)) {
         return true;
      }
      return false;
   }

   /**
    * Override all text types to be plain text.
    *
    * @param response
    * @param contentType
    */
   protected void setContentType(HttpServletResponse response, String contentType) {
      if (isTextType(contentType)) {
         response.setContentType("text/plain");
      } else {
         response.setContentType(contentType);
      }
   }

   private void streamFromRepo(HttpServletResponse response, Repository repository,
         RevCommit commit, String requestedPath) throws IOException {

      response.setDateHeader("Last-Modified", JGitUtils.getCommitDate(commit).getTime());
      response.setHeader("Cache-Control", "public, max-age=3600, must-revalidate");

      RevWalk rw = new RevWalk(repository);
      TreeWalk tw = new TreeWalk(repository);
      try {
         tw.reset();
         tw.addTree(commit.getTree());
         PathFilter f = PathFilter.create(requestedPath);
         tw.setFilter(f);
         tw.setRecursive(true);
         MutableObjectId id = new MutableObjectId();
         ObjectReader reader = tw.getObjectReader();
         while (tw.next()) {
            FileMode mode = tw.getFileMode(0);
            if (mode == FileMode.GITLINK || mode == FileMode.TREE) {
               continue;
            }
            tw.getObjectId(id, 0);

            long len = reader.getObjectSize(id, org.eclipse.jgit.lib.Constants.OBJ_BLOB);
            response.setIntHeader("Content-Length", (int) len);
            ObjectLoader ldr = repository.open(id);
            ldr.copyTo(response.getOutputStream());
         }
      } finally {
         tw.release();
         rw.dispose();
      }

      response.flushBuffer();
   }

   private void sendContent(HttpServletResponse response, Date date, InputStream is) throws ServletException, IOException {
      response.setDateHeader("Last-Modified", date.getTime());
      response.setHeader("Cache-Control", "public, max-age=3600, must-revalidate");