James Moger
2012-02-25 40538c57dd574d831d044cda50a8999941dc0a24
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,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());
@@ -103,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 '-':
@@ -125,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\">");