36 lines
940 B
TypeScript
36 lines
940 B
TypeScript
|
import { useMemo, useRef } from 'react';
|
||
|
|
|
||
|
|
// From https://github.com/alibaba/hooks/blob/master/packages/hooks/src/useMemoizedFn/index.ts
|
||
|
|
|
||
|
|
type Noop = (this: any, ...args: any[]) => any;
|
||
|
|
|
||
|
|
type PickFunction<T extends Noop> = (
|
||
|
|
this: ThisParameterType<T>,
|
||
|
|
...args: Parameters<T>
|
||
|
|
) => ReturnType<T>;
|
||
|
|
|
||
|
|
export function useEvent<T extends Noop>(fn: T) {
|
||
|
|
if (process.env.NODE_ENV === 'development') {
|
||
|
|
if (typeof fn !== 'function') {
|
||
|
|
console.error(
|
||
|
|
`useMemoizedFn expected parameter is a function, got ${typeof fn}`
|
||
|
|
);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
const fnRef = useRef<T>(fn);
|
||
|
|
|
||
|
|
// why not write `fnRef.current = fn`?
|
||
|
|
// https://github.com/alibaba/hooks/issues/728
|
||
|
|
fnRef.current = useMemo(() => fn, [fn]);
|
||
|
|
|
||
|
|
const memoizedFn = useRef<PickFunction<T>>();
|
||
|
|
if (!memoizedFn.current) {
|
||
|
|
memoizedFn.current = function (this, ...args) {
|
||
|
|
return fnRef.current.apply(this, args);
|
||
|
|
};
|
||
|
|
}
|
||
|
|
|
||
|
|
return memoizedFn.current as T;
|
||
|
|
}
|