From 4c21b172083a7b0cc48e795d20d93ed334dba392 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Edward=20Tj=C3=B6rnhammar?= <ed@cflags.cc>
Date: Fri, 4 Nov 2016 12:44:33 +0100
Subject: [PATCH] Add PlantUML support for code blocks

---
 plugins/PlantUML.hs | 56 +++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 56 insertions(+)
 create mode 100644 plugins/PlantUML.hs

diff --git a/plugins/PlantUML.hs b/plugins/PlantUML.hs
new file mode 100644
index 000000000..dfa73b344
--- /dev/null
+++ b/plugins/PlantUML.hs
@@ -0,0 +1,56 @@
+module PlantUML (plugin) where
+
+-- This plugin allows you to include a plantuml diagram
+-- in a page like this:
+--
+-- ~~~ {.puml}
+-- @startuml
+-- cloud cloud1
+-- cloud cloud2
+-- cloud cloud3
+-- cloud cloud4
+-- cloud cloud5
+-- cloud1 -0- cloud2
+-- cloud1 -0)- cloud3
+-- cloud1 -(0- cloud4
+-- cloud1 -(0)- cloud5
+-- @enduml
+-- ~~~
+--
+-- The "dot" and "plantuml" executable must be in the path.
+-- The generated png file will be saved in the static img directory.
+
+import GHC.IO.Handle
+import Network.Gitit.Interface
+import System.Process (readProcessWithExitCode)
+import System.Exit (ExitCode(ExitSuccess))
+-- from the temporary package on HackageDB
+import System.IO.Temp (withTempFile)
+-- from the utf8-string package on HackageDB:
+import Data.ByteString.Lazy.UTF8 (fromString)
+
+import System.Environment (unsetEnv)
+import System.FilePath ((</>))
+import System.FilePath
+
+plugin :: Plugin
+plugin = mkPageTransformM transformBlock
+
+transformBlock :: Block -> PluginM Block
+transformBlock (CodeBlock (id, classes, namevals) contents) | "puml" `elem` classes = do
+  cfg <- askConfig
+  let filetype = "svg"
+      outdir = staticDir cfg </> "img"
+  liftIO $ withTempFile outdir "diag.puml" $ \infile inhandle -> do
+    unsetEnv "DISPLAY"
+    hPutStr inhandle contents
+    hClose inhandle
+    (ec, stdout, stderr) <- readProcessWithExitCode "plantuml"
+      [ "-t" ++ filetype
+      , infile
+      ] ""
+    let outname = takeFileName $ infile -<.> filetype
+    if ec == ExitSuccess
+       then return $ Para [ Image (id,classes,namevals) [Str outname] ("/img" </> outname, "") ]
+       else error $ "plantuml returned an error status: " ++ stderr
+transformBlock x = return x