James Moger
2012-09-29 1e1b85270f93b3bca624c99b478f3a9a23be2395
src/com/gitblit/utils/GitWebDiffFormatter.java
@@ -15,6 +15,7 @@
 */
package com.gitblit.utils;
import static org.eclipse.jgit.lib.Constants.encode;
import static org.eclipse.jgit.lib.Constants.encodeASCII;
import java.io.ByteArrayOutputStream;
@@ -23,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;
@@ -47,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('@');
@@ -93,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 '-':
@@ -125,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');
         }