X-Git-Url: https://git.wpitchoune.net/gitweb/?a=blobdiff_plain;f=src%2Fmain%2Fjava%2FMain.java;h=5ce991290e49713252086aba92bd4d9892928ea5;hb=5404ccee528f0fb68cdc12e2db05ebef39ab7539;hp=042f2eea2a85b627683532ba4bb983ac35c29251;hpb=856e0ee4a51d4e79ea845ae5294588cb4c816633;p=asciidoctor_to_rss.git diff --git a/src/main/java/Main.java b/src/main/java/Main.java index 042f2ee..5ce9912 100644 --- a/src/main/java/Main.java +++ b/src/main/java/Main.java @@ -5,9 +5,12 @@ import java.io.FileReader; import java.io.IOException; import java.io.InputStream; import java.io.StringWriter; +import java.nio.charset.StandardCharsets; +import java.nio.file.Files; import java.util.ArrayList; import java.util.HashMap; import java.util.Properties; +import java.util.logging.Logger; import org.asciidoctor.Asciidoctor; import org.asciidoctor.Asciidoctor.Factory; @@ -22,6 +25,9 @@ import com.rometools.rome.io.FeedException; import com.rometools.rome.io.SyndFeedOutput; public class Main { + private static Logger LOG = Logger.getLogger(Main.class.getSimpleName()); + private static String KEY_HTML_HEADER_FILE = "html.header.file"; + private static File toHTMLFile(File dir, File adoc) { int idx; String name; @@ -36,10 +42,36 @@ public class Main { return new File(dir, name + ".html"); } + private static SyndContentImpl toSyndContentImpl(String description) { + SyndContentImpl ret; + + ret = new SyndContentImpl(); + ret.setType("text/html"); + ret.setValue(description); + + return ret; + } + + private static String getHTMLHeader(Properties props) throws IOException { + String fileName; + File f; + + fileName = props.getProperty(KEY_HTML_HEADER_FILE); + + if (fileName == null) { + LOG.info(KEY_HTML_HEADER_FILE + " is not set"); + return null; + } + + f = new File(fileName); + return new String(Files.readAllBytes(f.toPath()), + StandardCharsets.UTF_8); + } + public static void main(String[] args) throws FileNotFoundException, IOException, FeedException { File inDir, html, outDir; File[] adocs; - StringWriter sw; + StringWriter desc; Asciidoctor asciidoctor; SyndFeed feed; Properties props; @@ -48,6 +80,7 @@ public class Main { InputStream in; DocumentHeader h; SyndContentImpl c; + StringBuffer news; inDir = new File(args[0]); outDir = new File(args[1]); @@ -68,34 +101,50 @@ public class Main { entries = new ArrayList(); + news = new StringBuffer(); + news.append("\n"); + + getHTMLHeader(props); + + news.append("\n"); for (File adoc: adocs) { if (!adoc.getName().endsWith(".adoc")) continue; - sw = new StringWriter(); + desc = new StringWriter(); html = toHTMLFile(outDir, adoc); h = asciidoctor.readDocumentHeader(adoc); - asciidoctor.convert(new FileReader(adoc), sw, new HashMap()); + asciidoctor.convert(new FileReader(adoc), desc, new HashMap()); e = new SyndEntryImpl(); e.setTitle(h.getDocumentTitle().getMain()); e.setUri(props.getProperty("feed.baseurl") + "/" + html.getName()); - c = new SyndContentImpl(); - c.setType("text/html"); - c.setValue(sw.toString()); + c = toSyndContentImpl(desc.toString()); e.setDescription(c); entries.add(e); + + news.append("\n

\n"); + news.append("

"); + news.append(h.getDocumentTitle().getMain()); + news.append("

\n"); + news.append(desc.toString()); + news.append("\n

\n"); } + news.append("\n"); + news.append("\n"); feed.setEntries(entries); feed.setFeedType("rss_2.0"); SyndFeedOutput output = new SyndFeedOutput(); output.output(feed, new File(outDir, "feed.xml")); + + System.out.println(news.toString()); + Files.write(new File(outDir, "news.html").toPath(), news.toString().getBytes()); } } \ No newline at end of file