-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathindex.js
48 lines (38 loc) · 1.21 KB
/
index.js
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
import { useEffect, useReducer, useRef } from "react";
export function createCat(initialValue) {
let value = initialValue;
const listeners = new Set();
const get = () => value;
const set = (newValue) => {
if (newValue !== value) {
const oldValue = value;
value = newValue;
listeners.forEach((listener) => listener(oldValue, newValue));
}
};
const reset = () => set(initialValue);
const subscribe = (listener) => {
listeners.add(listener);
return () => listeners.delete(listener);
};
return { get, set, reset, subscribe };
}
export function useCat(cat, selector = (value) => value) {
const [, forceUpdate] = useReducer((x) => x + 1, 0);
const selectorRef = useRef(selector);
useEffect(() => {
selectorRef.current = selector;
}, [selector]);
useEffect(() => {
const handler = (oldValue, newValue) => {
const oldSlice = selectorRef.current(oldValue);
const newSlice = selectorRef.current(newValue);
if (oldSlice !== newSlice) {
forceUpdate();
}
};
const unsubscribe = cat.subscribe(handler);
return () => unsubscribe();
}, [cat]);
return selector(cat.get());
}