package com.gitblit.utils;
|
|
import java.io.IOException;
|
import java.io.OutputStream;
|
import java.text.MessageFormat;
|
import java.text.SimpleDateFormat;
|
import java.util.Date;
|
import java.util.HashMap;
|
import java.util.Map;
|
|
import org.eclipse.jgit.diff.DiffEntry;
|
import org.eclipse.jgit.diff.DiffFormatter;
|
import org.eclipse.jgit.diff.RawText;
|
import org.eclipse.jgit.revwalk.RevCommit;
|
|
import com.gitblit.Constants;
|
|
public class PatchFormatter extends DiffFormatter {
|
|
private final OutputStream os;
|
|
private PatchTouple currentTouple = null;
|
|
Map<String, PatchTouple> changes = new HashMap<String, PatchTouple>();
|
|
public PatchFormatter(OutputStream os) {
|
super(os);
|
this.os = os;
|
}
|
|
public void format(DiffEntry entry) throws IOException {
|
currentTouple = new PatchTouple();
|
changes.put(entry.getNewPath(), currentTouple);
|
super.format(entry);
|
}
|
|
@Override
|
protected void writeLine(final char prefix, final RawText text, final int cur) throws IOException {
|
switch (prefix) {
|
case '+':
|
currentTouple.insertions++;
|
break;
|
case '-':
|
currentTouple.deletions++;
|
break;
|
}
|
super.writeLine(prefix, text, cur);
|
}
|
|
public String getPatch(RevCommit commit) {
|
StringBuilder patch = new StringBuilder();
|
// hard-code the mon sep 17 2001 date string.
|
// I have no idea why that is there. it seems to be a constant.
|
patch.append("From " + commit.getName() + " Mon Sep 17 00:00:00 2001" + "\n");
|
patch.append("From: " + JGitUtils.getDisplayName(commit.getAuthorIdent()) + "\n");
|
patch.append("Date: " + (new SimpleDateFormat("EEE, dd MMM yyyy HH:mm:ss Z").format(new Date(commit.getCommitTime() * 1000l))) + "\n");
|
patch.append("Subject: [PATCH] " + commit.getShortMessage() + "\n");
|
patch.append("\n");
|
patch.append("---");
|
int maxPathLen = 0;
|
int files = 0;
|
int insertions = 0;
|
int deletions = 0;
|
for (String path : changes.keySet()) {
|
if (path.length() > maxPathLen) {
|
maxPathLen = path.length();
|
}
|
PatchTouple touple = changes.get(path);
|
files++;
|
insertions += touple.insertions;
|
deletions += touple.deletions;
|
}
|
int columns = 60;
|
int total = insertions + deletions;
|
int unit = total / columns + (total % columns > 0 ? 1 : 0);
|
if (unit == 0) {
|
unit = 1;
|
}
|
for (String path : changes.keySet()) {
|
PatchTouple touple = changes.get(path);
|
patch.append("\n " + Utils.rightPad(path, maxPathLen, ' ') + " | " + Utils.leftPad("" + touple.total(), 4, ' ') + " " + touple.relativeScale(unit));
|
}
|
patch.append(MessageFormat.format("\n {0} files changed, {1} insertions(+), {2} deletions(-)\n\n", files, insertions, deletions));
|
patch.append(os.toString());
|
patch.append("\n--\n");
|
patch.append(Constants.getRunningVersion());
|
return patch.toString();
|
}
|
|
private class PatchTouple {
|
int insertions = 0;
|
int deletions = 0;
|
|
int total() {
|
return insertions + deletions;
|
}
|
|
String relativeScale(int unit) {
|
int plus = (insertions / unit);
|
int minus = (deletions / unit);
|
StringBuilder sb = new StringBuilder();
|
for (int i = 0; i < plus; i++) {
|
sb.append('+');
|
}
|
for (int i = 0; i < minus; i++) {
|
sb.append('-');
|
}
|
return sb.toString();
|
}
|
}
|
}
|
