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 |   37 ++++++++++++++++++++++++++-----------
 1 files changed, 26 insertions(+), 11 deletions(-)

diff --git a/src/main/java/com/gitblit/wicket/pages/RawPage.java b/src/main/java/com/gitblit/wicket/pages/RawPage.java
index d322e95..1783764 100644
--- a/src/main/java/com/gitblit/wicket/pages/RawPage.java
+++ b/src/main/java/com/gitblit/wicket/pages/RawPage.java
@@ -44,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());
@@ -67,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
@@ -75,7 +77,7 @@
 					redirectToInterceptPage(new RepositoriesPage());
 					return;
 				}
-				
+
 				Repository r = GitBlit.self().getRepository(repositoryName);
 				if (r == null) {
 					error(getString("gb.canNotLoadRepository") + " " + repositoryName);
@@ -86,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);
@@ -125,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);
@@ -136,13 +140,14 @@
 						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; charset=UTF-8");
-							
+							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") + "\"");
@@ -157,7 +162,7 @@
 							catch (UnsupportedEncodingException e) {
 								response.setHeader("Content-Disposition", "attachment; filename=\"" + filename + "\"");
 							}
-							
+
 							try {
 								response.getOutputStream().write(binary);
 							} catch (IOException e) {
@@ -168,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) {
@@ -180,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) {
@@ -192,4 +199,12 @@
 			}
 		});
 	}
+
+	@Override
+	protected void setHeaders(WebResponse response) {
+		super.setHeaders(response);
+		if (!StringUtils.isEmpty(contentType)) {
+			response.setContentType(contentType);
+		}
+	}
 }

--
Gitblit v1.9.1