|
|
@ -1,12 +1,136 @@ |
|
|
|
const mdPlantUml = require('markdown-it-plantuml') |
|
|
|
const zlib = require('zlib') |
|
|
|
|
|
|
|
// ------------------------------------
|
|
|
|
// Markdown - PlantUML Preprocessor
|
|
|
|
// ------------------------------------
|
|
|
|
|
|
|
|
module.exports = { |
|
|
|
init (md, conf) { |
|
|
|
md.use(mdPlantUml, { |
|
|
|
init (mdinst, conf) { |
|
|
|
mdinst.use((md, opts) => { |
|
|
|
const openMarker = opts.openMarker || '@startuml' |
|
|
|
const openChar = openMarker.charCodeAt(0) |
|
|
|
const closeMarker = opts.closeMarker || '@enduml' |
|
|
|
const closeChar = closeMarker.charCodeAt(0) |
|
|
|
const imageFormat = opts.imageFormat || 'svg' |
|
|
|
const server = opts.server || 'https://www.plantuml.com/plantuml' |
|
|
|
|
|
|
|
md.block.ruler.before('fence', 'uml_diagram', (state, startLine, endLine, silent) => { |
|
|
|
let nextLine |
|
|
|
let markup |
|
|
|
let params |
|
|
|
let token |
|
|
|
let i |
|
|
|
let autoClosed = false |
|
|
|
let start = state.bMarks[startLine] + state.tShift[startLine] |
|
|
|
let max = state.eMarks[startLine] |
|
|
|
|
|
|
|
// Check out the first character quickly,
|
|
|
|
// this should filter out most of non-uml blocks
|
|
|
|
//
|
|
|
|
if (openChar !== state.src.charCodeAt(start)) { return false } |
|
|
|
|
|
|
|
// Check out the rest of the marker string
|
|
|
|
//
|
|
|
|
for (i = 0; i < openMarker.length; ++i) { |
|
|
|
if (openMarker[i] !== state.src[start + i]) { return false } |
|
|
|
} |
|
|
|
|
|
|
|
markup = state.src.slice(start, start + i) |
|
|
|
params = state.src.slice(start + i, max) |
|
|
|
|
|
|
|
// Since start is found, we can report success here in validation mode
|
|
|
|
//
|
|
|
|
if (silent) { return true } |
|
|
|
|
|
|
|
// Search for the end of the block
|
|
|
|
//
|
|
|
|
nextLine = startLine |
|
|
|
|
|
|
|
for (;;) { |
|
|
|
nextLine++ |
|
|
|
if (nextLine >= endLine) { |
|
|
|
// unclosed block should be autoclosed by end of document.
|
|
|
|
// also block seems to be autoclosed by end of parent
|
|
|
|
break |
|
|
|
} |
|
|
|
|
|
|
|
start = state.bMarks[nextLine] + state.tShift[nextLine] |
|
|
|
max = state.eMarks[nextLine] |
|
|
|
|
|
|
|
if (start < max && state.sCount[nextLine] < state.blkIndent) { |
|
|
|
// non-empty line with negative indent should stop the list:
|
|
|
|
// - ```
|
|
|
|
// test
|
|
|
|
break |
|
|
|
} |
|
|
|
|
|
|
|
if (closeChar !== state.src.charCodeAt(start)) { |
|
|
|
// didn't find the closing fence
|
|
|
|
continue |
|
|
|
} |
|
|
|
|
|
|
|
if (state.sCount[nextLine] > state.sCount[startLine]) { |
|
|
|
// closing fence should not be indented with respect of opening fence
|
|
|
|
continue |
|
|
|
} |
|
|
|
|
|
|
|
var closeMarkerMatched = true |
|
|
|
for (i = 0; i < closeMarker.length; ++i) { |
|
|
|
if (closeMarker[i] !== state.src[start + i]) { |
|
|
|
closeMarkerMatched = false |
|
|
|
break |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
if (!closeMarkerMatched) { |
|
|
|
continue |
|
|
|
} |
|
|
|
|
|
|
|
// make sure tail has spaces only
|
|
|
|
if (state.skipSpaces(start + i) < max) { |
|
|
|
continue |
|
|
|
} |
|
|
|
|
|
|
|
// found!
|
|
|
|
autoClosed = true |
|
|
|
break |
|
|
|
} |
|
|
|
|
|
|
|
const contents = state.src |
|
|
|
.split('\n') |
|
|
|
.slice(startLine + 1, nextLine) |
|
|
|
.join('\n') |
|
|
|
|
|
|
|
// We generate a token list for the alt property, to mimic what the image parser does.
|
|
|
|
let altToken = [] |
|
|
|
// Remove leading space if any.
|
|
|
|
let alt = params ? params.slice(1) : 'uml diagram' |
|
|
|
state.md.inline.parse( |
|
|
|
alt, |
|
|
|
state.md, |
|
|
|
state.env, |
|
|
|
altToken |
|
|
|
) |
|
|
|
|
|
|
|
var zippedCode = encode64(zlib.deflateSync('@startuml\n' + contents + '\n@enduml').toString('binary')) |
|
|
|
|
|
|
|
token = state.push('uml_diagram', 'img', 0) |
|
|
|
// alt is constructed from children. No point in populating it here.
|
|
|
|
token.attrs = [ [ 'src', `${server}/${imageFormat}/${zippedCode}` ], [ 'alt', '' ], ['class', 'uml-diagram'] ] |
|
|
|
token.block = true |
|
|
|
token.children = altToken |
|
|
|
token.info = params |
|
|
|
token.map = [ startLine, nextLine ] |
|
|
|
token.markup = markup |
|
|
|
|
|
|
|
state.line = nextLine + (autoClosed ? 1 : 0) |
|
|
|
|
|
|
|
return true |
|
|
|
}, { |
|
|
|
alt: [ 'paragraph', 'reference', 'blockquote', 'list' ] |
|
|
|
}) |
|
|
|
md.renderer.rules.uml_diagram = md.renderer.rules.image |
|
|
|
}, { |
|
|
|
openMarker: conf.openMarker, |
|
|
|
closeMarker: conf.closeMarker, |
|
|
|
imageFormat: conf.imageFormat, |
|
|
@ -14,3 +138,52 @@ module.exports = { |
|
|
|
}) |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
function encode64 (data) { |
|
|
|
let r = '' |
|
|
|
for (let i = 0; i < data.length; i += 3) { |
|
|
|
if (i + 2 === data.length) { |
|
|
|
r += append3bytes(data.charCodeAt(i), data.charCodeAt(i + 1), 0) |
|
|
|
} else if (i + 1 === data.length) { |
|
|
|
r += append3bytes(data.charCodeAt(i), 0, 0) |
|
|
|
} else { |
|
|
|
r += append3bytes(data.charCodeAt(i), data.charCodeAt(i + 1), data.charCodeAt(i + 2)) |
|
|
|
} |
|
|
|
} |
|
|
|
return r |
|
|
|
} |
|
|
|
|
|
|
|
function append3bytes (b1, b2, b3) { |
|
|
|
let c1 = b1 >> 2 |
|
|
|
let c2 = ((b1 & 0x3) << 4) | (b2 >> 4) |
|
|
|
let c3 = ((b2 & 0xF) << 2) | (b3 >> 6) |
|
|
|
let c4 = b3 & 0x3F |
|
|
|
let r = '' |
|
|
|
r += encode6bit(c1 & 0x3F) |
|
|
|
r += encode6bit(c2 & 0x3F) |
|
|
|
r += encode6bit(c3 & 0x3F) |
|
|
|
r += encode6bit(c4 & 0x3F) |
|
|
|
return r |
|
|
|
} |
|
|
|
|
|
|
|
function encode6bit(b) { |
|
|
|
if (b < 10) { |
|
|
|
return String.fromCharCode(48 + b) |
|
|
|
} |
|
|
|
b -= 10 |
|
|
|
if (b < 26) { |
|
|
|
return String.fromCharCode(65 + b) |
|
|
|
} |
|
|
|
b -= 26 |
|
|
|
if (b < 26) { |
|
|
|
return String.fromCharCode(97 + b) |
|
|
|
} |
|
|
|
b -= 26 |
|
|
|
if (b === 0) { |
|
|
|
return '-' |
|
|
|
} |
|
|
|
if (b === 1) { |
|
|
|
return '_' |
|
|
|
} |
|
|
|
return '?' |
|
|
|
} |