James Moger
2012-10-19 79dfe69726b6255464599ab852018e4d2ff96fdc
src/com/gitblit/wicket/pages/BlobPage.java
@@ -15,12 +15,14 @@
 */
package com.gitblit.wicket.pages;
import java.text.MessageFormat;
import java.util.HashMap;
import java.util.Map;
import org.apache.wicket.Component;
import org.apache.wicket.PageParameters;
import org.apache.wicket.markup.html.basic.Label;
import org.apache.wicket.markup.html.image.Image;
import org.apache.wicket.markup.html.link.BookmarkablePageLink;
import org.eclipse.jgit.lib.Constants;
import org.eclipse.jgit.lib.Repository;
@@ -30,6 +32,7 @@
import com.gitblit.Keys;
import com.gitblit.utils.JGitUtils;
import com.gitblit.utils.StringUtils;
import com.gitblit.wicket.ExternalImage;
import com.gitblit.wicket.WicketUtils;
import com.gitblit.wicket.panels.CommitHeaderPanel;
import com.gitblit.wicket.panels.PathBreadcrumbsPanel;
@@ -41,18 +44,21 @@
      Repository r = getRepository();
      final String blobPath = WicketUtils.getPath(params);
      String [] encodings = GitBlit.getEncodings();
      if (StringUtils.isEmpty(blobPath)) {
         // blob by objectid
         add(new Label("blameLink", getString("gb.blame")).setEnabled(false));
         add(new BookmarkablePageLink<Void>("blameLink", BlamePage.class,
               WicketUtils.newPathParameter(repositoryName, objectId, blobPath))
               .setEnabled(false));
         add(new BookmarkablePageLink<Void>("historyLink", HistoryPage.class).setEnabled(false));
         add(new BookmarkablePageLink<Void>("rawLink", RawPage.class,
               WicketUtils.newPathParameter(repositoryName, objectId, blobPath)));
         add(new BookmarkablePageLink<Void>("headLink", BlobPage.class).setEnabled(false));
         add(new CommitHeaderPanel("commitHeader", objectId));
         add(new PathBreadcrumbsPanel("breadcrumbs", repositoryName, blobPath, objectId));
         Component c = new Label("blobText", JGitUtils.getStringContent(r, objectId));
         Component c = new Label("blobText", JGitUtils.getStringContent(r, objectId, encodings));
         WicketUtils.setCssClass(c, "plainprint");
         add(c);
      } else {
@@ -74,7 +80,8 @@
         RevCommit commit = JGitUtils.getCommit(r, objectId);
         // blob page links
         add(new Label("blameLink", getString("gb.blame")));
         add(new BookmarkablePageLink<Void>("blameLink", BlamePage.class,
               WicketUtils.newPathParameter(repositoryName, objectId, blobPath)));
         add(new BookmarkablePageLink<Void>("historyLink", HistoryPage.class,
               WicketUtils.newPathParameter(repositoryName, objectId, blobPath)));
         add(new BookmarkablePageLink<Void>("rawLink", RawPage.class,
@@ -103,38 +110,82 @@
            if (map.containsKey(extension)) {
               type = map.get(extension);
            }
            Component c = null;
            switch (type) {
            case 1:
               // PrettyPrint blob text
               c = new Label("blobText", JGitUtils.getStringContent(r, commit.getTree(),
                     blobPath));
               WicketUtils.setCssClass(c, "prettyprint linenums");
               break;
            case 2:
               // TODO image blobs
               c = new Label("blobText", "Image File");
               // image blobs
               add(new Label("blobText").setVisible(false));
               add(new ExternalImage("blobImage", urlFor(RawPage.class, WicketUtils.newPathParameter(repositoryName, objectId, blobPath)).toString()));
               break;
            case 3:
               // TODO binary blobs
               c = new Label("blobText", "Binary File");
               // binary blobs
               add(new Label("blobText", "Binary File"));
               add(new Image("blobImage").setVisible(false));
               break;
            default:
               // plain text
               c = new Label("blobText", JGitUtils.getStringContent(r, commit.getTree(),
                     blobPath));
               WicketUtils.setCssClass(c, "plainprint");
               String source = JGitUtils.getStringContent(r, commit.getTree(), blobPath, encodings);
               String table = generateSourceView(source, type == 1);
               add(new Label("blobText", table).setEscapeModelStrings(false));
               add(new Image("blobImage").setVisible(false));
            }
            add(c);
         } else {
            // plain text
            Label blobLabel = new Label("blobText", JGitUtils.getStringContent(r,
                  commit.getTree(), blobPath));
            WicketUtils.setCssClass(blobLabel, "plainprint");
            add(blobLabel);
            String source = JGitUtils.getStringContent(r, commit.getTree(), blobPath, encodings);
            String table = generateSourceView(source, false);
            add(new Label("blobText", table).setEscapeModelStrings(false));
            add(new Image("blobImage").setVisible(false));
         }
      }
   }
   protected String generateSourceView(String source, boolean prettyPrint) {
      String [] lines = source.split("\n");
      StringBuilder sb = new StringBuilder();
      sb.append("<!-- start blob table -->");
      sb.append("<table width=\"100%\"><tbody><tr>");
      // nums column
      sb.append("<!-- start nums column -->");
      sb.append("<td id=\"nums\">");
      sb.append("<pre>");
      String numPattern = "<span id=\"L{0}\" class=\"num\">{0}</span>\n";
      for (int i = 0; i < lines.length; i++) {
         sb.append(MessageFormat.format(numPattern, "" + (i + 1)));
      }
      sb.append("</pre>");
      sb.append("<!-- end nums column -->");
      sb.append("</td>");
      sb.append("<!-- start lines column -->");
      sb.append("<td id=\"lines\">");
      sb.append("<div class=\"sourceview\">");
      if (prettyPrint) {
         sb.append("<pre class=\"prettyprint\">");
      } else {
         sb.append("<pre class=\"plainprint\">");
      }
      lines = StringUtils.escapeForHtml(source, true).split("\n");
      sb.append("<table width=\"100%\"><tbody>");
      String linePattern = "<tr class=\"{0}\"><td><a href=\"#L{2}\">{1}</a>\r</tr>";
      for (int i = 0; i < lines.length; i++) {
         String line = lines[i].replace('\r', ' ');
         String cssClass = (i % 2 == 0) ? "even" : "odd";
         sb.append(MessageFormat.format(linePattern, cssClass, line, "" + (i + 1)));
      }
      sb.append("</tbody></table></pre>");
      sb.append("</pre>");
      sb.append("</div>");
      sb.append("</td>");
      sb.append("<!-- end lines column -->");
      sb.append("</tr></tbody></table>");
      sb.append("<!-- end blob table -->");
      return sb.toString();
   }
   @Override
   protected String getPageName() {