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 <dai-shi@users.noreply.github.com>

* Update useSetAtom.ts

* move warning to store.ts

* move the warning in getAtomState

---------

Co-authored-by: Daishi Kato <dai-shi@users.noreply.github.com>
Co-authored-by: daishi <daishi@axlight.com>

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 }