From a9a2ffcf9a34bd25fe2e05bfdd4cde74725bb17d Mon Sep 17 00:00:00 2001 From: Milos Cubrilo <milos.cubrilo@gmail.com> Date: Sun, 11 Jan 2015 07:41:29 -0500 Subject: [PATCH] #230 - Improve empty folder navigation. --- src/main/java/com/gitblit/utils/DiffUtils.java | 134 +++++++++++++++++++++++++++++++++++++++++++- 1 files changed, 131 insertions(+), 3 deletions(-) diff --git a/src/main/java/com/gitblit/utils/DiffUtils.java b/src/main/java/com/gitblit/utils/DiffUtils.java index 107f6ab..b30a203 100644 --- a/src/main/java/com/gitblit/utils/DiffUtils.java +++ b/src/main/java/com/gitblit/utils/DiffUtils.java @@ -52,6 +52,27 @@ private static final Logger LOGGER = LoggerFactory.getLogger(DiffUtils.class); /** + * Callback interface for binary diffs. All the getDiff methods here take an optional handler; + * if given and the {@link DiffOutputType} is {@link DiffOutputType#HTML HTML}, it is responsible + * for displaying a binary diff. + */ + public interface BinaryDiffHandler { + + /** + * Renders a binary diff. The result must be valid HTML, it will be inserted into an HTML table cell. + * May return {@code null} if the default behavior (which is typically just a textual note "Bnary + * files differ") is desired. + * + * @param diffEntry + * current diff entry + * + * @return the rendered diff as HTML, or {@code null} if the default is desired. + */ + public String renderBinaryDiff(final DiffEntry diffEntry); + + } + + /** * Enumeration for the diff output types. */ public static enum DiffOutputType { @@ -181,6 +202,23 @@ } /** + * Returns the complete diff of the specified commit compared to its primary parent. + * + * @param repository + * @param commit + * @param outputType + * @param handler + * to use for rendering binary diffs if {@code outputType} is {@link DiffOutputType#HTML HTML}. + * May be {@code null}, resulting in the default behavior. + * @return the diff + */ + public static DiffOutput getCommitDiff(Repository repository, RevCommit commit, + DiffOutputType outputType, BinaryDiffHandler handler) { + return getDiff(repository, null, commit, null, outputType, handler); + } + + + /** * Returns the diff for the specified file or folder from the specified * commit compared to its primary parent. * @@ -196,6 +234,24 @@ } /** + * Returns the diff for the specified file or folder from the specified + * commit compared to its primary parent. + * + * @param repository + * @param commit + * @param path + * @param outputType + * @param handler + * to use for rendering binary diffs if {@code outputType} is {@link DiffOutputType#HTML HTML}. + * May be {@code null}, resulting in the default behavior. + * @return the diff + */ + public static DiffOutput getDiff(Repository repository, RevCommit commit, String path, + DiffOutputType outputType, BinaryDiffHandler handler) { + return getDiff(repository, null, commit, path, outputType, handler); + } + + /** * Returns the complete diff between the two specified commits. * * @param repository @@ -207,6 +263,23 @@ public static DiffOutput getDiff(Repository repository, RevCommit baseCommit, RevCommit commit, DiffOutputType outputType) { return getDiff(repository, baseCommit, commit, null, outputType); + } + + /** + * Returns the complete diff between the two specified commits. + * + * @param repository + * @param baseCommit + * @param commit + * @param outputType + * @param handler + * to use for rendering binary diffs if {@code outputType} is {@link DiffOutputType#HTML HTML}. + * May be {@code null}, resulting in the default behavior. + * @return the diff + */ + public static DiffOutput getDiff(Repository repository, RevCommit baseCommit, RevCommit commit, + DiffOutputType outputType, BinaryDiffHandler handler) { + return getDiff(repository, baseCommit, commit, null, outputType, handler); } /** @@ -225,18 +298,41 @@ */ public static DiffOutput getDiff(Repository repository, RevCommit baseCommit, RevCommit commit, String path, DiffOutputType outputType) { + return getDiff(repository, baseCommit, commit, path, outputType, null); + } + + /** + * Returns the diff between two commits for the specified file. + * + * @param repository + * @param baseCommit + * if base commit is null the diff is to the primary parent of + * the commit. + * @param commit + * @param path + * if the path is specified, the diff is restricted to that file + * or folder. if unspecified, the diff is for the entire commit. + * @param outputType + * @param handler + * to use for rendering binary diffs if {@code outputType} is {@link DiffOutputType#HTML HTML}. + * May be {@code null}, resulting in the default behavior. + * @return the diff + */ + public static DiffOutput getDiff(Repository repository, RevCommit baseCommit, RevCommit commit, String path, DiffOutputType outputType, + final BinaryDiffHandler handler) { DiffStat stat = null; String diff = null; try { - final ByteArrayOutputStream os = new ByteArrayOutputStream(); + ByteArrayOutputStream os = null; RawTextComparator cmp = RawTextComparator.DEFAULT; DiffFormatter df; switch (outputType) { case HTML: - df = new GitBlitDiffFormatter(os, commit.getName()); + df = new GitBlitDiffFormatter(commit.getName(), path, handler); break; case PLAIN: default: + os = new ByteArrayOutputStream(); df = new DiffFormatter(os); break; } @@ -271,6 +367,7 @@ } else { df.format(diffEntries); } + df.flush(); if (df instanceof GitBlitDiffFormatter) { // workaround for complex private methods in DiffFormatter diff = ((GitBlitDiffFormatter) df).getHtml(); @@ -278,7 +375,6 @@ } else { diff = os.toString(); } - df.flush(); } catch (Throwable t) { LOGGER.error("failed to generate commit diff!", t); } @@ -346,6 +442,38 @@ return diff; } + /** + * Returns the diffstat between the two commits for the specified file or folder. + * + * @param repository + * @param base + * if base commit is unspecified, the diffstat is generated against + * the primary parent of the specified tip. + * @param tip + * @param path + * if path is specified, the diffstat is generated only for the + * specified file or folder. if unspecified, the diffstat is + * generated for the entire diff between the two commits. + * @return patch as a string + */ + public static DiffStat getDiffStat(Repository repository, String base, String tip) { + RevCommit baseCommit = null; + RevCommit tipCommit = null; + RevWalk revWalk = new RevWalk(repository); + try { + tipCommit = revWalk.parseCommit(repository.resolve(tip)); + if (!StringUtils.isEmpty(base)) { + baseCommit = revWalk.parseCommit(repository.resolve(base)); + } + return getDiffStat(repository, baseCommit, tipCommit, null); + } catch (Exception e) { + LOGGER.error("failed to generate diffstat!", e); + } finally { + revWalk.dispose(); + } + return null; + } + public static DiffStat getDiffStat(Repository repository, RevCommit commit) { return getDiffStat(repository, null, commit, null); } -- Gitblit v1.9.1