James Moger
2012-06-18 67d4f89b0cddb3de05c20e08c20f1bea714c2a9e
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,7 +102,8 @@
   }
   @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());
@@ -88,11 +113,9 @@
         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,7 +133,8 @@
    * @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\">");