fix(client): initialize shared state once per key across trust flips

Author: antfubot

packages/devframe/src/client/rpc-shared-state.test.ts

@@ -0,0 +1,36 @@
+import { createEventEmitter } from 'devframe/utils/events'
+import { describe, expect, it } from 'vitest'
+import { createRpcSharedStateClientHost } from './rpc-shared-state'
+
+function makeFakeRpc() {
+  const events = createEventEmitter<any>()
+  const setCalls: any[][] = []
+  const rpc = {
+    connectionMeta: { backend: 'websocket' },
+    isTrusted: false,
+    events,
+    client: { register: () => {} },
+    callEvent: (name: string, ...args: any[]) => {
+      if (name === 'devframe:rpc:server-state:set')
+        setCalls.push(args)
+    },
+    call: async () => undefined,
+  } as any
+  return { rpc, events, setCalls }
+}
+
+describe('client shared state', () => {
+  it('registers the server-sync bridge once across repeated trust flips', async () => {
+    const { rpc, events, setCalls } = makeFakeRpc()
+    const host = createRpcSharedStateClientHost(rpc)
+    const state = await host.get('k', { initialValue: { a: 1 } })
+
+    events.emit('rpc:is-trusted:updated', true)
+    events.emit('rpc:is-trusted:updated', true) // second flip must not re-register
+
+    state.mutate((d: any) => {
+      d.a = 2
+    })
+    expect(setCalls).toHaveLength(1) // exactly one server-state:set, not two
+  })
+})

packages/devframe/src/client/rpc-shared-state.ts

@@ -114,8 +114,10 @@ export function createRpcSharedStateClientHost(rpc: DevframeRpcClient): RpcShare
       return new Promise<SharedState<T>>((resolve) => {
         if (!rpc.isTrusted) {
           resolve(state)
+          let initialized = false
           rpc.events.on('rpc:is-trusted:updated', (isTrusted) => {
-            if (isTrusted) {
+            if (isTrusted && !initialized) {
+              initialized = true
               initSharedState()
             }
           })