ant-design-vue/components/_util/createRef.ts
2020-10-17 12:14:13 +08:00

32 lines
619 B
TypeScript

export interface RefObject extends Function {
current?: any;
}
function createRef(): any {
const func: RefObject = (node: any) => {
func.current = node;
};
return func;
}
export function fillRef<T>(ref, node: T) {
if (typeof ref === 'function') {
ref(node);
} else if (typeof ref === 'object' && ref && 'current' in ref) {
(ref as any).current = node;
}
}
/**
* Merge refs into one ref function to support ref passing.
*/
export function composeRef<T>(...refs: any[]) {
return (node: T) => {
refs.forEach(ref => {
fillRef(ref, node);
});
};
}
export default createRef;