2022-11-09 12:28:04 +08:00
|
|
|
import fs from 'fs';
|
|
|
|
import type { IApi, IRoute } from 'dumi';
|
|
|
|
import { extractStyle } from '@ant-design/cssinjs';
|
|
|
|
import ReactTechStack from 'dumi/dist/techStacks/react';
|
|
|
|
import sylvanas from 'sylvanas';
|
|
|
|
|
|
|
|
/**
|
|
|
|
* extends dumi internal tech stack, for customize previewer props
|
|
|
|
*/
|
|
|
|
class AntdReactTechStack extends ReactTechStack {
|
|
|
|
// eslint-disable-next-line class-methods-use-this
|
|
|
|
generatePreviewerProps(...[props, opts]: any) {
|
|
|
|
if (opts.type === 'external') {
|
|
|
|
// try to find md file with the same name as the demo tsx file
|
|
|
|
const locale = opts.mdAbsPath.match(/index\.([a-z-]+)\.md$/i)?.[1];
|
|
|
|
const mdPath = opts.fileAbsPath!.replace(/\.\w+$/, '.md');
|
|
|
|
const md = fs.existsSync(mdPath) ? fs.readFileSync(mdPath, 'utf-8') : '';
|
|
|
|
|
|
|
|
const codePath = opts.fileAbsPath!.replace(/\.\w+$/, '.tsx');
|
|
|
|
const code = fs.existsSync(codePath) ? fs.readFileSync(codePath, 'utf-8') : '';
|
|
|
|
|
|
|
|
props.jsx = sylvanas.parseText(code);
|
|
|
|
|
|
|
|
if (md) {
|
|
|
|
// extract description & css style from md file
|
|
|
|
const description = md.match(
|
|
|
|
new RegExp(`(?:^|\\n)## ${locale}([^]+?)(\\n## [a-z]|\\n\`\`\`|\\n<style>|$)`),
|
|
|
|
)?.[1];
|
2023-01-06 12:36:43 +08:00
|
|
|
const style = md.match(/\r?\n(?:```css|<style>)\r?\n([^]+?)\r?\n(?:```|<\/style>)/)?.[1];
|
2022-11-09 12:28:04 +08:00
|
|
|
|
|
|
|
props.description ??= description?.trim();
|
|
|
|
props.style ??= style;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return props;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
const resolve = (path: string): string => require.resolve(path);
|
|
|
|
|
|
|
|
const RoutesPlugin = (api: IApi) => {
|
|
|
|
const ssrCssFileName = `ssr-${Date.now()}.css`;
|
|
|
|
|
|
|
|
api.registerTechStack(() => new AntdReactTechStack());
|
|
|
|
|
2022-11-18 13:10:45 +08:00
|
|
|
api.modifyRoutes((routes) => {
|
2022-11-09 12:28:04 +08:00
|
|
|
// TODO: append extra routes, such as home, changelog, form-v3
|
|
|
|
|
|
|
|
const extraRoutesList: IRoute[] = [
|
|
|
|
{
|
|
|
|
id: 'changelog-cn',
|
|
|
|
path: 'changelog-cn',
|
|
|
|
absPath: '/changelog-cn',
|
|
|
|
parentId: 'DocLayout',
|
|
|
|
file: resolve('../../CHANGELOG.zh-CN.md'),
|
|
|
|
},
|
|
|
|
{
|
|
|
|
id: 'changelog',
|
|
|
|
path: 'changelog',
|
|
|
|
absPath: '/changelog',
|
|
|
|
parentId: 'DocLayout',
|
|
|
|
file: resolve('../../CHANGELOG.en-US.md'),
|
|
|
|
},
|
|
|
|
];
|
|
|
|
|
2022-11-18 13:10:45 +08:00
|
|
|
extraRoutesList.forEach((itemRoute) => {
|
2022-11-09 12:28:04 +08:00
|
|
|
routes[itemRoute.path] = itemRoute;
|
|
|
|
});
|
|
|
|
|
|
|
|
return routes;
|
|
|
|
});
|
|
|
|
|
2022-11-18 13:10:45 +08:00
|
|
|
api.modifyExportHTMLFiles((files) =>
|
2022-11-09 12:28:04 +08:00
|
|
|
files
|
|
|
|
// exclude dynamic route path, to avoid deploy failed by `:id` directory
|
2022-11-18 13:10:45 +08:00
|
|
|
.filter((f) => !f.path.includes(':'))
|
2022-11-09 12:28:04 +08:00
|
|
|
// FIXME: workaround to make emotion support react 18 pipeableStream
|
|
|
|
// ref: https://github.com/emotion-js/emotion/issues/2800#issuecomment-1221296308
|
2022-11-18 13:10:45 +08:00
|
|
|
.map((file) => {
|
2022-11-09 12:28:04 +08:00
|
|
|
let styles = '';
|
|
|
|
|
|
|
|
// extract all emotion style tags from body
|
2022-12-01 16:10:53 +08:00
|
|
|
file.content = file.content.replace(/<style data-emotion[\S\s]+?<\/style>/g, (s) => {
|
|
|
|
styles += s;
|
|
|
|
|
|
|
|
return '';
|
|
|
|
});
|
2022-11-09 12:28:04 +08:00
|
|
|
|
|
|
|
// insert emotion style tags to head
|
|
|
|
file.content = file.content.replace('</head>', `${styles}</head>`);
|
|
|
|
|
|
|
|
return file;
|
|
|
|
}),
|
|
|
|
);
|
|
|
|
|
|
|
|
// add ssr css file to html
|
2022-11-18 13:10:45 +08:00
|
|
|
api.modifyConfig((memo) => {
|
2022-11-09 12:28:04 +08:00
|
|
|
memo.styles ??= [];
|
|
|
|
memo.styles.push(`/${ssrCssFileName}`);
|
|
|
|
|
|
|
|
return memo;
|
|
|
|
});
|
|
|
|
|
|
|
|
// generate ssr css file
|
|
|
|
api.onBuildHtmlComplete(() => {
|
2022-11-24 11:14:29 +08:00
|
|
|
// FIXME: This should not be empty @peachScript
|
|
|
|
const styleCache = (global as any)?.styleCache;
|
|
|
|
const styleText = styleCache ? extractStyle(styleCache) : '';
|
2022-11-09 12:28:04 +08:00
|
|
|
const styleTextWithoutStyleTag = styleText
|
|
|
|
.replace(/<style\s[^>]*>/g, '')
|
|
|
|
.replace(/<\/style>/g, '');
|
|
|
|
|
|
|
|
fs.writeFileSync(`./_site/${ssrCssFileName}`, styleTextWithoutStyleTag, 'utf8');
|
|
|
|
});
|
|
|
|
};
|
|
|
|
|
|
|
|
export default RoutesPlugin;
|