Raw latex support

pandoc_plantuml_filter.py

@@ -10,21 +10,31 @@
 import sys
 import subprocess
 
-from pandocfilters import toJSONFilter, Para, Image
+from pandocfilters import toJSONFilter, Para, Image, RawBlock
 from pandocfilters import get_filename4code, get_caption, get_extension
 
 PLANTUML_BIN = os.environ.get('PLANTUML_BIN', 'plantuml')
 
+pattern = re.compile('%\{(.*)\}$')
 
 def plantuml(key, value, format_, _):
+
+    if key == 'Header':
+        if 'tikz'in (str(_)):
+            os.environ["PLANTUML_LATEX_EXPORT"] = 'latex'        
     if key == 'CodeBlock':
         [[ident, classes, keyvals], code] = value
 
         if "plantuml" in classes:
             caption, typef, keyvals = get_caption(keyvals)
-
+            #
+            if "PLANTUML_LATEX_EXPORT" in os.environ:
+                plantuml_latex_export="latex"
+            else:
+                plantuml_latex_export="png"
+            #
             filename = get_filename4code("plantuml", code)
-            filetype = get_extension(format_, "png", html="svg", latex="png")
+            filetype = get_extension(format_, "png", html="svg", latex=plantuml_latex_export)
 
             src = filename + '.uml'
             dest = filename + '.' + filetype
@@ -35,12 +45,13 @@ def plantuml(key, value, format_, _):
                     txt = b"@startuml\n" + txt + b"\n@enduml\n"
                 with open(src, "wb") as f:
                     f.write(txt)
-
-                subprocess.check_call([
-                    PLANTUML_BIN, "-t" + filetype, src])
+                subprocess.check_call(PLANTUML_BIN.split() + ["-t" + filetype, src])
                 sys.stderr.write('Created image ' + dest + '\n')
-
-            return Para([Image([ident, [], keyvals], caption, [dest, typef])])
+            if (filetype=="latex") and (plantuml_latex_export=='latex'):
+                latex = open(dest).read()
+                return RawBlock('latex', latex.split("\\begin{document}")[-1].split("\\end{document}")[0])
+            else:           
+                return Para([Image([ident, [], keyvals], caption, [dest, typef])])
 
 
 def main():

tests/sample.md

@@ -8,4 +8,32 @@ Alice -> Bob: Another authentication Request
 Alice <-- Bob: another authentication Response
 ```
 
+```plantuml
+
+Alice -> Bob: Authentication Request
+
+alt successful case
+
+	Bob -> Alice: Authentication Accepted
+
+else some kind of failure
+
+	Bob -> Alice: Authentication Failure
+	group My own label
+		Alice -> Log : Log attack start
+	    loop 1000 times
+	        Alice -> Bob: DNS Attack
+	    end
+		Alice -> Log : Log attack end
+	end
+
+else Another type of failure
+
+   Bob -> Alice: Please repeat
+
+end
+
+
+```
+
 Nice, huh?

tests/sample_latex_uml.md

@@ -0,0 +1,43 @@
+---
+header-includes: |
+    \usepackage{tikz}
+---
+# Some PantUML sample
+
+```plantuml
+Alice -> Bob: Authentication Request
+Bob --> Alice: Authentication Response
+
+Alice -> Bob: Another authentication Request
+Alice <-- Bob: another authentication Response
+```
+
+```plantuml
+
+Alice -> Bob: Authentication Request
+
+alt successful case
+
+	Bob -> Alice: Authentication Accepted
+
+else some kind of failure
+
+	Bob -> Alice: Authentication Failure
+	group My own label
+		Alice -> Log : Log attack start
+	    loop 1000 times
+	        Alice -> Bob: DNS Attack
+	    end
+		Alice -> Log : Log attack end
+	end
+
+else Another type of failure
+
+   Bob -> Alice: Please repeat
+
+end
+
+
+```
+
+Nice, huh?