46 lines
1.2 KiB
JavaScript
46 lines
1.2 KiB
JavaScript
import { visit } from 'unist-util-visit'
|
|
import { selectAll } from 'hast-util-select'
|
|
import { parseSelector } from 'hast-util-parse-selector'
|
|
|
|
/*
|
|
* Attacher
|
|
*/
|
|
export default (options) => {
|
|
const selector = options.selector || options.select || 'body'
|
|
const wrapper = options.wrapper || options.wrap
|
|
const rewrite = options.rewrite
|
|
const additionalAttributes = options.wrapperAdditionalAttributes
|
|
|
|
/*
|
|
* Transformer
|
|
*/
|
|
return (tree) => {
|
|
if (wrapper && typeof wrapper !== 'string') {
|
|
throw new TypeError('Expected a `string` as wrapper')
|
|
}
|
|
|
|
if (typeof selector !== 'string') {
|
|
throw new TypeError('Expected a `string` as selector')
|
|
}
|
|
|
|
for (const match of selectAll(selector, tree)) {
|
|
visit(tree, match, (node, i, parent) => {
|
|
const parsedWrapper = parseSelector(wrapper)
|
|
if (additionalAttributes) {
|
|
parsedWrapper.properties = Object.assign(
|
|
{},
|
|
parsedWrapper.properties || {},
|
|
additionalAttributes
|
|
)
|
|
}
|
|
if (rewrite) {
|
|
parsedWrapper.children = node.children
|
|
} else {
|
|
parsedWrapper.children = [node]
|
|
}
|
|
parent.children[i] = parsedWrapper
|
|
})
|
|
}
|
|
}
|
|
}
|