-
-
Notifications
You must be signed in to change notification settings - Fork 215
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
docs: enable prefetch for dumi Link (#315)
- Loading branch information
Showing
9 changed files
with
56 additions
and
66 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,72 +1,56 @@ | ||
| import { Link as DumiLink, useLocation, useNavigate } from 'dumi'; | ||
| import nprogress from 'nprogress'; | ||
| import { Link as DumiLink, useAppData, useLocation, useNavigate } from 'dumi'; | ||
| import type { MouseEvent, MouseEventHandler } from 'react'; | ||
| import React, { forwardRef, useLayoutEffect, useTransition } from 'react'; | ||
| import React, { useMemo, forwardRef } from 'react'; | ||
|
|
||
| export interface LinkProps { | ||
| to: string | { pathname?: string; search?: string; hash?: string }; | ||
| style?: React.CSSProperties; | ||
| className?: string; | ||
| onClick?: MouseEventHandler; | ||
| component?: React.ComponentType<any>; | ||
| children?: React.ReactNode; | ||
| } | ||
|
|
||
| nprogress.configure({ showSpinner: false }); | ||
|
|
||
| const Link = forwardRef<HTMLAnchorElement, React.PropsWithChildren<LinkProps>>((props, ref) => { | ||
| const { to, children, component, ...rest } = props; | ||
| const [isPending, startTransition] = useTransition(); | ||
| const navigate = useNavigate(); | ||
| const { pathname } = useLocation(); | ||
|
|
||
| const href = React.useMemo<string>(() => { | ||
| if (typeof to === 'object') { | ||
| return `${to.pathname || pathname}${to.search || ''}${to.hash || ''}`; | ||
| } | ||
| return to; | ||
| }, [to]); | ||
|
|
||
| const handleClick = (e: MouseEvent<HTMLAnchorElement>) => { | ||
| props.onClick?.(e); | ||
| if (!href?.startsWith('http')) { | ||
| // Should support open in new tab | ||
| if (!e.metaKey && !e.ctrlKey && !e.shiftKey) { | ||
| e.preventDefault(); | ||
| startTransition(() => { | ||
| if (href) { | ||
| navigate(href); | ||
| } | ||
| }); | ||
| const Link = forwardRef<HTMLAnchorElement, React.PropsWithChildren<LinkProps>>( | ||
| ({ component, children, to, ...rest }, ref) => { | ||
| const { pathname } = useLocation(); | ||
| const { preloadRoute } = useAppData(); | ||
| const navigate = useNavigate(); | ||
| const href = useMemo<string>(() => { | ||
| if (typeof to === 'object') { | ||
| return `${to.pathname || pathname}${to.search || ''}${to.hash || ''}`; | ||
| } | ||
| return to; | ||
| }, [to]); | ||
| const onClick = (e: MouseEvent<HTMLAnchorElement>) => { | ||
| rest.onClick?.(e); | ||
| if (!href?.startsWith('http')) { | ||
| // Should support open in new tab | ||
| if (!e.metaKey && !e.ctrlKey && !e.shiftKey) { | ||
| e.preventDefault(); | ||
| navigate(href); | ||
| } | ||
| } | ||
| }; | ||
| if (component) { | ||
| return React.createElement( | ||
| component, | ||
| { | ||
| ...rest, | ||
| ref, | ||
| href, | ||
| onClick, | ||
| onMouseEnter: () => preloadRoute?.(href), | ||
| }, | ||
| children, | ||
| ); | ||
| } | ||
| }; | ||
|
|
||
| useLayoutEffect(() => { | ||
| if (isPending) { | ||
| nprogress.start(); | ||
| } else { | ||
| nprogress.done(); | ||
| } | ||
| }, [isPending]); | ||
|
|
||
| if (component) { | ||
| return React.createElement( | ||
| component, | ||
| { | ||
| ...rest, | ||
| ref, | ||
| onClick: handleClick, | ||
| href, | ||
| }, | ||
| children, | ||
| return ( | ||
| <DumiLink ref={ref} {...rest} to={href} prefetch> | ||
| {children} | ||
| </DumiLink> | ||
| ); | ||
| } | ||
|
|
||
| return ( | ||
| <DumiLink ref={ref} onClick={handleClick} {...rest} to={href} prefetch> | ||
| {children} | ||
| </DumiLink> | ||
| ); | ||
| }); | ||
| }, | ||
| ); | ||
|
|
||
| export default Link; |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters