27 lines
612 B
TypeScript
27 lines
612 B
TypeScript
|
import * as React from 'react';
|
||
|
|
|
||
|
|
// Restore focus on teardown
|
||
|
|
export const useRestoreFocus = (
|
||
|
|
// The ref for the element to receive initial focus
|
||
|
|
focusRef: React.RefObject<any>,
|
||
|
|
// Allow for an optional root element that must exist
|
||
|
|
root: boolean | HTMLElement | null = true
|
||
|
|
) => {
|
||
|
|
React.useEffect(() => {
|
||
|
|
if (!root) {
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
|
||
|
|
const lastFocused = document.activeElement as any;
|
||
|
|
if (focusRef.current) {
|
||
|
|
focusRef.current.focus();
|
||
|
|
}
|
||
|
|
|
||
|
|
return () => {
|
||
|
|
if (lastFocused && lastFocused.focus) {
|
||
|
|
lastFocused.focus();
|
||
|
|
}
|
||
|
|
};
|
||
|
|
}, [focusRef, root]);
|
||
|
|
};
|