Factored out cmark.py from test programs.

1 files changed, 38 insertions, 0 deletions

diff --git a/test/cmark.py b/test/cmark.py
new file mode 100644
index 0000000..432a3e0
--- /dev/null
+++ b/test/cmark.py
@@ -0,0 +1,38 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+
+from ctypes import CDLL, c_char_p, c_long
+from subprocess import *
+import platform
+
+def pipe_through_prog(prog, text):
+    p1 = Popen(prog.split(), stdout=PIPE, stdin=PIPE, stderr=PIPE)
+    [result, err] = p1.communicate(input=text)
+    return [p1.returncode, result, err]
+
+def use_library(lib, text):
+    return [0, lib(text, len(text)), '']
+
+class CMark:
+    def __init__(self, prog=None, library_dir=None):
+        self.prog = prog
+        if prog:
+            self.to_html = lambda x: pipe_through_prog(prog, x)
+        else:
+            sysname = platform.system()
+            libname = "libcmark"
+            if sysname == 'Darwin':
+                libname += ".dylib"
+            elif sysname == 'Windows':
+                libname += ".dll"
+            else:
+                libname += ".so"
+            if library_dir:
+                libpath = library_dir + "/" + libname
+            else:
+                libpath = "build/src/" + libname
+            cmark = CDLL(libpath)
+            markdown = cmark.cmark_markdown_to_html
+            markdown.restype = c_char_p
+            markdown.argtypes = [c_char_p, c_long]
+            self.to_html = lambda x: use_library(markdown, x)