Kotlin rewrite, adding diagram options

1 files changed, 21 insertions, 0 deletions

diff --git a/src/main/kotlin/org/pacien/pandoc/filter/plantuml/PlantUml.kt b/src/main/kotlin/org/pacien/pandoc/filter/plantuml/PlantUml.kt
new file mode 100644
index 0000000..2a8da1b
--- /dev/null
+++ b/src/main/kotlin/org/pacien/pandoc/filter/plantuml/PlantUml.kt
@@ -0,0 +1,21 @@
+package org.pacien.pandoc.filter.plantuml
+
+import net.sourceforge.plantuml.FileFormat
+import net.sourceforge.plantuml.FileFormatOption
+import net.sourceforge.plantuml.SourceStringReader
+import java.io.ByteArrayOutputStream
+
+object PlantUml {
+  private val OUTPUT_FORMAT = FileFormatOption(FileFormat.LATEX_NO_PREAMBLE)
+
+  private fun SourceStringReader.generateImage(outputFormat: FileFormatOption) =
+    ByteArrayOutputStream().use { buffer ->
+      generateImage(buffer, outputFormat)
+      buffer.toString().lineSequence()
+    }
+
+  fun renderTikz(plantuml: String) =
+    SourceStringReader(plantuml)
+      .generateImage(OUTPUT_FORMAT)
+      .toLatex()
+}