sphinx-d2lang/sphinx_d2lang/d2lang.py

from docutils import nodes

from docutils.parsers.rst import Directive

from shutil import which
import shlex
import subprocess
import tempfile

class D2langDirective(Directive):
    has_content = True
    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
        with tempfile.NamedTemporaryFile() as fp:
            for line in self.content.splitlines():
                fp.write(bytes(line,'utf-8'))
            fp.seek(0)
            cmd_line = '%s %s %s' % (d2_bin, fp.name, srcdir + "/test.svg")
            args = shlex.split(cmd_line)
            subprocess.run(args)
        image_node = nodes.image(uri="/test.svg")
        return [image_node]