ScreenFocusHook.js

import {  useEffect, useRef } from 'react';

const ScreenFocusHook = (navigation) => {
    const _didFocusSubscription = useRef();
    const callBack = useRef();
    useEffect(() => {
        _didFocusSubscription.current = navigation.addListener(
            'didFocus',
            payload => {
                console.log('focus');
                callBack.current()
            }
        );
        return () => {
            _didFocusSubscription.current && _didFocusSubscription.current.remove();

        }
    }, []);

    
    const setCallBack = cb => {
        callBack.current = cb
    }
    return [
        setCallBack
    ]
}

export default ScreenFocusHook;