From fabe060d3a435f116128851f828e35c2af5fde67 Mon Sep 17 00:00:00 2001
From: James Moger <james.moger@gitblit.com>
Date: Mon, 10 Sep 2012 16:26:27 -0400
Subject: [PATCH] Strip folder name and .git from repo links in the project view

---
 src/com/gitblit/wicket/pages/RawPage.java |  198 +++++++++++++++++++++++++++++++++----------------
 1 files changed, 134 insertions(+), 64 deletions(-)

diff --git a/src/com/gitblit/wicket/pages/RawPage.java b/src/com/gitblit/wicket/pages/RawPage.java
index 961e4c9..7f6ed13 100644
--- a/src/com/gitblit/wicket/pages/RawPage.java
+++ b/src/com/gitblit/wicket/pages/RawPage.java
@@ -1,91 +1,161 @@
+/*
+ * Copyright 2011 gitblit.com.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
 package com.gitblit.wicket.pages;
 
+import java.io.IOException;
 import java.util.HashMap;
 import java.util.Map;
 
-import javax.servlet.http.HttpServletRequest;
-
-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.servlet.ServletWebRequest;
+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.StoredSettings;
 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);
 
-		ServletWebRequest servletWebRequest = (ServletWebRequest) getRequest();
-		HttpServletRequest req = servletWebRequest.getHttpServletRequest();
-		req.getServerName();
-
-		Repository r = GitBlit.self().getRepository(req, 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 : StoredSettings.getStrings(Keys.web_imageExtensions)) {
-			map.put(ext.toLowerCase(), 2);
-		}
-		for (String ext : StoredSettings.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();
+		});
 	}
 }

--
Gitblit v1.9.1