mirror of
https://gitee.com/ant-design/ant-design.git
synced 2024-12-05 05:28:20 +08:00
ac5a06027e
This reverts commit ea8ed28209
.
29 lines
914 B
TypeScript
29 lines
914 B
TypeScript
import { composeRef } from 'rc-util/lib/ref';
|
|
import * as React from 'react';
|
|
import { FormContext } from '../context';
|
|
import type { InternalNamePath } from '../interface';
|
|
|
|
export default function useItemRef() {
|
|
const { itemRef } = React.useContext(FormContext);
|
|
const cacheRef = React.useRef<{
|
|
name?: string;
|
|
originRef?: React.Ref<any>;
|
|
ref?: React.Ref<any>;
|
|
}>({});
|
|
|
|
function getRef(name: InternalNamePath, children: any) {
|
|
const childrenRef: React.Ref<React.ReactElement> =
|
|
children && typeof children === 'object' && children.ref;
|
|
const nameStr = name.join('_');
|
|
if (cacheRef.current.name !== nameStr || cacheRef.current.originRef !== childrenRef) {
|
|
cacheRef.current.name = nameStr;
|
|
cacheRef.current.originRef = childrenRef;
|
|
cacheRef.current.ref = composeRef(itemRef(name), childrenRef);
|
|
}
|
|
|
|
return cacheRef.current.ref;
|
|
}
|
|
|
|
return getRef;
|
|
}
|