James Moger
2011-06-08 008322bec70a3a20bd00ed2219215a9f42fe0ca5
src/com/gitblit/BuildSite.java
@@ -37,6 +37,7 @@
import com.beust.jcommander.ParameterException;
import com.beust.jcommander.Parameters;
import com.gitblit.utils.MarkdownUtils;
import com.gitblit.utils.StringUtils;
public class BuildSite {
@@ -72,19 +73,21 @@
      StringBuilder sb = new StringBuilder();
      for (File file : markdownFiles) {
         String documentName = getDocumentName(file);
         String displayName = documentName;
         if (aliasMap.containsKey(documentName)) {
            displayName = aliasMap.get(documentName);
         if (!params.skips.contains(documentName)) {
            String displayName = documentName;
            if (aliasMap.containsKey(documentName)) {
               displayName = aliasMap.get(documentName);
            }
            String fileName = documentName + ".html";
            sb.append(MessageFormat.format(linkPattern, fileName, displayName));
            sb.append(" | ");
         }
         String fileName = documentName + ".html";
         sb.append(MessageFormat.format(linkPattern, fileName, displayName));
         sb.append(" | ");
      }
      sb.setLength(sb.length() - 3);
      sb.trimToSize();
      String htmlHeader = readContent(new File(params.pageHeader));
      String htmlFooter = readContent(new File(params.pageFooter));
      String htmlHeader = readContent(new File(params.pageHeader), "\n");
      String htmlFooter = readContent(new File(params.pageFooter), "\n");
      final String links = sb.toString();
      final String header = MessageFormat.format(htmlHeader, Constants.FULL_NAME, links);
      final String date = new SimpleDateFormat("yyyy-MM-dd").format(new Date());
@@ -92,22 +95,32 @@
      for (File file : markdownFiles) {
         try {
            String documentName = getDocumentName(file);
            String fileName = documentName + ".html";
            System.out.println(MessageFormat.format("  {0} => {1}", file.getName(), fileName));
            InputStreamReader reader = new InputStreamReader(new FileInputStream(file),
                  Charset.forName("UTF-8"));
            String content = MarkdownUtils.transformMarkdown(reader);
            for (String token : params.substitutions) {
               String[] kv = token.split("=");
               content = content.replace(kv[0], kv[1]);
            if (!params.skips.contains(documentName)) {
               String fileName = documentName + ".html";
               System.out.println(MessageFormat.format("  {0} => {1}", file.getName(),
                     fileName));
               InputStreamReader reader = new InputStreamReader(new FileInputStream(file),
                     Charset.forName("UTF-8"));
               String content = MarkdownUtils.transformMarkdown(reader);
               for (String token : params.substitutions) {
                  String[] kv = token.split("=");
                  content = content.replace(kv[0], kv[1]);
               }
               for (String alias : params.loads) {
                  String[] kv = alias.split("=");
                  String loadedContent = readContent(new File(kv[1]), "\n");
                  loadedContent = StringUtils.escapeForHtml(loadedContent, false);
                  loadedContent = StringUtils.breakLinesForHtml(loadedContent);
                  content = content.replace(kv[0], loadedContent);
               }
               OutputStreamWriter writer = new OutputStreamWriter(new FileOutputStream(
                     new File(destinationFolder, fileName)), Charset.forName("UTF-8"));
               writer.write(header);
               writer.write(content);
               writer.write(footer);
               reader.close();
               writer.close();
            }
            OutputStreamWriter writer = new OutputStreamWriter(new FileOutputStream(new File(
                  destinationFolder, fileName)), Charset.forName("UTF-8"));
            writer.write(header);
            writer.write(content);
            writer.write(footer);
            reader.close();
            writer.close();
         } catch (Throwable t) {
            System.err.println("Failed to transform " + file.getName());
            t.printStackTrace();
@@ -115,7 +128,7 @@
      }
   }
   private static String readContent(File file) {
   private static String readContent(File file, String lineEnding) {
      StringBuilder sb = new StringBuilder();
      try {
         InputStreamReader is = new InputStreamReader(new FileInputStream(file),
@@ -124,6 +137,9 @@
         String line = null;
         while ((line = reader.readLine()) != null) {
            sb.append(line);
            if (lineEnding != null) {
               sb.append(lineEnding);
            }
         }
         reader.close();
      } catch (Throwable t) {
@@ -168,11 +184,17 @@
      @Parameter(names = { "--pageFooter" }, description = "Page Footer HTML Snippet", required = true)
      public String pageFooter;
      @Parameter(names = { "--skip" }, description = "Filename to skip", required = false)
      public List<String> skips = new ArrayList<String>();
      @Parameter(names = { "--alias" }, description = "Filename=Linkname aliases", required = false)
      public List<String> aliases = new ArrayList<String>();
      @Parameter(names = { "--substitute" }, description = "@TOKEN@=value", required = false)
      @Parameter(names = { "--substitute" }, description = "%TOKEN%=value", required = false)
      public List<String> substitutions = new ArrayList<String>();
      @Parameter(names = { "--load" }, description = "%TOKEN%=filename", required = false)
      public List<String> loads = new ArrayList<String>();
   }
}