milka64/sphinx-d2lang
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

clean code && add arguments & options

Browse source
This commit is contained in:
Milka64 2023-12-11 11:31:13 +01:00
parent 3aa4cbf553
commit 6febc0a475
1 changed files with 34 additions and 13 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

35
sphinx_d2lang/d2lang.py
View file

@ -1,27 +1,48 @@
from docutils import nodes
from docutils.parsers.rst import Directive
from docutils.parsers.rst import Directive, directives
from sphinx.util.docutils import SphinxDirective
from shutil import which
import shlex
import subprocess
import tempfile
class D2langDirective(Directive):
class D2langDirective(SphinxDirective):
required_arguments = 0
has_content = True
optional_arguments = 3
option_spec = {
'layout': directives.unchanged_required,
'filename': directives.unchanged_required,
}
def run(self):
# TODO : add tempfile for diag_source
d2_bin = which('d2')
outdir = self.state.document.settings.env.app.builder.outdir
srcdir = self.state.document.settings.env.srcdir
diag_source = self.content
if 'filename' in self.options:
output_fname = self.options.get('filename')
else:
output_fname = 'test.svg'
if 'layout':
layout = self.options.get('layout')
else:
layout = 'dagre'
if self.arguments:
build_svg(self.arguments[0], srcdir, output_fname, layout)
else:
with tempfile.NamedTemporaryFile() as fp:
for line in self.content.data:
fp.write(bytes(line,'utf-8'))
fp.write(bytes('\n','utf-8'))
fp.seek(0)
cmd_line = '%s %s %s' % (d2_bin, fp.name, srcdir + "/test.svg")
build_svg(fp.name, srcdir, output_fname, layout)
image_node = nodes.image(uri=output_fname)
return [image_node]
def build_svg(diag_src, out_dir, filename, layout)
d2_bin = which('d2')
cmd_line = '%s -l %s %s %s' % (d2_bin, layout, diag_src, out_dir + "/" + filename)
args = shlex.split(cmd_line)
subprocess.run(args)
image_node = nodes.image(uri="/test.svg")
return [image_node]
return True