makespec.py: create toc without relying on pandoc.

We're still using pandoc, but this is one step closer to using
cmark to build spec.html.

1 files changed, 44 insertions, 5 deletions

diff --git a/makespec.py b/makespec.py
index 8e73cf4..c7b90b5 100644
--- a/makespec.py
+++ b/makespec.py
@@ -13,6 +13,9 @@ else:
     sys.stderr.write("Usage:  makespec.py SPECFILE [html|markdown]\n")
     exit(1)
 
+def toIdentifier(s):
+   return re.sub(r'\s+', '-', re.sub(r'\W+', ' ', s.strip().lower()))
+
 def pipe_through_prog(prog, text):
     p1 = Popen(prog.split(), stdout=PIPE, stdin=PIPE, stderr=PIPE)
     [result, err] = p1.communicate(input=text.encode('utf-8'))
@@ -28,12 +31,20 @@ def replaceAnchor(match):
 stage = 0
 example = 0
 section = ""
+sections = []
 mdlines = []
 refs = []
+lastnum = []
+finishedMeta = False
+yamllines = []
 
 with open(specfile, 'r', encoding='utf-8') as spec:
     for ln in spec:
-        if re.match(r'^\.$', ln):
+        if not finishedMeta:
+            yamllines.append(ln)
+            if re.match(r'^\.\.\.$', ln):
+                finishedMeta = True
+        elif re.match(r'^\.$', ln):
             if stage == 0:
                 example += 1
                 mdlines.append("\n<div class=\"example\" id=\"example-{0}\" data-section=\"{1}\">\n".format(example, section))
@@ -53,9 +64,30 @@ with open(specfile, 'r', encoding='utf-8') as spec:
                 sys.exit(1)
         else:
             if stage == 0:
-                match = re.match(r'^#{1,6} *(.*)', ln)
+                match = re.match(r'^(#{1,6}) *(.*)', ln)
                 if match:
-                    section = match.group(1)
+                    section = match.group(2)
+                    lastlevel = len(lastnum)
+                    level = len(match.group(1))
+                    if re.search(r'{-}$', section):
+                        section = re.sub(r' *{-} *$', '', section)
+                        number = ''
+                    else:
+                        if lastlevel == level:
+                            lastnum[level - 1] = lastnum[level - 1] + 1
+                        elif lastlevel < level:
+                            while len(lastnum) < level:
+                                lastnum.append(1)
+                        else: # lastlevel > level
+                            lastnum = lastnum[0:level]
+                            lastnum[level - 1] = lastnum[level - 1] + 1
+                        number = '.'.join([str(x) for x in lastnum])
+                    ident = toIdentifier(section)
+                    sections.append(dict(level=level,
+                                         contents=section,
+                                         ident=ident,
+                                         number=number))
+                    refs.append("[{0}]: #{1}".format(section, ident))
                 else:
                     ln = re.sub(r'\[([^]]*)\]\(@([^)]*)\)', replaceAnchor, ln)
             else:
@@ -67,8 +99,15 @@ mdtext = ''.join(mdlines) + '\n\n' + '\n'.join(refs) + '\n'
 if specformat == "markdown":
     sys.stdout.write(mdtext)
 elif specformat == "html":
-    prog = "pandoc -s --toc -S --no-highlight --number-sections --template template.html"
-    [retcode, result, err] = pipe_through_prog(prog, mdtext)
+    toclines = []
+    for section in sections:
+        indent = '    ' * (section['level'] - 1)
+        toclines.append(indent + '* [' + section['number'] + ' ' +
+                        section['contents'] + '](#' + section['ident'] + ')')
+    toc = '<div id="TOC">\n\n' + '\n'.join(toclines) + '\n\n</div>\n\n'
+    yaml = ''.join(yamllines) + '\n'
+    prog = "pandoc -s -S --no-highlight --number-sections --template template.html"
+    [retcode, result, err] = pipe_through_prog(prog, yaml + toc + mdtext)
     if retcode == 0:
         result = re.sub(r'␣', '<span class="space"> </span>', result)
         sys.stdout.write(result)