Merge pull request #115 from RavenColEvol/feat-set-callback-support

Author: zbeyens

.changeset/sweet-buses-behave.md

@@ -0,0 +1,5 @@
+---
+"zustand-x": patch
+---
+
+fix: add support for set callback support

.changeset/tall-brooms-enjoy.md

@@ -0,0 +1,5 @@
+---
+'zustand-x': patch
+---
+
+fix support for set callback value

package.json

@@ -88,6 +88,7 @@
     "eslint-plugin-unicorn": "^49.0.0",
     "eslint-plugin-unused-imports": "^3.0.0",
     "eslint-plugin-vitest": "0.5.4",
+    "glob": "8.1.0",
     "jsdom": "26.0.0",
     "prettier": "^3.1.0",
     "react": "^18.2.0",

packages/zustand-x/src/createStore.ts

@@ -149,6 +149,9 @@ export const createStore = <
     const typedKey = key as keyof StateType;
     const prevValue = store.getState()[typedKey];
 
+    if (typeof value === 'function') {
+      value = value(prevValue);
+    }
     if (prevValue === value) return;
 
     const actionKey = key.replace(/^\S/, (s) => s.toUpperCase());

packages/zustand-x/src/tests/createStore.spec.ts

@@ -285,6 +285,9 @@ describe('zustandX', () => {
         validName: (name: string) => {
           set('name', name.trim());
         },
+        validNameWithCallback: (lastName: string) => {
+          set('name', (firstName) => firstName + ' ' + lastName);
+        },
         replace2ndTag: (tag: string) => {
           set('state', (draft) => {
             draft.tags?.splice(1, 1, tag);
@@ -311,6 +314,11 @@ describe('zustandX', () => {
       expect(store.get('name')).toBe('test');
     });
 
+    it('should execute action with callback', () => {
+      store.actions.validNameWithCallback('test');
+      expect(store.get('name')).toBe('test test');
+    });
+
     it('should execute action with immer', () => {
       store.actions.replace2ndTag('tag3');
       expect(store.get('tags')).toEqual(['tag1', 'tag3']);

packages/zustand-x/src/types/store.ts

@@ -49,6 +49,10 @@ export type TStoreApiSet<
     key: K,
     ...args: Parameters<TActions[K]>
   ): ReturnType<TActions[K]>;
+  <K extends keyof StateType>(
+    key: K,
+    callback: (prevVal: StateType[K]) => StateType[K]
+  ): void;
   (
     key: 'state',
     value: Parameters<TCreatedStoreType<StateType, Mutators>['setState']>[0]

yarn.lock

@@ -5098,6 +5098,19 @@ __metadata:
   languageName: node
   linkType: hard
 
+"glob@npm:8.1.0, glob@npm:^8.0.0, glob@npm:^8.0.3":
+  version: 8.1.0
+  resolution: "glob@npm:8.1.0"
+  dependencies:
+    fs.realpath: "npm:^1.0.0"
+    inflight: "npm:^1.0.4"
+    inherits: "npm:2"
+    minimatch: "npm:^5.0.1"
+    once: "npm:^1.3.0"
+  checksum: 10c0/cb0b5cab17a59c57299376abe5646c7070f8acb89df5595b492dba3bfb43d301a46c01e5695f01154e6553168207cb60d4eaf07d3be4bc3eb9b0457c5c561d0f
+  languageName: node
+  linkType: hard
+
 "glob@npm:^10.2.2, glob@npm:^10.3.10, glob@npm:^10.3.7":
   version: 10.3.10
   resolution: "glob@npm:10.3.10"
@@ -5127,19 +5140,6 @@ __metadata:
   languageName: node
   linkType: hard
 
-"glob@npm:^8.0.0, glob@npm:^8.0.3":
-  version: 8.1.0
-  resolution: "glob@npm:8.1.0"
-  dependencies:
-    fs.realpath: "npm:^1.0.0"
-    inflight: "npm:^1.0.4"
-    inherits: "npm:2"
-    minimatch: "npm:^5.0.1"
-    once: "npm:^1.3.0"
-  checksum: 10c0/cb0b5cab17a59c57299376abe5646c7070f8acb89df5595b492dba3bfb43d301a46c01e5695f01154e6553168207cb60d4eaf07d3be4bc3eb9b0457c5c561d0f
-  languageName: node
-  linkType: hard
-
 "glob@npm:^9.3.1":
   version: 9.3.5
   resolution: "glob@npm:9.3.5"
@@ -8802,6 +8802,7 @@ __metadata:
     eslint-plugin-unicorn: "npm:^49.0.0"
     eslint-plugin-unused-imports: "npm:^3.0.0"
     eslint-plugin-vitest: "npm:0.5.4"
+    glob: "npm:8.1.0"
     jsdom: "npm:26.0.0"
     prettier: "npm:^3.1.0"
     react: "npm:^18.2.0"