From 3b0bd4bcd3c7e8103cd9247b6a72b6826282f206 Mon Sep 17 00:00:00 2001
From: Oliver Marks <oly@digitaloctave.com>
Date: Thu, 18 Oct 2018 22:23:58 +0100
Subject: [PATCH] Tweak the html to apply classes

---
 eorg/generate.py | 25 +++++++++++++++----------
 1 file changed, 15 insertions(+), 10 deletions(-)

diff --git a/eorg/generate.py b/eorg/generate.py
index 988c2c6..538e847 100644
--- a/eorg/generate.py
+++ b/eorg/generate.py
@@ -9,26 +9,31 @@ def src(code):
     return highlight(code, lexer, HtmlFormatter())
 
 builddoc ={
-    "HEADER1": "h2",
-    "HEADER2": "h3",
-    "HEADER3": "h4",
-    "BREAK": "br",
-    "TEXT": "p",
-    "SRC_BEGIN": src,
-    "EXAMPLE": 'pre',
+    "HEADER1": ("h2", None),
+    "HEADER2": ("h3", None),
+    "HEADER3": ("h4", None),
+#    "BREAK": "br",
+    "TEXT": ("p", "flow-text"),
+    "SRC_BEGIN": (src, None),
+    "EXAMPLE": ('vblockquote', None),
 }
 
 
 def html(doc):
     response = StringIO()
     for item in doc:
-        tag = builddoc.get(item.token)
-        if not tag:
+        match = builddoc.get(item.token)
+        if not match:
             continue
+        tag, cls = match
+        if cls:
+            cls = f' class="{cls}"'
+        else:
+            cls = ''
         if callable(tag):
             response.write(tag(item.value))
             continue
         else:
-            response.write('<%s>%s<%s/>\n' % (tag, item.value, tag))
+            response.write('<%s%s>%s<%s/>\n' % (tag, cls, item.value, tag))
     response.seek(0)
     return response