Throw error on useAtom(undefined) or useAtom(null) (pmndrs#2778)

* Fix undefined or null atoms * Simplify conditional, only in dev mode Co-authored-by: Daishi Kato <[email protected]> * Update useSetAtom.ts * move warning to store.ts * move the warning in getAtomState --------- Co-authored-by: Daishi Kato <[email protected]> Co-authored-by: daishi <[email protected]>
dmaskasky · Nov 14, 2024 · db988fe · db988fe
1 parent 33260c6
commit db988fe
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/vanilla/store.ts b/src/vanilla/store.ts
@@ -695,6 +695,9 @@ const buildStore = (
 export const createStore = (): Store => {
   const atomStateMap = new WeakMap()
   const getAtomState = <Value>(atom: Atom<Value>) => {
+    if (import.meta.env?.MODE !== 'production' && !atom) {
+      throw new Error('Atom is undefined or null')
+    }
     let atomState = atomStateMap.get(atom) as AtomState<Value> | undefined
     if (!atomState) {
       atomState = { d: new Map(), p: new Set(), n: 0 }