mirror of
https://github.com/ant-design/ant-design.git
synced 2024-11-28 13:09:40 +08:00
ffbb19e090
* chore: enable tsc for demos * chore: site lint * chore: fix lint
69 lines
2.1 KiB
TypeScript
69 lines
2.1 KiB
TypeScript
import assert from 'assert';
|
|
import { type HastRoot, type UnifiedTransformer, unistUtilVisit } from 'dumi';
|
|
|
|
/**
|
|
* plugin for modify hast tree when docs compiling
|
|
*/
|
|
function rehypeAntd(): UnifiedTransformer<HastRoot> {
|
|
return (tree, vFile) => {
|
|
const { filename } = vFile.data.frontmatter as any;
|
|
|
|
unistUtilVisit.visit(tree, 'element', (node) => {
|
|
if (node.tagName === 'DumiDemoGrid') {
|
|
// replace DumiDemoGrid to DemoWrapper, to implement demo toolbar
|
|
node.tagName = 'DemoWrapper';
|
|
} else if (node.tagName === 'ResourceCards') {
|
|
const propNames = ['title', 'cover', 'description', 'src', 'official'];
|
|
const contentNode = node.children[0];
|
|
|
|
assert(
|
|
contentNode.type === 'text',
|
|
`ResourceCards content must be plain text!\nat ${filename}`,
|
|
);
|
|
|
|
// clear children
|
|
node.children = [];
|
|
|
|
// generate JSX props
|
|
(node as any).JSXAttributes = [
|
|
{
|
|
type: 'JSXAttribute',
|
|
name: 'resources',
|
|
value: JSON.stringify(
|
|
contentNode.value
|
|
.trim()
|
|
.split('\n')
|
|
.reduce<any>((acc, cur) => {
|
|
// match text from ` - 桌面组件 Sketch 模板包`
|
|
const [, isProp, val] = cur.match(/(\s+)?-\s(.+)/)!;
|
|
|
|
if (!isProp) {
|
|
// create items when match title
|
|
acc.push({ [propNames[0]]: val });
|
|
} else {
|
|
// add props when match others
|
|
const prev = acc[acc.length - 1];
|
|
|
|
prev[propNames[Object.keys(prev).length]] = val;
|
|
}
|
|
|
|
return acc;
|
|
}, []),
|
|
),
|
|
},
|
|
];
|
|
} else if (
|
|
node.type === 'element' &&
|
|
node.tagName === 'Table' &&
|
|
/^components/.test(filename)
|
|
) {
|
|
if (!node.properties) return;
|
|
node.properties.className ??= [];
|
|
(node.properties.className as string[]).push('component-api-table');
|
|
}
|
|
});
|
|
};
|
|
}
|
|
|
|
export default rehypeAntd;
|