From 3daba4304dba818d54eb9b74d8d35df5aab8fee5 Mon Sep 17 00:00:00 2001 From: James Moger <james.moger@gitblit.com> Date: Fri, 22 Nov 2013 10:36:05 -0500 Subject: [PATCH] Add pull request linking in changelog to generated documentation --- src/main/java/com/gitblit/wicket/pages/RawPage.java | 55 ++++++++++++++++++++++++++++++++++++++++++++++--------- 1 files changed, 46 insertions(+), 9 deletions(-) diff --git a/src/main/java/com/gitblit/wicket/pages/RawPage.java b/src/main/java/com/gitblit/wicket/pages/RawPage.java index 27a01f9..1783764 100644 --- a/src/main/java/com/gitblit/wicket/pages/RawPage.java +++ b/src/main/java/com/gitblit/wicket/pages/RawPage.java @@ -16,12 +16,15 @@ package com.gitblit.wicket.pages; import java.io.IOException; +import java.io.UnsupportedEncodingException; +import java.net.URLEncoder; import java.util.HashMap; import java.util.Map; import org.apache.wicket.IRequestTarget; import org.apache.wicket.PageParameters; import org.apache.wicket.RequestCycle; +import org.apache.wicket.protocol.http.WebRequest; import org.apache.wicket.protocol.http.WebResponse; import org.eclipse.jgit.lib.Repository; import org.eclipse.jgit.revwalk.RevCommit; @@ -41,9 +44,11 @@ private final Logger logger = LoggerFactory.getLogger(getClass().getSimpleName()); + String contentType; + public RawPage(final PageParameters params) { super(params); - + if (!params.containsKey("r")) { error(getString("gb.repositoryNotSpecified")); redirectToInterceptPage(new RepositoriesPage()); @@ -64,7 +69,7 @@ String[] encodings = GitBlit.getEncodings(); GitBlitWebSession session = GitBlitWebSession.get(); UserModel user = session.getUser(); - + RepositoryModel model = GitBlit.self().getRepositoryModel(user, repositoryName); if (model == null) { // user does not have permission @@ -72,7 +77,7 @@ redirectToInterceptPage(new RepositoriesPage()); return; } - + Repository r = GitBlit.self().getRepository(repositoryName); if (r == null) { error(getString("gb.canNotLoadRepository") + " " + repositoryName); @@ -83,7 +88,8 @@ if (StringUtils.isEmpty(blobPath)) { // objectid referenced raw view byte [] binary = JGitUtils.getByteContent(r, objectId); - response.setContentType("application/octet-stream"); + contentType = "application/octet-stream"; + response.setContentType(contentType); response.setContentLength(binary.length); try { response.getOutputStream().write(binary); @@ -122,7 +128,8 @@ case 2: // image blobs byte[] image = JGitUtils.getByteContent(r, commit.getTree(), blobPath, true); - response.setContentType("image/" + extension.toLowerCase()); + contentType = "image/" + extension.toLowerCase(); + response.setContentType(contentType); response.setContentLength(image.length); try { response.getOutputStream().write(image); @@ -133,9 +140,29 @@ case 3: // binary blobs (download) byte[] binary = JGitUtils.getByteContent(r, commit.getTree(), blobPath, true); + contentType = "application/octet-stream"; response.setContentLength(binary.length); - response.setContentType("application/octet-stream"); - response.setHeader("Content-Disposition", "attachment; filename=\"" + filename + "\""); + response.setContentType(contentType); + + try { + WebRequest request = (WebRequest) requestCycle.getRequest(); + String userAgent = request.getHttpServletRequest().getHeader("User-Agent"); + + if (userAgent != null && userAgent.indexOf("MSIE 5.5") > -1) { + response.setHeader("Content-Disposition", "filename=\"" + + URLEncoder.encode(filename, "UTF-8") + "\""); + } else if (userAgent != null && userAgent.indexOf("MSIE") > -1) { + response.setHeader("Content-Disposition", "attachment; filename=\"" + + URLEncoder.encode(filename, "UTF-8") + "\""); + } else { + response.setHeader("Content-Disposition", "attachment; filename=\"" + + new String(filename.getBytes("UTF-8"), "latin1") + "\""); + } + } + catch (UnsupportedEncodingException e) { + response.setHeader("Content-Disposition", "attachment; filename=\"" + filename + "\""); + } + try { response.getOutputStream().write(binary); } catch (IOException e) { @@ -146,7 +173,8 @@ // plain text String content = JGitUtils.getStringContent(r, commit.getTree(), blobPath, encodings); - response.setContentType("text/plain; charset=UTF-8"); + contentType = "text/plain; charset=UTF-8"; + response.setContentType(contentType); try { response.getOutputStream().write(content.getBytes("UTF-8")); } catch (Exception e) { @@ -158,7 +186,8 @@ // plain text String content = JGitUtils.getStringContent(r, commit.getTree(), blobPath, encodings); - response.setContentType("text/plain; charset=UTF-8"); + contentType = "text/plain; charset=UTF-8"; + response.setContentType(contentType); try { response.getOutputStream().write(content.getBytes("UTF-8")); } catch (Exception e) { @@ -170,4 +199,12 @@ } }); } + + @Override + protected void setHeaders(WebResponse response) { + super.setHeaders(response); + if (!StringUtils.isEmpty(contentType)) { + response.setContentType(contentType); + } + } } -- Gitblit v1.9.1