Fix types and use more shared utils

mntm-lib · Feb 6, 2021 · a99c7e0 · a99c7e0
1 parent b29861e
commit a99c7e0
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 7 deletions.
diff --git a/package.json b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "@mntm/precoil",
-  "version": "1.4.0",
+  "version": "1.5.0",
   "main": "dist/index.js",
   "module": "dist/index.js",
   "jsnext:main": "dist/index.js",

diff --git a/src/store.ts b/src/store.ts
@@ -1,7 +1,8 @@
-import mitt from 'mitt';
-
 import type { Store, Atom, AtomUpdater, AtomValOrUpdater, Selector } from './types';
 
+import { weakUniqueId } from '@mntm/shared';
+
+import mitt from 'mitt';
 export const updater = mitt();
 
 export const store: Store = {};
@@ -27,9 +28,7 @@ export const setter = <T>(key: string, value: AtomValOrUpdater<T>): T => {
   return next;
 };
 
-let id = 0;
-const generateKey = () => (++id) + Math.floor(Math.random() * 1E6).toString(32);
-export const atom = <T>(defaultValue: T, key = generateKey()): Atom<T> => {
+export const atom = <T>(defaultValue: T, key = weakUniqueId()): Atom<T> => {
   store[key] = defaultValue;
 
   return {
@@ -38,7 +37,7 @@ export const atom = <T>(defaultValue: T, key = generateKey()): Atom<T> => {
     get() {
       return getter(key);
     },
-    set(value: T) {
+    set(value: AtomValOrUpdater<T>) {
       return setter(key, value);
     }
   };