40 lines
1.2 KiB
JavaScript
40 lines
1.2 KiB
JavaScript
import { useState } from 'react';
|
|
/**
|
|
* creates a MutableRef with ref change callback
|
|
* @param initialValue - initial ref value
|
|
* @param {Function} callback - a callback to run when value changes
|
|
*
|
|
* @example
|
|
* const ref = useCallbackRef(0, (newValue, oldValue) => console.log(oldValue, '->', newValue);
|
|
* ref.current = 1;
|
|
* // prints 0 -> 1
|
|
*
|
|
* @see https://reactjs.org/docs/hooks-reference.html#useref
|
|
* @see https://github.com/theKashey/use-callback-ref#usecallbackref---to-replace-reactuseref
|
|
* @returns {MutableRefObject}
|
|
*/
|
|
export function useCallbackRef(initialValue, callback) {
|
|
var ref = useState(function () { return ({
|
|
// value
|
|
value: initialValue,
|
|
// last callback
|
|
callback: callback,
|
|
// "memoized" public interface
|
|
facade: {
|
|
get current() {
|
|
return ref.value;
|
|
},
|
|
set current(value) {
|
|
var last = ref.value;
|
|
if (last !== value) {
|
|
ref.value = value;
|
|
ref.callback(value, last);
|
|
}
|
|
},
|
|
},
|
|
}); })[0];
|
|
// update callback
|
|
ref.callback = callback;
|
|
return ref.facade;
|
|
}
|