ant-design/.dumi/theme/common/Link.tsx

59 lines
1.5 KiB
TypeScript
Raw Normal View History

import type { MouseEvent, MouseEventHandler } from 'react';
import React, { forwardRef, useLayoutEffect, useTransition } from 'react';
2023-08-02 14:04:29 +08:00
import { useLocation, useNavigate } from 'dumi';
import nprogress from 'nprogress';
2023-09-17 23:32:14 +08:00
export interface LinkProps {
to: string | { pathname?: string; search?: string; hash?: string };
style?: React.CSSProperties;
2023-04-18 15:29:34 +08:00
className?: string;
onClick?: MouseEventHandler;
2023-09-17 23:32:14 +08:00
}
nprogress.configure({ showSpinner: false });
const Link = forwardRef<HTMLAnchorElement, React.PropsWithChildren<LinkProps>>((props, ref) => {
2023-04-18 15:29:34 +08:00
const { to, children, ...rest } = props;
2023-08-02 14:04:29 +08:00
const [isPending, startTransition] = useTransition();
const navigate = useNavigate();
2023-08-02 14:04:29 +08:00
const { pathname } = useLocation();
const href = React.useMemo<string>(() => {
2023-08-02 14:04:29 +08:00
if (typeof to === 'object') {
return `${to.pathname || pathname}${to.search || ''}${to.hash || ''}`;
}
return to;
}, [to]);
const handleClick = (e: MouseEvent<HTMLAnchorElement>) => {
props.onClick?.(e);
2023-09-17 23:32:14 +08:00
if (!href?.startsWith('http')) {
// Should support open in new tab
if (!e.metaKey && !e.ctrlKey && !e.shiftKey) {
e.preventDefault();
startTransition(() => {
2023-09-17 23:32:14 +08:00
if (href) {
navigate(href);
}
});
}
}
};
2023-08-02 14:04:29 +08:00
useLayoutEffect(() => {
if (isPending) {
nprogress.start();
} else {
nprogress.done();
}
}, [isPending]);
return (
<a ref={ref} onClick={handleClick} {...rest} href={href}>
{children}
</a>
);
});
export default Link;