blob: ee3f19159347ebf643d8dea6110ed9158cf2bf0e (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
|
import React, { useCallback, useEffect, useRef } from 'react';
export function useMounted() {
const mountedRef = useRef(false);
const isMounted = useCallback(() => mountedRef.current, []);
useEffect(() => {
mountedRef.current = true;
return () => {
mountedRef.current = false;
};
}, []);
return isMounted;
}
export function useCombinedRefs<T>(...refs: (React.Ref<T> | undefined)[]): React.RefCallback<T> {
return useCallback((element: T | null) => refs.forEach((ref) => assignToRef(element, ref)), []);
}
export function assignToRef<T>(element: T | null, ref?: React.Ref<T>) {
if (typeof ref === 'function') {
ref(element);
} else if (ref && element) {
(ref as React.MutableRefObject<T>).current = element;
}
}
|
