[@xstate/store] Fix createStoreWithProducer(…) types (#4890)

* Fix createStoreWithProducer() types

* Changeset

* Update packages/xstate-store/src/store.ts

Co-authored-by: Mateusz Burzyński <mateuszburzynski@gmail.com>

---------

Co-authored-by: Mateusz Burzyński <mateuszburzynski@gmail.com>

.changeset/nice-spiders-perform.md

@@ -0,0 +1,21 @@
+---
+'@xstate/store': patch
+---
+
+The `context` type for `createStoreWithProducer(producer, context, transitions)` will now be properly inferred.
+
+```ts
+const store = createStoreWithProducer(
+  produce,
+  {
+    count: 0
+  },
+  {
+    // ...
+  }
+);
+
+store.getSnapshot().context;
+// BEFORE: StoreContext
+// NOW: { count: number }
+```

packages/xstate-store/src/store.ts

@@ -187,10 +187,10 @@ export function createStoreWithProducer<
   TContext extends StoreContext,
   TEventPayloadMap extends EventPayloadMap
 >(
-  producer: (
+  producer: NoInfer<(
     context: TContext,
     recipe: (context: TContext) => void
-  ) => TContext,
+  ) => TContext>,
   initialContext: TContext,
   transitions: {
     [K in keyof TEventPayloadMap & string]: (

packages/xstate-store/test/store.test.ts

@@ -97,7 +97,7 @@ it('updates state from sent events', () => {
   expect(store.getSnapshot().context).toEqual({ count: 0 });
 });
 
-it('works with immer', () => {
+it('createStoreWithProducer(…) works with an immer producer', () => {
   const store = createStoreWithProducer(
     produce,
     {
@@ -120,6 +120,22 @@ it('works with immer', () => {
   expect(store.getInitialSnapshot().context).toEqual({ count: 0 });
 });
 
+it('createStoreWithProducer(…) infers the context type properly with a producer', () => {
+  const store = createStoreWithProducer(
+    produce,
+    {
+      count: 0
+    },
+    {
+      inc: (ctx, ev: { by: number }) => {
+        ctx.count += ev.by;
+      }
+    }
+  );
+
+  store.getSnapshot().context satisfies { count: number };
+});
+
 it('can be observed', () => {
   const store = createStore(
     {