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

54 lines
1.4 KiB
TypeScript
Raw Normal View History

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