fix(zustand): publish shared main external writes

Author: unadlib

packages/coaction-zustand/src/index.ts

@@ -33,14 +33,16 @@ export const bindZustand = ((initializer: StateCreator<any, [], []>) =>
         internal.rootState = zustandStore.getState() as object;
         const unsubscribe = zustandStore.subscribe(() => {
           if (!isCoactionUpdated) {
-            internal.rootState = zustandStore.getState() as object;
+            const nextState = zustandStore.getState() as object;
             if (coactionStore.share === 'client') {
+              internal.rootState = nextState;
               throw new Error('client zustand store cannot be updated');
             } else if (coactionStore.share === 'main') {
               // emit to all clients
-              coactionStore.setState(zustandStore.getState()!);
+              coactionStore.setState(nextState);
               return;
             }
+            internal.rootState = nextState;
           }
           internal.notifyStateChange();
         });

packages/coaction-zustand/test/index.test.ts

@@ -84,13 +84,17 @@ test('initializer get callback reads latest state', () => {
   expect(useStore.getState().count).toBe(2);
 });
 
-test('worker main propagates direct zustand mutations', () => {
+test('worker main propagates direct zustand mutations', async () => {
   type Counter = {
     count: number;
     increment: () => void;
   };
   const ports = mockPorts();
   const serverTransport = createTransport('WebWorkerInternal', ports.main);
+  const clientTransport = createTransport(
+    'WebWorkerClient',
+    ports.create() as WorkerMainTransportOptions
+  );
   const counter: StateCreator<Counter, [], []> = (set) => ({
     count: 0,
     increment() {
@@ -102,8 +106,38 @@ test('worker main propagates direct zustand mutations', () => {
     transport: serverTransport,
     name: 'test-worker-main'
   });
+  const serverListener = jest.fn();
+  const signalValues: number[] = [];
+  useServerStore.subscribe(serverListener);
+  const stop = effect(() => {
+    signalValues.push(useServerStore.getState().count);
+  });
+  const useClientStore = create(
+    () => adapt(createWithZustand(bindZustand(counter))),
+    {
+      clientTransport,
+      name: 'test-worker-main'
+    }
+  );
+  await new Promise((resolve) => {
+    clientTransport.onConnect(() => {
+      setTimeout(resolve);
+    });
+  });
+  const clientListener = jest.fn();
+  useClientStore.subscribe(clientListener);
+
   underlyingStore.setState({ count: 6 });
+  await new Promise((resolve) => {
+    setTimeout(resolve);
+  });
+  stop();
+
   expect(useServerStore.getState().count).toBe(6);
+  expect(serverListener).toHaveBeenCalledTimes(1);
+  expect(signalValues).toEqual([0, 6]);
+  expect(useClientStore.getState().count).toBe(6);
+  expect(clientListener).toHaveBeenCalled();
 });
 
 test('base direct zustand mutation syncs without forwarding', () => {