diff options
| -rw-r--r-- | makespec.py | 49 |
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) |