fix

Author: zbeyens

.changeset/loud-crabs-raise.md

@@ -0,0 +1,5 @@
+---
+'zustand-x': patch
+---
+
+Fixes #102

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

@@ -248,6 +248,24 @@ describe('zustandX', () => {
       extendedStore.set('validName', 'foo ');
       expect(extendedStore.get('name')).toBe('foo foo');
     });
+
+    it('should pass all parameters to actions', () => {
+      const another = createStore(
+        {
+          value: '',
+        },
+        {
+          name: 'test-params',
+        }
+      ).extendActions(({ set }) => ({
+        setValue: (a: string, b: string, c: string) => {
+          set('value', `${a}-${b}-${c}`);
+        },
+      }));
+
+      another.set('setValue', 'a', 'b', 'c');
+      expect(another.get('value')).toBe('a-b-c');
+    });
   });
 
   describe('when extending selectors', () => {

packages/zustand-x/src/utils/extendActions.ts

@@ -25,17 +25,17 @@ export const extendActions = <
     actions,
     set: <K extends keyof (StateType | TActions)>(
       key: K,
-      value: K extends keyof TActions
-        ? Parameters<TActions[K]>[0]
+      ...args: K extends keyof TActions
+        ? Parameters<TActions[K]>
         : K extends keyof StateType
-          ? StateType[K]
+          ? [StateType[K]]
           : never
     ) => {
       if (key in actions) {
         const action = actions[key as keyof typeof actions];
-        return action(value);
+        return action(...args);
       }
-      return api.set(key as any, value);
+      return api.set(key as any, args[0]);
     },
   };
 };