add immer

Ruben-Arushanyan · Oct 3, 2023 · 130c91c · 130c91c
1 parent 6c9672e
commit 130c91c
Show file tree

Hide file tree

Showing 3 changed files with 24 additions and 1 deletion.
diff --git a/package-lock.json b/package-lock.json
diff --git a/package.json b/package.json
@@ -57,6 +57,7 @@
         "jest": "^29.0.3"
     },
     "dependencies": {
+        "immer": "^10.0.3",
         "secure-event-emitter": "^4.3.5"
     }
 }
diff --git a/src/Store.js b/src/Store.js
@@ -1,6 +1,6 @@
 import { SingularEventEmitter } from 'secure-event-emitter'
 import { isFunction, eq, memoizeByArgs } from './utils'
-
+import { produce } from 'immer'
 
 class Store {
     #emitter_key = Symbol()
@@ -26,6 +26,13 @@ class Store {
         }
     }
 
+    updateStateImmer = (cb) => {
+        if (!isFunction(cb)) {
+            throw new TypeError('[[updateStateImmer()]] argument must be a function');
+        }
+        this.updateState(() => produce(this.state, cb))
+    }
+
     subscribe = (cb) => {
         cb = memoizeByArgs(cb)
         const _cb = (order) => {