useCallbackRef.ts

import { useLayoutEffect, useRef } from "react";
type AnyFunction = (...args: any[]) => any;
const useLayoutEffectRef = typeof window !== "undefined" ? useLayoutEffect : () => {};

export function useCallbackRef<TCallback extends AnyFunction>(callback: TCallback): TCallback {
  const latestRef = useRef<TCallback>(useCallbackRef_shouldNotBeInvokedBeforeMount as any);
  useLayoutEffectRef(() => {
    latestRef.current = callback;
  }, [callback]);

  const stableRef = useRef<TCallback>(null as any);
  if (!stableRef.current) {
    stableRef.current = function(this: any) {
      return latestRef.current.apply(this, arguments as any);
    } as TCallback;
  }

  return stableRef.current;
}

function useCallbackRef_shouldNotBeInvokedBeforeMount() {
  throw new Error("useCallbackRef Error");
}

export default useCallbackRef;