semi-design/packages/semi-ui/_utils/hooks/usePrevFocus.ts

import { useState, useEffect } from 'react';
import { getActiveElement } from '../index';
import { get, isFunction } from 'lodash';

export function usePrevFocus() {
    const [prevFocusElement, setPrevFocus] = useState<HTMLElement>(getActiveElement());

    useEffect(() => {
        return function cleanup() {
            const blur = get(prevFocusElement, 'blur');
            isFunction(blur) && blur();
        };
    }, [prevFocusElement]);

    return [prevFocusElement, setPrevFocus];
}