From 13417cf9c6eec555b51da49742e47939d2f5715b Mon Sep 17 00:00:00 2001
From: James Moger <james.moger@gitblit.com>
Date: Fri, 19 Oct 2012 22:47:33 -0400
Subject: [PATCH] Exclude submodules from zip downloads (issue 151)

---
 src/com/gitblit/utils/GitWebDiffFormatter.java |   48 ++++++++++++++++++++++++++++++++++++------------
 1 files changed, 36 insertions(+), 12 deletions(-)

diff --git a/src/com/gitblit/utils/GitWebDiffFormatter.java b/src/com/gitblit/utils/GitWebDiffFormatter.java
index 2ac0a56..e657dc5 100644
--- a/src/com/gitblit/utils/GitWebDiffFormatter.java
+++ b/src/com/gitblit/utils/GitWebDiffFormatter.java
@@ -1,5 +1,21 @@
+/*
+ * 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.utils;
 
+import static org.eclipse.jgit.lib.Constants.encode;
 import static org.eclipse.jgit.lib.Constants.encodeASCII;
 
 import java.io.ByteArrayOutputStream;
@@ -8,7 +24,14 @@
 
 import org.eclipse.jgit.diff.DiffFormatter;
 import org.eclipse.jgit.diff.RawText;
+import org.eclipse.jgit.util.RawParseUtils;
 
+/**
+ * Returns an html snippet of the diff in the standard Gitweb style.
+ * 
+ * @author James Moger
+ * 
+ */
 public class GitWebDiffFormatter extends DiffFormatter {
 
 	private final OutputStream os;
@@ -32,7 +55,8 @@
 	 * @throws IOException
 	 */
 	@Override
-	protected void writeHunkHeader(int aStartLine, int aEndLine, int bStartLine, int bEndLine) throws IOException {
+	protected void writeHunkHeader(int aStartLine, int aEndLine, int bStartLine, int bEndLine)
+			throws IOException {
 		os.write("<div class=\"diff hunk_header\"><span class=\"diff hunk_info\">".getBytes());
 		os.write('@');
 		os.write('@');
@@ -78,21 +102,20 @@
 	}
 
 	@Override
-	protected void writeLine(final char prefix, final RawText text, final int cur) throws IOException {
+	protected void writeLine(final char prefix, final RawText text, final int cur)
+			throws IOException {
 		switch (prefix) {
 		case '+':
-			os.write("<span class=\"diff add\">".getBytes());
+			os.write("<span style=\"color:#008000;\">".getBytes());
 			break;
 		case '-':
-			os.write("<span class=\"diff remove\">".getBytes());
+			os.write("<span style=\"color:#800000;\">".getBytes());
 			break;
 		}
 		os.write(prefix);
-		ByteArrayOutputStream bos = new ByteArrayOutputStream();
-		text.writeLine(bos, cur);
-		String line = bos.toString();
+		String line = text.getString(cur);
 		line = StringUtils.escapeForHtml(line, false);
-		os.write(line.getBytes());
+		os.write(encode(line));
 		switch (prefix) {
 		case '+':
 		case '-':
@@ -110,17 +133,18 @@
 	 * @return
 	 */
 	public String getHtml() {
-		String html = os.toString();
+		ByteArrayOutputStream bos = (ByteArrayOutputStream) os;
+		String html = RawParseUtils.decode(bos.toByteArray());
 		String[] lines = html.split("\n");
 		StringBuilder sb = new StringBuilder();
 		sb.append("<div class=\"diff\">");
 		for (String line : lines) {
 			if (line.startsWith("diff")) {
-				sb.append("<div class=\"diff header\">").append(line).append("</div>");
+				sb.append("<div class=\"diff header\">").append(StringUtils.convertOctal(line)).append("</div>");
 			} else if (line.startsWith("---")) {
-				sb.append("<span class=\"diff remove\">").append(line).append("</span><br/>");
+				sb.append("<span style=\"color:#800000;\">").append(StringUtils.convertOctal(line)).append("</span><br/>");
 			} else if (line.startsWith("+++")) {
-				sb.append("<span class=\"diff add\">").append(line).append("</span><br/>");
+				sb.append("<span style=\"color:#008000;\">").append(StringUtils.convertOctal(line)).append("</span><br/>");
 			} else {
 				sb.append(line).append('\n');
 			}

--
Gitblit v1.9.1