ant-design/.dumi/theme/builtins/LocaleLink/index.tsx
二货爱吃白萝卜 3097ec39e1
fix: Site link auto switch by locale ()
* Revert "Revert "docs: auto trans link" ()"

This reverts commit fa90b25b3e.

* docs: site update link auto switch logic
2023-05-16 08:42:18 +08:00

50 lines
1.1 KiB
TypeScript

import { Link } from 'dumi';
import * as React from 'react';
import useLocale from '../../../hooks/useLocale';
type LinkProps = Parameters<typeof Link>[0];
export interface LocaleLinkProps extends LinkProps {
sourceType: 'a' | 'Link';
children?: React.ReactNode;
}
export default function LocaleLink({ sourceType, to, ...props }: LocaleLinkProps) {
const Component = sourceType === 'a' ? 'a' : Link;
const [, localeType] = useLocale();
const localeTo = React.useMemo(() => {
if (!to || typeof to !== 'string') {
return to;
}
// Auto locale switch
const cells = to.match(/(\/[^#]*)(#.*)?/);
if (cells) {
let path = cells[1].replace(/\/$/, '');
const hash = cells[2] || '';
if (localeType === 'cn' && !path.endsWith('-cn')) {
path = `${path}-cn`;
} else if (localeType === 'en' && path.endsWith('-cn')) {
path = path.replace(/-cn$/, '');
}
return `${path}${hash}`;
}
return to;
}, [to]);
const linkProps: LocaleLinkProps = {
...props,
} as LocaleLinkProps;
if (to) {
linkProps.to = localeTo;
}
return <Component {...linkProps} />;
}