mirror of
https://github.com/mermaid-js/mermaid.git
synced 2024-09-06 17:43:35 +02:00
214 lines
5.4 KiB
JavaScript
214 lines
5.4 KiB
JavaScript
![]() |
/**
|
||
|
* Credits:
|
||
|
* - SVG Processing from the NYTimes svg-crowbar, under an MIT license
|
||
|
* https://github.com/NYTimes/svg-crowbar
|
||
|
* - Thanks to the grunticon project for some guidance
|
||
|
* https://github.com/filamentgroup/grunticon
|
||
|
*/
|
||
|
|
||
|
phantom.onError = function(msg, trace) {
|
||
|
var msgStack = ['PHANTOM ERROR: ' + msg]
|
||
|
if (trace && trace.length) {
|
||
|
msgStack.push('TRACE:')
|
||
|
trace.forEach(function(t) {
|
||
|
msgStack.push(
|
||
|
' -> '
|
||
|
+ (t.file || t.sourceURL)
|
||
|
+ ': '
|
||
|
+ t.line
|
||
|
+ (t.function ? ' (in function ' + t.function +')' : '')
|
||
|
)
|
||
|
})
|
||
|
}
|
||
|
system.stderr.write(msgStack.join('\n'))
|
||
|
phantom.exit(1)
|
||
|
}
|
||
|
|
||
|
var system = require('system')
|
||
|
, fs = require('fs')
|
||
|
, webpage = require('webpage')
|
||
|
|
||
|
var page = webpage.create()
|
||
|
, files = phantom.args.slice(4, phantom.args.length)
|
||
|
, options = {
|
||
|
outputDir: phantom.args[0]
|
||
|
, png: phantom.args[1] === 'true' ? true : false
|
||
|
, svg: phantom.args[2] === 'true' ? true : false
|
||
|
, verbose: phantom.args[3] === 'true' ? true : false
|
||
|
}
|
||
|
, log = logger(options.verbose)
|
||
|
|
||
|
page.content = [
|
||
|
'<html>'
|
||
|
, '<head>'
|
||
|
, '<style type="text/css">'
|
||
|
, '* { margin: 0; padding: 0; }'
|
||
|
, '</style>'
|
||
|
, '</head>'
|
||
|
, '<body>'
|
||
|
, '</body>'
|
||
|
, '</html>'
|
||
|
].join('\n')
|
||
|
|
||
|
page.injectJs('../dist/mermaid.full.js')
|
||
|
|
||
|
files.forEach(function(file) {
|
||
|
var contents = fs.read(file)
|
||
|
, filename = file.split(fs.separator).slice(-1)
|
||
|
, oParser = new DOMParser()
|
||
|
, oDOM
|
||
|
, svgContent
|
||
|
, allElements
|
||
|
|
||
|
// this JS is executed in this statement is sandboxed, even though it doesn't
|
||
|
// look like it. we need to serialize then unserialize the svgContent that's
|
||
|
// taken from the DOM
|
||
|
svgContent = page.evaluate(executeInPage, contents)
|
||
|
oDOM = oParser.parseFromString(svgContent, "text/xml")
|
||
|
|
||
|
resolveSVGElement(oDOM.firstChild)
|
||
|
|
||
|
// traverse the SVG, and replace all foreignObject elements
|
||
|
// can be removed when https://github.com/knsv/mermaid/issues/58 is resolved
|
||
|
allElements = traverse(oDOM)
|
||
|
for (var i = 0, len = allElements.length; i < len; i++) {
|
||
|
resolveForeignObjects(allElements[i])
|
||
|
}
|
||
|
|
||
|
if (options.png) {
|
||
|
page.viewportSize = {
|
||
|
width: ~~oDOM.documentElement.attributes.getNamedItem('width').value
|
||
|
, height: ~~oDOM.documentElement.attributes.getNamedItem('height').value
|
||
|
}
|
||
|
|
||
|
page.render(options.outputDir + fs.separator + filename + '.png')
|
||
|
log('saved png: ' + filename + '.png')
|
||
|
}
|
||
|
|
||
|
if (options.svg) {
|
||
|
var serialize = new XMLSerializer()
|
||
|
fs.write(
|
||
|
options.outputDir + fs.separator + filename + '.svg'
|
||
|
, serialize.serializeToString(oDOM)
|
||
|
, 'w'
|
||
|
)
|
||
|
log('saved svg: ' + filename + '.svg')
|
||
|
}
|
||
|
})
|
||
|
|
||
|
phantom.exit()
|
||
|
|
||
|
function logger(_verbose) {
|
||
|
var verbose = _verbose
|
||
|
|
||
|
return function(_message, _level) {
|
||
|
var level = level
|
||
|
, message = _message
|
||
|
, log
|
||
|
|
||
|
log = level === 'error' ? system.stderr : system.stdout
|
||
|
|
||
|
if (verbose) {
|
||
|
log.write(message + '\n')
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
function traverse(obj){
|
||
|
var tree = []
|
||
|
|
||
|
tree.push(obj)
|
||
|
visit(obj)
|
||
|
|
||
|
function visit(node) {
|
||
|
if (node && node.hasChildNodes()) {
|
||
|
var child = node.firstChild
|
||
|
while (child) {
|
||
|
if (child.nodeType === 1 && child.nodeName != 'SCRIPT'){
|
||
|
tree.push(child)
|
||
|
visit(child)
|
||
|
}
|
||
|
child = child.nextSibling
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
return tree
|
||
|
}
|
||
|
|
||
|
function resolveSVGElement(element) {
|
||
|
var prefix = {
|
||
|
xmlns: "http://www.w3.org/2000/xmlns/"
|
||
|
, xlink: "http://www.w3.org/1999/xlink"
|
||
|
, svg: "http://www.w3.org/2000/svg"
|
||
|
}
|
||
|
, doctype = '<!DOCTYPE svg:svg PUBLIC'
|
||
|
+ ' "-//W3C//DTD XHTML 1.1 plus MathML 2.0 plus SVG 1.1//EN"'
|
||
|
+ ' "http://www.w3.org/2002/04/xhtml-math-svg/xhtml-math-svg.dtd">'
|
||
|
|
||
|
element.setAttribute("version", "1.1")
|
||
|
// removing attributes so they aren't doubled up
|
||
|
element.removeAttribute("xmlns")
|
||
|
element.removeAttribute("xlink")
|
||
|
// These are needed for the svg
|
||
|
if (!element.hasAttributeNS(prefix.xmlns, "xmlns")) {
|
||
|
element.setAttributeNS(prefix.xmlns, "xmlns", prefix.svg)
|
||
|
}
|
||
|
if (!element.hasAttributeNS(prefix.xmlns, "xmlns:xlink")) {
|
||
|
element.setAttributeNS(prefix.xmlns, "xmlns:xlink", prefix.xlink)
|
||
|
}
|
||
|
}
|
||
|
|
||
|
function resolveForeignObjects(element) {
|
||
|
var children
|
||
|
, textElement
|
||
|
, textSpan
|
||
|
|
||
|
if (element.tagName === 'foreignObject') {
|
||
|
textElement = document.createElement('text')
|
||
|
textSpan = document.createElement('tspan')
|
||
|
textSpan.setAttribute(
|
||
|
'style'
|
||
|
, 'font-size: 11.5pt; font-family: "sans-serif";'
|
||
|
)
|
||
|
textSpan.setAttribute('x', 0)
|
||
|
textSpan.setAttribute('y', 14.5)
|
||
|
textSpan.textContent = element.textContent
|
||
|
|
||
|
textElement.appendChild(textSpan)
|
||
|
element.parentElement.appendChild(textElement)
|
||
|
element.parentElement.removeChild(element)
|
||
|
}
|
||
|
}
|
||
|
|
||
|
// The sandboxed function that's executed in-page by phantom
|
||
|
function executeInPage(contents) {
|
||
|
var xmlSerializer = new XMLSerializer()
|
||
|
, toRemove
|
||
|
, el
|
||
|
, elContent
|
||
|
, svg
|
||
|
, svgValue
|
||
|
|
||
|
toRemove = document.getElementsByClassName('mermaid')
|
||
|
if (toRemove && toRemove.length) {
|
||
|
for (var i = 0, len = toRemove.length; i < len; i++) {
|
||
|
toRemove[i].parentNode.removeChild(toRemove[i])
|
||
|
}
|
||
|
}
|
||
|
|
||
|
el = document.createElement("div")
|
||
|
el.className = 'mermaid'
|
||
|
elContent = document.createTextNode(contents)
|
||
|
el.appendChild(elContent)
|
||
|
|
||
|
document.body.appendChild(el)
|
||
|
|
||
|
mermaid.init()
|
||
|
|
||
|
svg = document.querySelector('svg')
|
||
|
svgValue = xmlSerializer.serializeToString(svg)
|
||
|
|
||
|
return svgValue
|
||
|
}
|