feat(react): Support Mutiple useOnSelectionChange Hook subscribers
This commit is contained in:
@@ -4,14 +4,19 @@ import { useStoreApi } from './useStore';
|
||||
import type { OnSelectionChangeFunc } from '../types';
|
||||
|
||||
export type UseOnSelectionChangeOptions = {
|
||||
onChange?: OnSelectionChangeFunc;
|
||||
onChange: OnSelectionChangeFunc;
|
||||
};
|
||||
|
||||
function useOnSelectionChange({ onChange }: UseOnSelectionChangeOptions) {
|
||||
const store = useStoreApi();
|
||||
|
||||
useEffect(() => {
|
||||
store.setState({ onSelectionChange: onChange });
|
||||
const addedOnSelectionChange = [...store.getState().onSelectionChange, onChange];
|
||||
store.setState({ onSelectionChange: addedOnSelectionChange });
|
||||
return () => {
|
||||
const removedOnSelectionChange = store.getState().onSelectionChange.filter((fn) => fn !== onChange);
|
||||
store.setState({ onSelectionChange: removedOnSelectionChange });
|
||||
};
|
||||
}, [onChange]);
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user