You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
51 lines
1.4 KiB
51 lines
1.4 KiB
/**
|
|
* Copyright (c) Meta Platforms, Inc. and affiliates.
|
|
*
|
|
* This source code is licensed under the MIT license found in the
|
|
* LICENSE file in the root directory of this source tree.
|
|
*
|
|
*
|
|
*/
|
|
import createEventHandle from '../createEventHandle';
|
|
import useLayoutEffect from '../useLayoutEffect';
|
|
import useStable from '../useStable';
|
|
|
|
/**
|
|
* This can be used with any event type include custom events.
|
|
*
|
|
* const click = useEvent('click', options);
|
|
* useEffect(() => {
|
|
* click.setListener(target, onClick);
|
|
* return () => click.clear();
|
|
* }).
|
|
*/
|
|
export default function useEvent(event, options) {
|
|
var targetListeners = useStable(() => new Map());
|
|
var addListener = useStable(() => {
|
|
var addEventListener = createEventHandle(event, options);
|
|
return (target, callback) => {
|
|
var removeTargetListener = targetListeners.get(target);
|
|
|
|
if (removeTargetListener != null) {
|
|
removeTargetListener();
|
|
}
|
|
|
|
if (callback == null) {
|
|
targetListeners.delete(target);
|
|
}
|
|
|
|
var removeEventListener = addEventListener(target, callback);
|
|
targetListeners.set(target, removeEventListener);
|
|
return removeEventListener;
|
|
};
|
|
});
|
|
useLayoutEffect(() => {
|
|
return () => {
|
|
targetListeners.forEach(removeListener => {
|
|
removeListener();
|
|
});
|
|
targetListeners.clear();
|
|
};
|
|
}, [targetListeners]);
|
|
return addListener;
|
|
} |