mirror of
https://github.com/ant-design/ant-design.git
synced 2024-12-04 00:49:39 +08:00
54 lines
1.4 KiB
TypeScript
54 lines
1.4 KiB
TypeScript
import type { MouseEvent } from 'react';
|
|
import React, { forwardRef, useLayoutEffect, useMemo, useTransition } from 'react';
|
|
import { useLocation, useNavigate } from 'dumi';
|
|
import nprogress from 'nprogress';
|
|
|
|
export type LinkProps = {
|
|
to?: string | { pathname?: string; search?: string; hash?: string };
|
|
children?: React.ReactNode;
|
|
style?: React.CSSProperties;
|
|
className?: string;
|
|
};
|
|
|
|
const Link = forwardRef<HTMLAnchorElement, LinkProps>((props, ref) => {
|
|
const { to, children, ...rest } = props;
|
|
const [isPending, startTransition] = useTransition();
|
|
const navigate = useNavigate();
|
|
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>) => {
|
|
if (!href.startsWith('http')) {
|
|
// Should support open in new tab
|
|
if (!e.metaKey && !e.ctrlKey && !e.shiftKey) {
|
|
e.preventDefault();
|
|
startTransition(() => {
|
|
navigate(href);
|
|
});
|
|
}
|
|
}
|
|
};
|
|
|
|
useLayoutEffect(() => {
|
|
if (isPending) {
|
|
nprogress.start();
|
|
} else {
|
|
nprogress.done();
|
|
}
|
|
}, [isPending]);
|
|
|
|
return (
|
|
<a ref={ref} href={href} onClick={handleClick} {...rest}>
|
|
{children}
|
|
</a>
|
|
);
|
|
});
|
|
|
|
export default Link;
|