You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
wiki/server/modules/rendering/markdown-wireviz/renderer.js

69 lines
1.8 KiB

const { spawnSync } = require('child_process')
module.exports = {
init(mdinst, conf = {}) {
this.md = mdinst
this.conf = conf
this.logger = (typeof WIKI !== 'undefined' && WIKI.logger) ? WIKI.logger : console
this.wirevizWebUrl =
conf.wirevizWebUrl ||
'http://localhost:3005'
const originalFence = mdinst.renderer.rules.fence || null
mdinst.renderer.rules.fence = (tokens, idx, options, env, slf) => {
const token = tokens[idx]
const info = (token.info || '').trim()
if (!info || !info.startsWith('wireviz')) {
return originalFence
? originalFence(tokens, idx, options, env, slf)
: `<pre><code>${mdinst.utils.escapeHtml(token.content)}</code></pre>`
}
const wirevizCode = token.content || ''
try {
const html = this.renderViaWirevizWebSync(wirevizCode)
return `<div class="wireviz-diagram">${html}</div>`
} catch (err) {
this.logger.error?.(`WireViz rendering error: ${err.message}`)
return `<pre>WireViz Error: ${mdinst.utils.escapeHtml(err.message)}</pre>`
}
}
},
renderViaWirevizWebSync(wirevizCode) {
// Use spawnSync to call curl synchronously and not add a new wikijs dependency
const curlArgs = [
'-s', // silent
'-X', 'POST',
'-F', `yml_file=@-`, // read from stdin
'-H', 'Accept: text/html',
this.wirevizWebUrl.replace(/\/$/, '') + '/render'
]
const res = spawnSync('curl', curlArgs, {
input: wirevizCode,
encoding: 'utf8'
})
if (res.error) {
throw new Error(`curl failed: ${res.error.message}`)
}
const stdout = res.stdout
const stderr = res.stderr
if (stderr) {
this.logger.warn?.('curl stderr: ' + stderr)
}
if (!stdout || !stdout.includes('<svg')) {
throw new Error(stdout)
}
return stdout
}
}