31 lines
817 B
JavaScript
31 lines
817 B
JavaScript
|
import tokenizer from './customBlock'
|
||
|
|
import visit from 'unist-util-visit'
|
||
|
|
import { render } from './page'
|
||
|
|
|
||
|
|
const NAME = 'customblock'
|
||
|
|
|
||
|
|
export default function attacher () {
|
||
|
|
let proto = this.Parser.prototype
|
||
|
|
|
||
|
|
proto.blockTokenizers[NAME] = tokenizer
|
||
|
|
proto.interruptParagraph.push([NAME])
|
||
|
|
proto.interruptList.push([NAME])
|
||
|
|
proto.interruptBlockquote.push([NAME])
|
||
|
|
|
||
|
|
let methods = proto.blockMethods
|
||
|
|
methods.unshift(NAME)
|
||
|
|
|
||
|
|
return (tree, file) => {
|
||
|
|
let { path } = file
|
||
|
|
|
||
|
|
visit(tree, NAME, ({ className, value }, index, parent) => {
|
||
|
|
let { contents } = render(value, path, {})
|
||
|
|
className = className ? `${className} custom-block` : 'custom-block'
|
||
|
|
parent.children.splice(index, 1, {
|
||
|
|
type: 'html',
|
||
|
|
value: `<div class="${className}">${contents}</div>`
|
||
|
|
})
|
||
|
|
})
|
||
|
|
}
|
||
|
|
}
|