@derivation/rpc 0.6.0 → 0.8.0

package/dist/websocket-client.cjs.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/websocket/web-socket-client.ts","../src/client-message.ts"],"sourcesContent":["import { ClientMessage } from \"../client-message.js\";\nimport { ServerMessage } from \"../server-message.js\";\nimport type { Graph } from \"derivation\";\nimport type {\n  Sink,\n  StreamSinks,\n  RPCDefinition,\n  MutationResult,\n} from \"../stream-types.js\";\n\nfunction changer<T extends object, I extends object>(\n  sink: Sink<T, I>,\n  input: WeakRef<I>,\n): (change: object) => void {\n  return (change) => {\n    const i = input.deref();\n    if (i) {\n      sink.apply(change, i);\n    }\n  };\n}\n\nexport class WebSocketClient<Defs extends RPCDefinition> {\n  private nextId = 1;\n  private pendingStreams = new Map<number, (snapshot: object) => void>();\n  private pendingMutations = new Map<\n    number,\n    (result: MutationResult<unknown>) => void\n  >();\n  private activeStreams = new Map<number, (change: object) => void>();\n  private heartbeatTimeout: NodeJS.Timeout | undefined;\n  private inactivityTimeout: NodeJS.Timeout | undefined;\n\n  private registry = new FinalizationRegistry<[number, string]>(\n    ([id, name]) => {\n      console.log(`🧹 Stream ${id} (${name}) collected — unsubscribing`);\n      this.sendMessage(ClientMessage.unsubscribe(id));\n      this.activeStreams.delete(id);\n    },\n  );\n\n  private resetHeartbeat() {\n    if (this.heartbeatTimeout) {\n      clearTimeout(this.heartbeatTimeout);\n    }\n\n    this.heartbeatTimeout = setTimeout(() => {\n      this.sendMessage(ClientMessage.heartbeat());\n    }, 10_000);\n  }\n\n  private resetInactivity() {\n    if (this.inactivityTimeout) {\n      clearTimeout(this.inactivityTimeout);\n    }\n\n    this.inactivityTimeout = setTimeout(() => {\n      this.close();\n    }, 30_000);\n  }\n\n  constructor(\n    private ws: globalThis.WebSocket,\n    private sinks: StreamSinks<Defs[\"streams\"]>,\n    private graph: Graph,\n  ) {\n    this.ws.onmessage = (event: MessageEvent) => {\n      const message = JSON.parse(event.data as string) as ServerMessage;\n      this.handleMessage(message);\n    };\n    this.resetHeartbeat();\n    this.resetInactivity();\n  }\n\n  private handleMessage(message: ServerMessage) {\n    this.resetInactivity();\n\n    switch (message.type) {\n      case \"snapshot\": {\n        console.log(`[WebSocketClient] Received snapshot for stream ${message.id}`);\n        const resolve = this.pendingStreams.get(message.id);\n        if (resolve) {\n          resolve(message.snapshot as object);\n          this.pendingStreams.delete(message.id);\n        }\n        break;\n      }\n      case \"delta\": {\n        const changeCount = Object.keys(message.changes).length;\n        console.log(`[WebSocketClient] Received delta with ${changeCount} changes for streams: ${Object.keys(message.changes).join(\", \")}`);\n        for (const [idStr, change] of Object.entries(message.changes)) {\n          const id = Number(idStr);\n          const sink = this.activeStreams.get(id);\n\n          if (sink && change && typeof change === \"object\") {\n            sink(change);\n          } else if (!sink) {\n            console.log(`🧹 Sink ${id} GC'd — auto-unsubscribing`);\n            this.sendMessage(ClientMessage.unsubscribe(id));\n            this.activeStreams.delete(id);\n          }\n        }\n        console.log(\"[WebSocketClient] Stepping graph\");\n        this.graph.step();\n        break;\n      }\n      case \"result\": {\n        console.log(`[WebSocketClient] Received mutation result for call ${message.id}:`, message.success ? \"success\" : \"error\");\n        const resolve = this.pendingMutations.get(message.id);\n        if (resolve) {\n          if (message.success) {\n            resolve({ success: true, value: message.value });\n          } else {\n            resolve({\n              success: false,\n              error: message.error || \"Unknown error\",\n            });\n          }\n          this.pendingMutations.delete(message.id);\n        }\n        break;\n      }\n      case \"heartbeat\":\n        console.log(\"[WebSocketClient] Received heartbeat\");\n        break;\n    }\n  }\n\n  private sendMessage(message: ClientMessage) {\n    this.resetHeartbeat();\n    this.ws.send(JSON.stringify(message));\n  }\n\n  async run<Key extends keyof Defs[\"streams\"]>(\n    key: Key,\n    args: Defs[\"streams\"][Key][\"args\"],\n  ): Promise<Defs[\"streams\"][Key][\"returnType\"]> {\n    console.log(\n      `Running stream ${String(key)} with args ${JSON.stringify(args)}`,\n    );\n    const id = this.nextId++;\n\n    this.sendMessage(ClientMessage.subscribe(id, String(key), args));\n\n    const snapshot = await new Promise<object>((resolve) => {\n      this.pendingStreams.set(id, resolve);\n    });\n\n    const endpoint = this.sinks[key];\n    const sinkAdapter = endpoint(snapshot);\n    const { stream, input } = sinkAdapter.build();\n    const inputRef = new WeakRef(input);\n    this.activeStreams.set(id, changer(sinkAdapter, inputRef));\n    this.registry.register(input, [id, String(key)]);\n\n    return stream;\n  }\n\n  async call<Key extends keyof Defs[\"mutations\"]>(\n    key: Key,\n    args: Defs[\"mutations\"][Key][\"args\"],\n  ): Promise<MutationResult<Defs[\"mutations\"][Key][\"result\"]>> {\n    console.log(\n      `Calling mutation ${String(key)} with args ${JSON.stringify(args)}`,\n    );\n    const id = this.nextId++;\n\n    this.sendMessage(\n      ClientMessage.call(id, String(key), args as Record<string, unknown>),\n    );\n\n    const result = await new Promise<\n      MutationResult<Defs[\"mutations\"][Key][\"result\"]>\n    >((resolve) => {\n      this.pendingMutations.set(\n        id,\n        resolve as (result: MutationResult<unknown>) => void,\n      );\n    });\n\n    return result;\n  }\n\n  close() {\n    clearTimeout(this.heartbeatTimeout);\n    clearTimeout(this.inactivityTimeout);\n    try {\n      this.ws.close();\n    } catch {}\n  }\n\n  setPresence(value: Record<string, unknown>): void {\n    this.sendMessage(ClientMessage.presence(value));\n  }\n}\n","import { z } from \"zod\";\n\nexport const SubscribeMessageSchema = z.object({\n  type: z.literal(\"subscribe\"),\n  id: z.number(),\n  name: z.string(),\n  args: z.looseObject({}),\n});\nexport type SubscribeMessage = z.infer<typeof SubscribeMessageSchema>;\n\nexport const UnsubscribeMessageSchema = z.object({\n  type: z.literal(\"unsubscribe\"),\n  id: z.number(),\n});\nexport type UnsubscribeMessage = z.infer<typeof UnsubscribeMessageSchema>;\n\nexport const HeartbeatMessageSchema = z.object({\n  type: z.literal(\"heartbeat\"),\n});\nexport type HeartbeatMessage = z.infer<typeof HeartbeatMessageSchema>;\n\nexport const CallMessageSchema = z.object({\n  type: z.literal(\"call\"),\n  id: z.number(),\n  name: z.string(),\n  args: z.looseObject({}),\n});\nexport type CallMessage = z.infer<typeof CallMessageSchema>;\n\nexport const PresenceMessageSchema = z.object({\n  type: z.literal(\"presence\"),\n  data: z.looseObject({}),\n});\nexport type PresenceMessage = z.infer<typeof PresenceMessageSchema>;\n\nexport const ClientMessageSchema = z.discriminatedUnion(\"type\", [\n  SubscribeMessageSchema,\n  UnsubscribeMessageSchema,\n  HeartbeatMessageSchema,\n  CallMessageSchema,\n  PresenceMessageSchema,\n]);\nexport type ClientMessage = z.infer<typeof ClientMessageSchema>;\n\nexport function parseClientMessage(data: unknown): ClientMessage {\n  return ClientMessageSchema.parse(data);\n}\n\nexport const ClientMessage = {\n  subscribe: (\n    id: number,\n    name: string,\n    args: object,\n  ): SubscribeMessage => ({\n    type: \"subscribe\",\n    id,\n    name,\n    args: args as Record<string, unknown>,\n  }),\n  unsubscribe: (id: number): UnsubscribeMessage => ({\n    type: \"unsubscribe\",\n    id,\n  }),\n  call: (\n    id: number,\n    name: string,\n    args: object,\n  ): CallMessage => ({\n    type: \"call\",\n    id,\n    name,\n    args: args as Record<string, unknown>,\n  }),\n  heartbeat: (): HeartbeatMessage => ({\n    type: \"heartbeat\",\n  }),\n  presence: (data: object): PresenceMessage => ({\n    type: \"presence\",\n    data: data as Record<string, unknown>,\n  }),\n};\n"],"mappings":";;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;;;ACAA,iBAAkB;AAEX,IAAM,yBAAyB,aAAE,OAAO;AAAA,EAC7C,MAAM,aAAE,QAAQ,WAAW;AAAA,EAC3B,IAAI,aAAE,OAAO;AAAA,EACb,MAAM,aAAE,OAAO;AAAA,EACf,MAAM,aAAE,YAAY,CAAC,CAAC;AACxB,CAAC;AAGM,IAAM,2BAA2B,aAAE,OAAO;AAAA,EAC/C,MAAM,aAAE,QAAQ,aAAa;AAAA,EAC7B,IAAI,aAAE,OAAO;AACf,CAAC;AAGM,IAAM,yBAAyB,aAAE,OAAO;AAAA,EAC7C,MAAM,aAAE,QAAQ,WAAW;AAC7B,CAAC;AAGM,IAAM,oBAAoB,aAAE,OAAO;AAAA,EACxC,MAAM,aAAE,QAAQ,MAAM;AAAA,EACtB,IAAI,aAAE,OAAO;AAAA,EACb,MAAM,aAAE,OAAO;AAAA,EACf,MAAM,aAAE,YAAY,CAAC,CAAC;AACxB,CAAC;AAGM,IAAM,wBAAwB,aAAE,OAAO;AAAA,EAC5C,MAAM,aAAE,QAAQ,UAAU;AAAA,EAC1B,MAAM,aAAE,YAAY,CAAC,CAAC;AACxB,CAAC;AAGM,IAAM,sBAAsB,aAAE,mBAAmB,QAAQ;AAAA,EAC9D;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AACF,CAAC;AAOM,IAAM,gBAAgB;AAAA,EAC3B,WAAW,CACT,IACA,MACA,UACsB;AAAA,IACtB,MAAM;AAAA,IACN;AAAA,IACA;AAAA,IACA;AAAA,EACF;AAAA,EACA,aAAa,CAAC,QAAoC;AAAA,IAChD,MAAM;AAAA,IACN;AAAA,EACF;AAAA,EACA,MAAM,CACJ,IACA,MACA,UACiB;AAAA,IACjB,MAAM;AAAA,IACN;AAAA,IACA;AAAA,IACA;AAAA,EACF;AAAA,EACA,WAAW,OAAyB;AAAA,IAClC,MAAM;AAAA,EACR;AAAA,EACA,UAAU,CAAC,UAAmC;AAAA,IAC5C,MAAM;AAAA,IACN;AAAA,EACF;AACF;;;ADtEA,SAAS,QACP,MACA,OAC0B;AAC1B,SAAO,CAAC,WAAW;AACjB,UAAM,IAAI,MAAM,MAAM;AACtB,QAAI,GAAG;AACL,WAAK,MAAM,QAAQ,CAAC;AAAA,IACtB;AAAA,EACF;AACF;AAEO,IAAM,kBAAN,MAAkD;AAAA,EAuCvD,YACU,IACA,OACA,OACR;AAHQ;AACA;AACA;AAzCV,SAAQ,SAAS;AACjB,SAAQ,iBAAiB,oBAAI,IAAwC;AACrE,SAAQ,mBAAmB,oBAAI,IAG7B;AACF,SAAQ,gBAAgB,oBAAI,IAAsC;AAIlE,SAAQ,WAAW,IAAI;AAAA,MACrB,CAAC,CAAC,IAAI,IAAI,MAAM;AACd,gBAAQ,IAAI,oBAAa,EAAE,KAAK,IAAI,kCAA6B;AACjE,aAAK,YAAY,cAAc,YAAY,EAAE,CAAC;AAC9C,aAAK,cAAc,OAAO,EAAE;AAAA,MAC9B;AAAA,IACF;AA2BE,SAAK,GAAG,YAAY,CAAC,UAAwB;AAC3C,YAAM,UAAU,KAAK,MAAM,MAAM,IAAc;AAC/C,WAAK,cAAc,OAAO;AAAA,IAC5B;AACA,SAAK,eAAe;AACpB,SAAK,gBAAgB;AAAA,EACvB;AAAA,EA/BQ,iBAAiB;AACvB,QAAI,KAAK,kBAAkB;AACzB,mBAAa,KAAK,gBAAgB;AAAA,IACpC;AAEA,SAAK,mBAAmB,WAAW,MAAM;AACvC,WAAK,YAAY,cAAc,UAAU,CAAC;AAAA,IAC5C,GAAG,GAAM;AAAA,EACX;AAAA,EAEQ,kBAAkB;AACxB,QAAI,KAAK,mBAAmB;AAC1B,mBAAa,KAAK,iBAAiB;AAAA,IACrC;AAEA,SAAK,oBAAoB,WAAW,MAAM;AACxC,WAAK,MAAM;AAAA,IACb,GAAG,GAAM;AAAA,EACX;AAAA,EAeQ,cAAc,SAAwB;AAC5C,SAAK,gBAAgB;AAErB,YAAQ,QAAQ,MAAM;AAAA,MACpB,KAAK,YAAY;AACf,gBAAQ,IAAI,kDAAkD,QAAQ,EAAE,EAAE;AAC1E,cAAM,UAAU,KAAK,eAAe,IAAI,QAAQ,EAAE;AAClD,YAAI,SAAS;AACX,kBAAQ,QAAQ,QAAkB;AAClC,eAAK,eAAe,OAAO,QAAQ,EAAE;AAAA,QACvC;AACA;AAAA,MACF;AAAA,MACA,KAAK,SAAS;AACZ,cAAM,cAAc,OAAO,KAAK,QAAQ,OAAO,EAAE;AACjD,gBAAQ,IAAI,yCAAyC,WAAW,yBAAyB,OAAO,KAAK,QAAQ,OAAO,EAAE,KAAK,IAAI,CAAC,EAAE;AAClI,mBAAW,CAAC,OAAO,MAAM,KAAK,OAAO,QAAQ,QAAQ,OAAO,GAAG;AAC7D,gBAAM,KAAK,OAAO,KAAK;AACvB,gBAAM,OAAO,KAAK,cAAc,IAAI,EAAE;AAEtC,cAAI,QAAQ,UAAU,OAAO,WAAW,UAAU;AAChD,iBAAK,MAAM;AAAA,UACb,WAAW,CAAC,MAAM;AAChB,oBAAQ,IAAI,kBAAW,EAAE,iCAA4B;AACrD,iBAAK,YAAY,cAAc,YAAY,EAAE,CAAC;AAC9C,iBAAK,cAAc,OAAO,EAAE;AAAA,UAC9B;AAAA,QACF;AACA,gBAAQ,IAAI,kCAAkC;AAC9C,aAAK,MAAM,KAAK;AAChB;AAAA,MACF;AAAA,MACA,KAAK,UAAU;AACb,gBAAQ,IAAI,uDAAuD,QAAQ,EAAE,KAAK,QAAQ,UAAU,YAAY,OAAO;AACvH,cAAM,UAAU,KAAK,iBAAiB,IAAI,QAAQ,EAAE;AACpD,YAAI,SAAS;AACX,cAAI,QAAQ,SAAS;AACnB,oBAAQ,EAAE,SAAS,MAAM,OAAO,QAAQ,MAAM,CAAC;AAAA,UACjD,OAAO;AACL,oBAAQ;AAAA,cACN,SAAS;AAAA,cACT,OAAO,QAAQ,SAAS;AAAA,YAC1B,CAAC;AAAA,UACH;AACA,eAAK,iBAAiB,OAAO,QAAQ,EAAE;AAAA,QACzC;AACA;AAAA,MACF;AAAA,MACA,KAAK;AACH,gBAAQ,IAAI,sCAAsC;AAClD;AAAA,IACJ;AAAA,EACF;AAAA,EAEQ,YAAY,SAAwB;AAC1C,SAAK,eAAe;AACpB,SAAK,GAAG,KAAK,KAAK,UAAU,OAAO,CAAC;AAAA,EACtC;AAAA,EAEA,MAAM,IACJ,KACA,MAC6C;AAC7C,YAAQ;AAAA,MACN,kBAAkB,OAAO,GAAG,CAAC,cAAc,KAAK,UAAU,IAAI,CAAC;AAAA,IACjE;AACA,UAAM,KAAK,KAAK;AAEhB,SAAK,YAAY,cAAc,UAAU,IAAI,OAAO,GAAG,GAAG,IAAI,CAAC;AAE/D,UAAM,WAAW,MAAM,IAAI,QAAgB,CAAC,YAAY;AACtD,WAAK,eAAe,IAAI,IAAI,OAAO;AAAA,IACrC,CAAC;AAED,UAAM,WAAW,KAAK,MAAM,GAAG;AAC/B,UAAM,cAAc,SAAS,QAAQ;AACrC,UAAM,EAAE,QAAQ,MAAM,IAAI,YAAY,MAAM;AAC5C,UAAM,WAAW,IAAI,QAAQ,KAAK;AAClC,SAAK,cAAc,IAAI,IAAI,QAAQ,aAAa,QAAQ,CAAC;AACzD,SAAK,SAAS,SAAS,OAAO,CAAC,IAAI,OAAO,GAAG,CAAC,CAAC;AAE/C,WAAO;AAAA,EACT;AAAA,EAEA,MAAM,KACJ,KACA,MAC2D;AAC3D,YAAQ;AAAA,MACN,oBAAoB,OAAO,GAAG,CAAC,cAAc,KAAK,UAAU,IAAI,CAAC;AAAA,IACnE;AACA,UAAM,KAAK,KAAK;AAEhB,SAAK;AAAA,MACH,cAAc,KAAK,IAAI,OAAO,GAAG,GAAG,IAA+B;AAAA,IACrE;AAEA,UAAM,SAAS,MAAM,IAAI,QAEvB,CAAC,YAAY;AACb,WAAK,iBAAiB;AAAA,QACpB;AAAA,QACA;AAAA,MACF;AAAA,IACF,CAAC;AAED,WAAO;AAAA,EACT;AAAA,EAEA,QAAQ;AACN,iBAAa,KAAK,gBAAgB;AAClC,iBAAa,KAAK,iBAAiB;AACnC,QAAI;AACF,WAAK,GAAG,MAAM;AAAA,IAChB,SAAQ;AAAA,IAAC;AAAA,EACX;AAAA,EAEA,YAAY,OAAsC;AAChD,SAAK,YAAY,cAAc,SAAS,KAAK,CAAC;AAAA,EAChD;AACF;","names":[]}
+{"version":3,"sources":["../src/websocket/web-socket-client.ts","../src/client-message.ts"],"sourcesContent":["import { ClientMessage } from \"../client-message.js\";\nimport { ServerMessage } from \"../server-message.js\";\nimport type { Graph } from \"derivation\";\nimport type {\n  Sink,\n  StreamSinks,\n  RPCDefinition,\n  MutationResult,\n} from \"../stream-types.js\";\n\nconst DISCONNECTED_ERROR = \"WebSocket is disconnected\";\nconst CONNECTION_CLOSED_ERROR = \"WebSocket connection closed\";\n\ntype PendingStream = {\n  name: string;\n  args: Record<string, unknown>;\n  resolve: (snapshot: object) => void;\n  reject: (error: Error) => void;\n};\n\ntype ActiveStream = {\n  name: string;\n  args: Record<string, unknown>;\n  sink: Sink<object, object>;\n  input: WeakRef<object>;\n};\n\nexport type WebSocketReconnectOptions = {\n  baseDelayMs?: number;\n  maxDelayMs?: number;\n  jitterRatio?: number;\n  maxReconnectAttempts?: number;\n};\n\nfunction randomJitter(range: number): number {\n  if (range <= 0) return 0;\n  return (Math.random() * 2 - 1) * range;\n}\n\nexport class WebSocketClient<Defs extends RPCDefinition> {\n  private ws: globalThis.WebSocket;\n  private nextId = 1;\n  private pendingStreams = new Map<number, PendingStream>();\n  private pendingMutations = new Map<\n    number,\n    (result: MutationResult<unknown>) => void\n  >();\n  private activeStreams = new Map<number, ActiveStream>();\n  private heartbeatTimeout: NodeJS.Timeout | undefined;\n  private inactivityTimeout: NodeJS.Timeout | undefined;\n  private reconnectTimeout: NodeJS.Timeout | undefined;\n  private reconnectAttempts = 0;\n  private connected = false;\n  private closed = false;\n  private currentPresence: Record<string, unknown> | undefined;\n\n  private readonly reconnectOptions: {\n    baseDelayMs: number;\n    maxDelayMs: number;\n    jitterRatio: number;\n    maxReconnectAttempts: number;\n  };\n\n  private registry = new FinalizationRegistry<[number, string]>(\n    ([id, name]) => {\n      console.log(`🧹 Stream ${id} (${name}) collected — unsubscribing`);\n      this.sendMessage(ClientMessage.unsubscribe(id));\n      this.activeStreams.delete(id);\n    },\n  );\n\n  constructor(\n    private url: string,\n    private sinks: StreamSinks<Defs[\"streams\"]>,\n    private graph: Graph,\n    reconnectOptions: WebSocketReconnectOptions = {},\n  ) {\n    this.reconnectOptions = {\n      baseDelayMs: reconnectOptions.baseDelayMs ?? 500,\n      maxDelayMs: reconnectOptions.maxDelayMs ?? 15_000,\n      jitterRatio: reconnectOptions.jitterRatio ?? 0.2,\n      maxReconnectAttempts:\n        reconnectOptions.maxReconnectAttempts ?? Number.POSITIVE_INFINITY,\n    };\n\n    this.ws = this.createSocket();\n    this.bindSocket(this.ws);\n    this.connected = this.isSocketOpen();\n    if (this.connected) {\n      this.resetHeartbeat();\n      this.resetInactivity();\n    }\n  }\n\n  private createSocket(): globalThis.WebSocket {\n    return new globalThis.WebSocket(this.url);\n  }\n\n  private bindSocket(socket: globalThis.WebSocket) {\n    this.ws = socket;\n\n    socket.onopen = () => {\n      if (this.closed || this.ws !== socket) {\n        return;\n      }\n      this.handleSocketOpen();\n    };\n\n    socket.onmessage = (event: MessageEvent) => {\n      if (this.closed || this.ws !== socket) {\n        return;\n      }\n      const message = JSON.parse(event.data as string) as ServerMessage;\n      this.handleMessage(message);\n    };\n\n    socket.onerror = () => {\n      if (this.closed || this.ws !== socket) {\n        return;\n      }\n      this.handleTransportClose();\n    };\n\n    socket.onclose = () => {\n      if (this.closed || this.ws !== socket) {\n        return;\n      }\n      this.handleTransportClose();\n    };\n  }\n\n  private handleSocketOpen() {\n    this.connected = true;\n    this.reconnectAttempts = 0;\n\n    if (this.reconnectTimeout) {\n      clearTimeout(this.reconnectTimeout);\n      this.reconnectTimeout = undefined;\n    }\n\n    this.resetHeartbeat();\n    this.resetInactivity();\n    this.resubscribeAll();\n\n    if (this.currentPresence) {\n      this.sendMessage(ClientMessage.presence(this.currentPresence));\n    }\n  }\n\n  private handleTransportClose() {\n    if (!this.connected && this.reconnectTimeout) {\n      return;\n    }\n\n    this.connected = false;\n\n    if (this.heartbeatTimeout) {\n      clearTimeout(this.heartbeatTimeout);\n      this.heartbeatTimeout = undefined;\n    }\n    if (this.inactivityTimeout) {\n      clearTimeout(this.inactivityTimeout);\n      this.inactivityTimeout = undefined;\n    }\n\n    this.failPendingMutations(DISCONNECTED_ERROR);\n\n    this.scheduleReconnect();\n  }\n\n  private scheduleReconnect() {\n    if (this.closed || this.reconnectTimeout) {\n      return;\n    }\n\n    if (this.reconnectAttempts >= this.reconnectOptions.maxReconnectAttempts) {\n      this.failPendingStreams(CONNECTION_CLOSED_ERROR);\n      return;\n    }\n\n    const exponent = Math.min(this.reconnectAttempts, 30);\n    const baseDelay = this.reconnectOptions.baseDelayMs * 2 ** exponent;\n    const cappedDelay = Math.min(baseDelay, this.reconnectOptions.maxDelayMs);\n    const jitter = cappedDelay * this.reconnectOptions.jitterRatio;\n    const delay = Math.max(0, Math.round(cappedDelay + randomJitter(jitter)));\n\n    this.reconnectAttempts += 1;\n\n    this.reconnectTimeout = setTimeout(() => {\n      this.reconnectTimeout = undefined;\n      if (this.closed) {\n        return;\n      }\n\n      try {\n        const socket = this.createSocket();\n        this.bindSocket(socket);\n        this.connected = this.isSocketOpen();\n        if (this.connected) {\n          this.handleSocketOpen();\n        }\n      } catch (err) {\n        console.error(\"Failed to create reconnect socket:\", err);\n        this.scheduleReconnect();\n      }\n    }, delay);\n  }\n\n  private resubscribeAll() {\n    for (const [id, pending] of this.pendingStreams) {\n      this.sendMessage(ClientMessage.subscribe(id, pending.name, pending.args));\n    }\n\n    for (const [id, active] of this.activeStreams) {\n      if (!active.input.deref()) {\n        this.activeStreams.delete(id);\n        continue;\n      }\n      this.sendMessage(ClientMessage.subscribe(id, active.name, active.args));\n    }\n  }\n\n  private failPendingStreams(message: string) {\n    for (const [id, pending] of this.pendingStreams) {\n      pending.reject(new Error(message));\n      this.pendingStreams.delete(id);\n    }\n  }\n\n  private failPendingMutations(message: string) {\n    for (const [id, resolve] of this.pendingMutations) {\n      resolve({ success: false, error: message });\n      this.pendingMutations.delete(id);\n    }\n  }\n\n  private resetHeartbeat() {\n    if (!this.connected || this.closed) {\n      return;\n    }\n\n    if (this.heartbeatTimeout) {\n      clearTimeout(this.heartbeatTimeout);\n    }\n\n    this.heartbeatTimeout = setTimeout(() => {\n      this.sendMessage(ClientMessage.heartbeat());\n    }, 10_000);\n  }\n\n  private resetInactivity() {\n    if (!this.connected || this.closed) {\n      return;\n    }\n\n    if (this.inactivityTimeout) {\n      clearTimeout(this.inactivityTimeout);\n    }\n\n    this.inactivityTimeout = setTimeout(() => {\n      try {\n        this.ws.close();\n      } catch {\n        this.handleTransportClose();\n      }\n    }, 30_000);\n  }\n\n  private handleMessage(message: ServerMessage) {\n    this.resetInactivity();\n\n    switch (message.type) {\n      case \"snapshot\": {\n        console.log(`[WebSocketClient] Received snapshot for stream ${message.id}`);\n\n        const pending = this.pendingStreams.get(message.id);\n        if (pending) {\n          pending.resolve(message.snapshot as object);\n          this.pendingStreams.delete(message.id);\n          break;\n        }\n\n        const active = this.activeStreams.get(message.id);\n        if (!active) {\n          break;\n        }\n\n        const input = active.input.deref();\n        if (!input) {\n          this.sendMessage(ClientMessage.unsubscribe(message.id));\n          this.activeStreams.delete(message.id);\n          break;\n        }\n\n        active.sink.reset(message.snapshot as object, input);\n        this.graph.step();\n        break;\n      }\n      case \"delta\": {\n        const changeCount = Object.keys(message.changes).length;\n        console.log(\n          `[WebSocketClient] Received delta with ${changeCount} changes for streams: ${Object.keys(message.changes).join(\", \")}`,\n        );\n\n        for (const [idStr, change] of Object.entries(message.changes)) {\n          const id = Number(idStr);\n          const active = this.activeStreams.get(id);\n\n          if (!active) {\n            this.sendMessage(ClientMessage.unsubscribe(id));\n            continue;\n          }\n\n          const input = active.input.deref();\n          if (!input) {\n            this.sendMessage(ClientMessage.unsubscribe(id));\n            this.activeStreams.delete(id);\n            continue;\n          }\n\n          if (change && typeof change === \"object\") {\n            active.sink.apply(change, input);\n          }\n        }\n\n        console.log(\"[WebSocketClient] Stepping graph\");\n        this.graph.step();\n        break;\n      }\n      case \"result\": {\n        console.log(\n          `[WebSocketClient] Received mutation result for call ${message.id}:`,\n          message.success ? \"success\" : \"error\",\n        );\n\n        const resolve = this.pendingMutations.get(message.id);\n        if (!resolve) {\n          break;\n        }\n\n        if (message.success) {\n          resolve({ success: true, value: message.value });\n        } else {\n          resolve({\n            success: false,\n            error: message.error || \"Unknown error\",\n          });\n        }\n        this.pendingMutations.delete(message.id);\n        break;\n      }\n      case \"heartbeat\":\n        console.log(\"[WebSocketClient] Received heartbeat\");\n        break;\n    }\n  }\n\n  private isSocketOpen() {\n    return this.ws.readyState === this.ws.OPEN;\n  }\n\n  private sendMessage(message: ClientMessage): boolean {\n    if (this.closed) {\n      return false;\n    }\n\n    if (!this.isSocketOpen()) {\n      return false;\n    }\n\n    this.resetHeartbeat();\n\n    try {\n      this.ws.send(JSON.stringify(message));\n      return true;\n    } catch {\n      this.handleTransportClose();\n      return false;\n    }\n  }\n\n  async run<Key extends keyof Defs[\"streams\"]>(\n    key: Key,\n    args: Defs[\"streams\"][Key][\"args\"],\n  ): Promise<Defs[\"streams\"][Key][\"returnType\"]> {\n    console.log(\n      `Running stream ${String(key)} with args ${JSON.stringify(args)}`,\n    );\n\n    const id = this.nextId++;\n    const name = String(key);\n    const typedArgs = args as Record<string, unknown>;\n\n    const snapshot = await new Promise<object>((resolve, reject) => {\n      this.pendingStreams.set(id, { name, args: typedArgs, resolve, reject });\n      this.sendMessage(ClientMessage.subscribe(id, name, typedArgs));\n    });\n\n    const endpoint = this.sinks[key];\n    const sinkAdapter = endpoint(snapshot);\n    const { stream, input } = sinkAdapter.build();\n    const inputRef = new WeakRef(input);\n\n    this.activeStreams.set(id, {\n      name,\n      args: typedArgs,\n      sink: sinkAdapter as Sink<object, object>,\n      input: inputRef as WeakRef<object>,\n    });\n\n    this.registry.register(input, [id, name]);\n    return stream;\n  }\n\n  async call<Key extends keyof Defs[\"mutations\"]>(\n    key: Key,\n    args: Defs[\"mutations\"][Key][\"args\"],\n  ): Promise<MutationResult<Defs[\"mutations\"][Key][\"result\"]>> {\n    console.log(\n      `Calling mutation ${String(key)} with args ${JSON.stringify(args)}`,\n    );\n\n    if (!this.connected || !this.isSocketOpen()) {\n      return {\n        success: false,\n        error: DISCONNECTED_ERROR,\n      };\n    }\n\n    const id = this.nextId++;\n\n    const resultPromise = new Promise<\n      MutationResult<Defs[\"mutations\"][Key][\"result\"]>\n    >((resolve) => {\n      this.pendingMutations.set(\n        id,\n        resolve as (result: MutationResult<unknown>) => void,\n      );\n    });\n\n    const sent = this.sendMessage(\n      ClientMessage.call(id, String(key), args as Record<string, unknown>),\n    );\n\n    if (!sent) {\n      this.pendingMutations.delete(id);\n      return {\n        success: false,\n        error: DISCONNECTED_ERROR,\n      };\n    }\n\n    return resultPromise;\n  }\n\n  close() {\n    if (this.closed) {\n      return;\n    }\n\n    this.closed = true;\n    this.connected = false;\n\n    if (this.heartbeatTimeout) {\n      clearTimeout(this.heartbeatTimeout);\n      this.heartbeatTimeout = undefined;\n    }\n    if (this.inactivityTimeout) {\n      clearTimeout(this.inactivityTimeout);\n      this.inactivityTimeout = undefined;\n    }\n    if (this.reconnectTimeout) {\n      clearTimeout(this.reconnectTimeout);\n      this.reconnectTimeout = undefined;\n    }\n\n    this.failPendingStreams(CONNECTION_CLOSED_ERROR);\n    this.failPendingMutations(CONNECTION_CLOSED_ERROR);\n\n    try {\n      this.ws.close();\n    } catch {}\n  }\n\n  setPresence(value: Record<string, unknown>): void {\n    this.currentPresence = value;\n    this.sendMessage(ClientMessage.presence(value));\n  }\n}\n","import { z } from \"zod\";\n\nexport const SubscribeMessageSchema = z.object({\n  type: z.literal(\"subscribe\"),\n  id: z.number(),\n  name: z.string(),\n  args: z.looseObject({}),\n});\nexport type SubscribeMessage = z.infer<typeof SubscribeMessageSchema>;\n\nexport const UnsubscribeMessageSchema = z.object({\n  type: z.literal(\"unsubscribe\"),\n  id: z.number(),\n});\nexport type UnsubscribeMessage = z.infer<typeof UnsubscribeMessageSchema>;\n\nexport const HeartbeatMessageSchema = z.object({\n  type: z.literal(\"heartbeat\"),\n});\nexport type HeartbeatMessage = z.infer<typeof HeartbeatMessageSchema>;\n\nexport const CallMessageSchema = z.object({\n  type: z.literal(\"call\"),\n  id: z.number(),\n  name: z.string(),\n  args: z.looseObject({}),\n});\nexport type CallMessage = z.infer<typeof CallMessageSchema>;\n\nexport const PresenceMessageSchema = z.object({\n  type: z.literal(\"presence\"),\n  data: z.looseObject({}),\n});\nexport type PresenceMessage = z.infer<typeof PresenceMessageSchema>;\n\nexport const ClientMessageSchema = z.discriminatedUnion(\"type\", [\n  SubscribeMessageSchema,\n  UnsubscribeMessageSchema,\n  HeartbeatMessageSchema,\n  CallMessageSchema,\n  PresenceMessageSchema,\n]);\nexport type ClientMessage = z.infer<typeof ClientMessageSchema>;\n\nexport function parseClientMessage(data: unknown): ClientMessage {\n  return ClientMessageSchema.parse(data);\n}\n\nexport const ClientMessage = {\n  subscribe: (\n    id: number,\n    name: string,\n    args: object,\n  ): SubscribeMessage => ({\n    type: \"subscribe\",\n    id,\n    name,\n    args: args as Record<string, unknown>,\n  }),\n  unsubscribe: (id: number): UnsubscribeMessage => ({\n    type: \"unsubscribe\",\n    id,\n  }),\n  call: (\n    id: number,\n    name: string,\n    args: object,\n  ): CallMessage => ({\n    type: \"call\",\n    id,\n    name,\n    args: args as Record<string, unknown>,\n  }),\n  heartbeat: (): HeartbeatMessage => ({\n    type: \"heartbeat\",\n  }),\n  presence: (data: object): PresenceMessage => ({\n    type: \"presence\",\n    data: data as Record<string, unknown>,\n  }),\n};\n"],"mappings":";;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;;;ACAA,iBAAkB;AAEX,IAAM,yBAAyB,aAAE,OAAO;AAAA,EAC7C,MAAM,aAAE,QAAQ,WAAW;AAAA,EAC3B,IAAI,aAAE,OAAO;AAAA,EACb,MAAM,aAAE,OAAO;AAAA,EACf,MAAM,aAAE,YAAY,CAAC,CAAC;AACxB,CAAC;AAGM,IAAM,2BAA2B,aAAE,OAAO;AAAA,EAC/C,MAAM,aAAE,QAAQ,aAAa;AAAA,EAC7B,IAAI,aAAE,OAAO;AACf,CAAC;AAGM,IAAM,yBAAyB,aAAE,OAAO;AAAA,EAC7C,MAAM,aAAE,QAAQ,WAAW;AAC7B,CAAC;AAGM,IAAM,oBAAoB,aAAE,OAAO;AAAA,EACxC,MAAM,aAAE,QAAQ,MAAM;AAAA,EACtB,IAAI,aAAE,OAAO;AAAA,EACb,MAAM,aAAE,OAAO;AAAA,EACf,MAAM,aAAE,YAAY,CAAC,CAAC;AACxB,CAAC;AAGM,IAAM,wBAAwB,aAAE,OAAO;AAAA,EAC5C,MAAM,aAAE,QAAQ,UAAU;AAAA,EAC1B,MAAM,aAAE,YAAY,CAAC,CAAC;AACxB,CAAC;AAGM,IAAM,sBAAsB,aAAE,mBAAmB,QAAQ;AAAA,EAC9D;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AACF,CAAC;AAOM,IAAM,gBAAgB;AAAA,EAC3B,WAAW,CACT,IACA,MACA,UACsB;AAAA,IACtB,MAAM;AAAA,IACN;AAAA,IACA;AAAA,IACA;AAAA,EACF;AAAA,EACA,aAAa,CAAC,QAAoC;AAAA,IAChD,MAAM;AAAA,IACN;AAAA,EACF;AAAA,EACA,MAAM,CACJ,IACA,MACA,UACiB;AAAA,IACjB,MAAM;AAAA,IACN;AAAA,IACA;AAAA,IACA;AAAA,EACF;AAAA,EACA,WAAW,OAAyB;AAAA,IAClC,MAAM;AAAA,EACR;AAAA,EACA,UAAU,CAAC,UAAmC;AAAA,IAC5C,MAAM;AAAA,IACN;AAAA,EACF;AACF;;;ADtEA,IAAM,qBAAqB;AAC3B,IAAM,0BAA0B;AAuBhC,SAAS,aAAa,OAAuB;AAC3C,MAAI,SAAS,EAAG,QAAO;AACvB,UAAQ,KAAK,OAAO,IAAI,IAAI,KAAK;AACnC;AAEO,IAAM,kBAAN,MAAkD;AAAA,EAgCvD,YACU,KACA,OACA,OACR,mBAA8C,CAAC,GAC/C;AAJQ;AACA;AACA;AAjCV,SAAQ,SAAS;AACjB,SAAQ,iBAAiB,oBAAI,IAA2B;AACxD,SAAQ,mBAAmB,oBAAI,IAG7B;AACF,SAAQ,gBAAgB,oBAAI,IAA0B;AAItD,SAAQ,oBAAoB;AAC5B,SAAQ,YAAY;AACpB,SAAQ,SAAS;AAUjB,SAAQ,WAAW,IAAI;AAAA,MACrB,CAAC,CAAC,IAAI,IAAI,MAAM;AACd,gBAAQ,IAAI,oBAAa,EAAE,KAAK,IAAI,kCAA6B;AACjE,aAAK,YAAY,cAAc,YAAY,EAAE,CAAC;AAC9C,aAAK,cAAc,OAAO,EAAE;AAAA,MAC9B;AAAA,IACF;AArEF;AA6EI,SAAK,mBAAmB;AAAA,MACtB,cAAa,sBAAiB,gBAAjB,YAAgC;AAAA,MAC7C,aAAY,sBAAiB,eAAjB,YAA+B;AAAA,MAC3C,cAAa,sBAAiB,gBAAjB,YAAgC;AAAA,MAC7C,uBACE,sBAAiB,yBAAjB,YAAyC,OAAO;AAAA,IACpD;AAEA,SAAK,KAAK,KAAK,aAAa;AAC5B,SAAK,WAAW,KAAK,EAAE;AACvB,SAAK,YAAY,KAAK,aAAa;AACnC,QAAI,KAAK,WAAW;AAClB,WAAK,eAAe;AACpB,WAAK,gBAAgB;AAAA,IACvB;AAAA,EACF;AAAA,EAEQ,eAAqC;AAC3C,WAAO,IAAI,WAAW,UAAU,KAAK,GAAG;AAAA,EAC1C;AAAA,EAEQ,WAAW,QAA8B;AAC/C,SAAK,KAAK;AAEV,WAAO,SAAS,MAAM;AACpB,UAAI,KAAK,UAAU,KAAK,OAAO,QAAQ;AACrC;AAAA,MACF;AACA,WAAK,iBAAiB;AAAA,IACxB;AAEA,WAAO,YAAY,CAAC,UAAwB;AAC1C,UAAI,KAAK,UAAU,KAAK,OAAO,QAAQ;AACrC;AAAA,MACF;AACA,YAAM,UAAU,KAAK,MAAM,MAAM,IAAc;AAC/C,WAAK,cAAc,OAAO;AAAA,IAC5B;AAEA,WAAO,UAAU,MAAM;AACrB,UAAI,KAAK,UAAU,KAAK,OAAO,QAAQ;AACrC;AAAA,MACF;AACA,WAAK,qBAAqB;AAAA,IAC5B;AAEA,WAAO,UAAU,MAAM;AACrB,UAAI,KAAK,UAAU,KAAK,OAAO,QAAQ;AACrC;AAAA,MACF;AACA,WAAK,qBAAqB;AAAA,IAC5B;AAAA,EACF;AAAA,EAEQ,mBAAmB;AACzB,SAAK,YAAY;AACjB,SAAK,oBAAoB;AAEzB,QAAI,KAAK,kBAAkB;AACzB,mBAAa,KAAK,gBAAgB;AAClC,WAAK,mBAAmB;AAAA,IAC1B;AAEA,SAAK,eAAe;AACpB,SAAK,gBAAgB;AACrB,SAAK,eAAe;AAEpB,QAAI,KAAK,iBAAiB;AACxB,WAAK,YAAY,cAAc,SAAS,KAAK,eAAe,CAAC;AAAA,IAC/D;AAAA,EACF;AAAA,EAEQ,uBAAuB;AAC7B,QAAI,CAAC,KAAK,aAAa,KAAK,kBAAkB;AAC5C;AAAA,IACF;AAEA,SAAK,YAAY;AAEjB,QAAI,KAAK,kBAAkB;AACzB,mBAAa,KAAK,gBAAgB;AAClC,WAAK,mBAAmB;AAAA,IAC1B;AACA,QAAI,KAAK,mBAAmB;AAC1B,mBAAa,KAAK,iBAAiB;AACnC,WAAK,oBAAoB;AAAA,IAC3B;AAEA,SAAK,qBAAqB,kBAAkB;AAE5C,SAAK,kBAAkB;AAAA,EACzB;AAAA,EAEQ,oBAAoB;AAC1B,QAAI,KAAK,UAAU,KAAK,kBAAkB;AACxC;AAAA,IACF;AAEA,QAAI,KAAK,qBAAqB,KAAK,iBAAiB,sBAAsB;AACxE,WAAK,mBAAmB,uBAAuB;AAC/C;AAAA,IACF;AAEA,UAAM,WAAW,KAAK,IAAI,KAAK,mBAAmB,EAAE;AACpD,UAAM,YAAY,KAAK,iBAAiB,cAAc,KAAK;AAC3D,UAAM,cAAc,KAAK,IAAI,WAAW,KAAK,iBAAiB,UAAU;AACxE,UAAM,SAAS,cAAc,KAAK,iBAAiB;AACnD,UAAM,QAAQ,KAAK,IAAI,GAAG,KAAK,MAAM,cAAc,aAAa,MAAM,CAAC,CAAC;AAExE,SAAK,qBAAqB;AAE1B,SAAK,mBAAmB,WAAW,MAAM;AACvC,WAAK,mBAAmB;AACxB,UAAI,KAAK,QAAQ;AACf;AAAA,MACF;AAEA,UAAI;AACF,cAAM,SAAS,KAAK,aAAa;AACjC,aAAK,WAAW,MAAM;AACtB,aAAK,YAAY,KAAK,aAAa;AACnC,YAAI,KAAK,WAAW;AAClB,eAAK,iBAAiB;AAAA,QACxB;AAAA,MACF,SAAS,KAAK;AACZ,gBAAQ,MAAM,sCAAsC,GAAG;AACvD,aAAK,kBAAkB;AAAA,MACzB;AAAA,IACF,GAAG,KAAK;AAAA,EACV;AAAA,EAEQ,iBAAiB;AACvB,eAAW,CAAC,IAAI,OAAO,KAAK,KAAK,gBAAgB;AAC/C,WAAK,YAAY,cAAc,UAAU,IAAI,QAAQ,MAAM,QAAQ,IAAI,CAAC;AAAA,IAC1E;AAEA,eAAW,CAAC,IAAI,MAAM,KAAK,KAAK,eAAe;AAC7C,UAAI,CAAC,OAAO,MAAM,MAAM,GAAG;AACzB,aAAK,cAAc,OAAO,EAAE;AAC5B;AAAA,MACF;AACA,WAAK,YAAY,cAAc,UAAU,IAAI,OAAO,MAAM,OAAO,IAAI,CAAC;AAAA,IACxE;AAAA,EACF;AAAA,EAEQ,mBAAmB,SAAiB;AAC1C,eAAW,CAAC,IAAI,OAAO,KAAK,KAAK,gBAAgB;AAC/C,cAAQ,OAAO,IAAI,MAAM,OAAO,CAAC;AACjC,WAAK,eAAe,OAAO,EAAE;AAAA,IAC/B;AAAA,EACF;AAAA,EAEQ,qBAAqB,SAAiB;AAC5C,eAAW,CAAC,IAAI,OAAO,KAAK,KAAK,kBAAkB;AACjD,cAAQ,EAAE,SAAS,OAAO,OAAO,QAAQ,CAAC;AAC1C,WAAK,iBAAiB,OAAO,EAAE;AAAA,IACjC;AAAA,EACF;AAAA,EAEQ,iBAAiB;AACvB,QAAI,CAAC,KAAK,aAAa,KAAK,QAAQ;AAClC;AAAA,IACF;AAEA,QAAI,KAAK,kBAAkB;AACzB,mBAAa,KAAK,gBAAgB;AAAA,IACpC;AAEA,SAAK,mBAAmB,WAAW,MAAM;AACvC,WAAK,YAAY,cAAc,UAAU,CAAC;AAAA,IAC5C,GAAG,GAAM;AAAA,EACX;AAAA,EAEQ,kBAAkB;AACxB,QAAI,CAAC,KAAK,aAAa,KAAK,QAAQ;AAClC;AAAA,IACF;AAEA,QAAI,KAAK,mBAAmB;AAC1B,mBAAa,KAAK,iBAAiB;AAAA,IACrC;AAEA,SAAK,oBAAoB,WAAW,MAAM;AACxC,UAAI;AACF,aAAK,GAAG,MAAM;AAAA,MAChB,SAAQ;AACN,aAAK,qBAAqB;AAAA,MAC5B;AAAA,IACF,GAAG,GAAM;AAAA,EACX;AAAA,EAEQ,cAAc,SAAwB;AAC5C,SAAK,gBAAgB;AAErB,YAAQ,QAAQ,MAAM;AAAA,MACpB,KAAK,YAAY;AACf,gBAAQ,IAAI,kDAAkD,QAAQ,EAAE,EAAE;AAE1E,cAAM,UAAU,KAAK,eAAe,IAAI,QAAQ,EAAE;AAClD,YAAI,SAAS;AACX,kBAAQ,QAAQ,QAAQ,QAAkB;AAC1C,eAAK,eAAe,OAAO,QAAQ,EAAE;AACrC;AAAA,QACF;AAEA,cAAM,SAAS,KAAK,cAAc,IAAI,QAAQ,EAAE;AAChD,YAAI,CAAC,QAAQ;AACX;AAAA,QACF;AAEA,cAAM,QAAQ,OAAO,MAAM,MAAM;AACjC,YAAI,CAAC,OAAO;AACV,eAAK,YAAY,cAAc,YAAY,QAAQ,EAAE,CAAC;AACtD,eAAK,cAAc,OAAO,QAAQ,EAAE;AACpC;AAAA,QACF;AAEA,eAAO,KAAK,MAAM,QAAQ,UAAoB,KAAK;AACnD,aAAK,MAAM,KAAK;AAChB;AAAA,MACF;AAAA,MACA,KAAK,SAAS;AACZ,cAAM,cAAc,OAAO,KAAK,QAAQ,OAAO,EAAE;AACjD,gBAAQ;AAAA,UACN,yCAAyC,WAAW,yBAAyB,OAAO,KAAK,QAAQ,OAAO,EAAE,KAAK,IAAI,CAAC;AAAA,QACtH;AAEA,mBAAW,CAAC,OAAO,MAAM,KAAK,OAAO,QAAQ,QAAQ,OAAO,GAAG;AAC7D,gBAAM,KAAK,OAAO,KAAK;AACvB,gBAAM,SAAS,KAAK,cAAc,IAAI,EAAE;AAExC,cAAI,CAAC,QAAQ;AACX,iBAAK,YAAY,cAAc,YAAY,EAAE,CAAC;AAC9C;AAAA,UACF;AAEA,gBAAM,QAAQ,OAAO,MAAM,MAAM;AACjC,cAAI,CAAC,OAAO;AACV,iBAAK,YAAY,cAAc,YAAY,EAAE,CAAC;AAC9C,iBAAK,cAAc,OAAO,EAAE;AAC5B;AAAA,UACF;AAEA,cAAI,UAAU,OAAO,WAAW,UAAU;AACxC,mBAAO,KAAK,MAAM,QAAQ,KAAK;AAAA,UACjC;AAAA,QACF;AAEA,gBAAQ,IAAI,kCAAkC;AAC9C,aAAK,MAAM,KAAK;AAChB;AAAA,MACF;AAAA,MACA,KAAK,UAAU;AACb,gBAAQ;AAAA,UACN,uDAAuD,QAAQ,EAAE;AAAA,UACjE,QAAQ,UAAU,YAAY;AAAA,QAChC;AAEA,cAAM,UAAU,KAAK,iBAAiB,IAAI,QAAQ,EAAE;AACpD,YAAI,CAAC,SAAS;AACZ;AAAA,QACF;AAEA,YAAI,QAAQ,SAAS;AACnB,kBAAQ,EAAE,SAAS,MAAM,OAAO,QAAQ,MAAM,CAAC;AAAA,QACjD,OAAO;AACL,kBAAQ;AAAA,YACN,SAAS;AAAA,YACT,OAAO,QAAQ,SAAS;AAAA,UAC1B,CAAC;AAAA,QACH;AACA,aAAK,iBAAiB,OAAO,QAAQ,EAAE;AACvC;AAAA,MACF;AAAA,MACA,KAAK;AACH,gBAAQ,IAAI,sCAAsC;AAClD;AAAA,IACJ;AAAA,EACF;AAAA,EAEQ,eAAe;AACrB,WAAO,KAAK,GAAG,eAAe,KAAK,GAAG;AAAA,EACxC;AAAA,EAEQ,YAAY,SAAiC;AACnD,QAAI,KAAK,QAAQ;AACf,aAAO;AAAA,IACT;AAEA,QAAI,CAAC,KAAK,aAAa,GAAG;AACxB,aAAO;AAAA,IACT;AAEA,SAAK,eAAe;AAEpB,QAAI;AACF,WAAK,GAAG,KAAK,KAAK,UAAU,OAAO,CAAC;AACpC,aAAO;AAAA,IACT,SAAQ;AACN,WAAK,qBAAqB;AAC1B,aAAO;AAAA,IACT;AAAA,EACF;AAAA,EAEA,MAAM,IACJ,KACA,MAC6C;AAC7C,YAAQ;AAAA,MACN,kBAAkB,OAAO,GAAG,CAAC,cAAc,KAAK,UAAU,IAAI,CAAC;AAAA,IACjE;AAEA,UAAM,KAAK,KAAK;AAChB,UAAM,OAAO,OAAO,GAAG;AACvB,UAAM,YAAY;AAElB,UAAM,WAAW,MAAM,IAAI,QAAgB,CAAC,SAAS,WAAW;AAC9D,WAAK,eAAe,IAAI,IAAI,EAAE,MAAM,MAAM,WAAW,SAAS,OAAO,CAAC;AACtE,WAAK,YAAY,cAAc,UAAU,IAAI,MAAM,SAAS,CAAC;AAAA,IAC/D,CAAC;AAED,UAAM,WAAW,KAAK,MAAM,GAAG;AAC/B,UAAM,cAAc,SAAS,QAAQ;AACrC,UAAM,EAAE,QAAQ,MAAM,IAAI,YAAY,MAAM;AAC5C,UAAM,WAAW,IAAI,QAAQ,KAAK;AAElC,SAAK,cAAc,IAAI,IAAI;AAAA,MACzB;AAAA,MACA,MAAM;AAAA,MACN,MAAM;AAAA,MACN,OAAO;AAAA,IACT,CAAC;AAED,SAAK,SAAS,SAAS,OAAO,CAAC,IAAI,IAAI,CAAC;AACxC,WAAO;AAAA,EACT;AAAA,EAEA,MAAM,KACJ,KACA,MAC2D;AAC3D,YAAQ;AAAA,MACN,oBAAoB,OAAO,GAAG,CAAC,cAAc,KAAK,UAAU,IAAI,CAAC;AAAA,IACnE;AAEA,QAAI,CAAC,KAAK,aAAa,CAAC,KAAK,aAAa,GAAG;AAC3C,aAAO;AAAA,QACL,SAAS;AAAA,QACT,OAAO;AAAA,MACT;AAAA,IACF;AAEA,UAAM,KAAK,KAAK;AAEhB,UAAM,gBAAgB,IAAI,QAExB,CAAC,YAAY;AACb,WAAK,iBAAiB;AAAA,QACpB;AAAA,QACA;AAAA,MACF;AAAA,IACF,CAAC;AAED,UAAM,OAAO,KAAK;AAAA,MAChB,cAAc,KAAK,IAAI,OAAO,GAAG,GAAG,IAA+B;AAAA,IACrE;AAEA,QAAI,CAAC,MAAM;AACT,WAAK,iBAAiB,OAAO,EAAE;AAC/B,aAAO;AAAA,QACL,SAAS;AAAA,QACT,OAAO;AAAA,MACT;AAAA,IACF;AAEA,WAAO;AAAA,EACT;AAAA,EAEA,QAAQ;AACN,QAAI,KAAK,QAAQ;AACf;AAAA,IACF;AAEA,SAAK,SAAS;AACd,SAAK,YAAY;AAEjB,QAAI,KAAK,kBAAkB;AACzB,mBAAa,KAAK,gBAAgB;AAClC,WAAK,mBAAmB;AAAA,IAC1B;AACA,QAAI,KAAK,mBAAmB;AAC1B,mBAAa,KAAK,iBAAiB;AACnC,WAAK,oBAAoB;AAAA,IAC3B;AACA,QAAI,KAAK,kBAAkB;AACzB,mBAAa,KAAK,gBAAgB;AAClC,WAAK,mBAAmB;AAAA,IAC1B;AAEA,SAAK,mBAAmB,uBAAuB;AAC/C,SAAK,qBAAqB,uBAAuB;AAEjD,QAAI;AACF,WAAK,GAAG,MAAM;AAAA,IAChB,SAAQ;AAAA,IAAC;AAAA,EACX;AAAA,EAEA,YAAY,OAAsC;AAChD,SAAK,kBAAkB;AACvB,SAAK,YAAY,cAAc,SAAS,KAAK,CAAC;AAAA,EAChD;AACF;","names":[]}

package/dist/websocket-client.d.cts

@@ -1,21 +1,43 @@
 import { Graph } from 'derivation';
-import { R as RPCDefinition, h as StreamSinks, d as MutationResult } from './stream-types-Q_EqNLtO.cjs';
+import { R as RPCDefinition, h as StreamSinks, d as MutationResult } from './stream-types-Cn-LsIb9.cjs';
 
+type WebSocketReconnectOptions = {
+    baseDelayMs?: number;
+    maxDelayMs?: number;
+    jitterRatio?: number;
+    maxReconnectAttempts?: number;
+};
 declare class WebSocketClient<Defs extends RPCDefinition> {
-    private ws;
+    private url;
     private sinks;
     private graph;
+    private ws;
     private nextId;
     private pendingStreams;
     private pendingMutations;
     private activeStreams;
     private heartbeatTimeout;
     private inactivityTimeout;
+    private reconnectTimeout;
+    private reconnectAttempts;
+    private connected;
+    private closed;
+    private currentPresence;
+    private readonly reconnectOptions;
     private registry;
+    constructor(url: string, sinks: StreamSinks<Defs["streams"]>, graph: Graph, reconnectOptions?: WebSocketReconnectOptions);
+    private createSocket;
+    private bindSocket;
+    private handleSocketOpen;
+    private handleTransportClose;
+    private scheduleReconnect;
+    private resubscribeAll;
+    private failPendingStreams;
+    private failPendingMutations;
     private resetHeartbeat;
     private resetInactivity;
-    constructor(ws: globalThis.WebSocket, sinks: StreamSinks<Defs["streams"]>, graph: Graph);
     private handleMessage;
+    private isSocketOpen;
     private sendMessage;
     run<Key extends keyof Defs["streams"]>(key: Key, args: Defs["streams"][Key]["args"]): Promise<Defs["streams"][Key]["returnType"]>;
     call<Key extends keyof Defs["mutations"]>(key: Key, args: Defs["mutations"][Key]["args"]): Promise<MutationResult<Defs["mutations"][Key]["result"]>>;
@@ -23,4 +45,4 @@ declare class WebSocketClient<Defs extends RPCDefinition> {
     setPresence(value: Record<string, unknown>): void;
 }
 
-export { WebSocketClient };
+export { WebSocketClient, type WebSocketReconnectOptions };

package/dist/websocket-client.d.ts

@@ -1,21 +1,43 @@
 import { Graph } from 'derivation';
-import { R as RPCDefinition, h as StreamSinks, d as MutationResult } from './stream-types-Q_EqNLtO.js';
+import { R as RPCDefinition, h as StreamSinks, d as MutationResult } from './stream-types-Cn-LsIb9.js';
 
+type WebSocketReconnectOptions = {
+    baseDelayMs?: number;
+    maxDelayMs?: number;
+    jitterRatio?: number;
+    maxReconnectAttempts?: number;
+};
 declare class WebSocketClient<Defs extends RPCDefinition> {
-    private ws;
+    private url;
     private sinks;
     private graph;
+    private ws;
     private nextId;
     private pendingStreams;
     private pendingMutations;
     private activeStreams;
     private heartbeatTimeout;
     private inactivityTimeout;
+    private reconnectTimeout;
+    private reconnectAttempts;
+    private connected;
+    private closed;
+    private currentPresence;
+    private readonly reconnectOptions;
     private registry;
+    constructor(url: string, sinks: StreamSinks<Defs["streams"]>, graph: Graph, reconnectOptions?: WebSocketReconnectOptions);
+    private createSocket;
+    private bindSocket;
+    private handleSocketOpen;
+    private handleTransportClose;
+    private scheduleReconnect;
+    private resubscribeAll;
+    private failPendingStreams;
+    private failPendingMutations;
     private resetHeartbeat;
     private resetInactivity;
-    constructor(ws: globalThis.WebSocket, sinks: StreamSinks<Defs["streams"]>, graph: Graph);
     private handleMessage;
+    private isSocketOpen;
     private sendMessage;
     run<Key extends keyof Defs["streams"]>(key: Key, args: Defs["streams"][Key]["args"]): Promise<Defs["streams"][Key]["returnType"]>;
     call<Key extends keyof Defs["mutations"]>(key: Key, args: Defs["mutations"][Key]["args"]): Promise<MutationResult<Defs["mutations"][Key]["result"]>>;
@@ -23,4 +45,4 @@ declare class WebSocketClient<Defs extends RPCDefinition> {
     setPresence(value: Record<string, unknown>): void;
 }
 
-export { WebSocketClient };
+export { WebSocketClient, type WebSocketReconnectOptions };

package/dist/websocket-client.js

@@ -57,23 +57,24 @@ var ClientMessage = {
 };
 
 // src/websocket/web-socket-client.ts
-function changer(sink, input) {
-  return (change) => {
-    const i = input.deref();
-    if (i) {
-      sink.apply(change, i);
-    }
-  };
+var DISCONNECTED_ERROR = "WebSocket is disconnected";
+var CONNECTION_CLOSED_ERROR = "WebSocket connection closed";
+function randomJitter(range) {
+  if (range <= 0) return 0;
+  return (Math.random() * 2 - 1) * range;
 }
 var WebSocketClient = class {
-  constructor(ws, sinks, graph) {
-    this.ws = ws;
+  constructor(url, sinks, graph, reconnectOptions = {}) {
+    this.url = url;
     this.sinks = sinks;
     this.graph = graph;
     this.nextId = 1;
     this.pendingStreams = /* @__PURE__ */ new Map();
     this.pendingMutations = /* @__PURE__ */ new Map();
     this.activeStreams = /* @__PURE__ */ new Map();
+    this.reconnectAttempts = 0;
+    this.connected = false;
+    this.closed = false;
     this.registry = new FinalizationRegistry(
       ([id, name]) => {
         console.log(`\u{1F9F9} Stream ${id} (${name}) collected \u2014 unsubscribing`);
@@ -81,14 +82,142 @@ var WebSocketClient = class {
         this.activeStreams.delete(id);
       }
     );
-    this.ws.onmessage = (event) => {
+    var _a, _b, _c, _d;
+    this.reconnectOptions = {
+      baseDelayMs: (_a = reconnectOptions.baseDelayMs) != null ? _a : 500,
+      maxDelayMs: (_b = reconnectOptions.maxDelayMs) != null ? _b : 15e3,
+      jitterRatio: (_c = reconnectOptions.jitterRatio) != null ? _c : 0.2,
+      maxReconnectAttempts: (_d = reconnectOptions.maxReconnectAttempts) != null ? _d : Number.POSITIVE_INFINITY
+    };
+    this.ws = this.createSocket();
+    this.bindSocket(this.ws);
+    this.connected = this.isSocketOpen();
+    if (this.connected) {
+      this.resetHeartbeat();
+      this.resetInactivity();
+    }
+  }
+  createSocket() {
+    return new globalThis.WebSocket(this.url);
+  }
+  bindSocket(socket) {
+    this.ws = socket;
+    socket.onopen = () => {
+      if (this.closed || this.ws !== socket) {
+        return;
+      }
+      this.handleSocketOpen();
+    };
+    socket.onmessage = (event) => {
+      if (this.closed || this.ws !== socket) {
+        return;
+      }
       const message = JSON.parse(event.data);
       this.handleMessage(message);
     };
+    socket.onerror = () => {
+      if (this.closed || this.ws !== socket) {
+        return;
+      }
+      this.handleTransportClose();
+    };
+    socket.onclose = () => {
+      if (this.closed || this.ws !== socket) {
+        return;
+      }
+      this.handleTransportClose();
+    };
+  }
+  handleSocketOpen() {
+    this.connected = true;
+    this.reconnectAttempts = 0;
+    if (this.reconnectTimeout) {
+      clearTimeout(this.reconnectTimeout);
+      this.reconnectTimeout = void 0;
+    }
     this.resetHeartbeat();
     this.resetInactivity();
+    this.resubscribeAll();
+    if (this.currentPresence) {
+      this.sendMessage(ClientMessage.presence(this.currentPresence));
+    }
+  }
+  handleTransportClose() {
+    if (!this.connected && this.reconnectTimeout) {
+      return;
+    }
+    this.connected = false;
+    if (this.heartbeatTimeout) {
+      clearTimeout(this.heartbeatTimeout);
+      this.heartbeatTimeout = void 0;
+    }
+    if (this.inactivityTimeout) {
+      clearTimeout(this.inactivityTimeout);
+      this.inactivityTimeout = void 0;
+    }
+    this.failPendingMutations(DISCONNECTED_ERROR);
+    this.scheduleReconnect();
+  }
+  scheduleReconnect() {
+    if (this.closed || this.reconnectTimeout) {
+      return;
+    }
+    if (this.reconnectAttempts >= this.reconnectOptions.maxReconnectAttempts) {
+      this.failPendingStreams(CONNECTION_CLOSED_ERROR);
+      return;
+    }
+    const exponent = Math.min(this.reconnectAttempts, 30);
+    const baseDelay = this.reconnectOptions.baseDelayMs * 2 ** exponent;
+    const cappedDelay = Math.min(baseDelay, this.reconnectOptions.maxDelayMs);
+    const jitter = cappedDelay * this.reconnectOptions.jitterRatio;
+    const delay = Math.max(0, Math.round(cappedDelay + randomJitter(jitter)));
+    this.reconnectAttempts += 1;
+    this.reconnectTimeout = setTimeout(() => {
+      this.reconnectTimeout = void 0;
+      if (this.closed) {
+        return;
+      }
+      try {
+        const socket = this.createSocket();
+        this.bindSocket(socket);
+        this.connected = this.isSocketOpen();
+        if (this.connected) {
+          this.handleSocketOpen();
+        }
+      } catch (err) {
+        console.error("Failed to create reconnect socket:", err);
+        this.scheduleReconnect();
+      }
+    }, delay);
+  }
+  resubscribeAll() {
+    for (const [id, pending] of this.pendingStreams) {
+      this.sendMessage(ClientMessage.subscribe(id, pending.name, pending.args));
+    }
+    for (const [id, active] of this.activeStreams) {
+      if (!active.input.deref()) {
+        this.activeStreams.delete(id);
+        continue;
+      }
+      this.sendMessage(ClientMessage.subscribe(id, active.name, active.args));
+    }
+  }
+  failPendingStreams(message) {
+    for (const [id, pending] of this.pendingStreams) {
+      pending.reject(new Error(message));
+      this.pendingStreams.delete(id);
+    }
+  }
+  failPendingMutations(message) {
+    for (const [id, resolve] of this.pendingMutations) {
+      resolve({ success: false, error: message });
+      this.pendingMutations.delete(id);
+    }
   }
   resetHeartbeat() {
+    if (!this.connected || this.closed) {
+      return;
+    }
     if (this.heartbeatTimeout) {
       clearTimeout(this.heartbeatTimeout);
     }
@@ -97,11 +226,18 @@ var WebSocketClient = class {
     }, 1e4);
   }
   resetInactivity() {
+    if (!this.connected || this.closed) {
+      return;
+    }
     if (this.inactivityTimeout) {
       clearTimeout(this.inactivityTimeout);
     }
     this.inactivityTimeout = setTimeout(() => {
-      this.close();
+      try {
+        this.ws.close();
+      } catch (e) {
+        this.handleTransportClose();
+      }
     }, 3e4);
   }
   handleMessage(message) {
@@ -109,25 +245,46 @@ var WebSocketClient = class {
     switch (message.type) {
       case "snapshot": {
         console.log(`[WebSocketClient] Received snapshot for stream ${message.id}`);
-        const resolve = this.pendingStreams.get(message.id);
-        if (resolve) {
-          resolve(message.snapshot);
+        const pending = this.pendingStreams.get(message.id);
+        if (pending) {
+          pending.resolve(message.snapshot);
           this.pendingStreams.delete(message.id);
+          break;
+        }
+        const active = this.activeStreams.get(message.id);
+        if (!active) {
+          break;
         }
+        const input = active.input.deref();
+        if (!input) {
+          this.sendMessage(ClientMessage.unsubscribe(message.id));
+          this.activeStreams.delete(message.id);
+          break;
+        }
+        active.sink.reset(message.snapshot, input);
+        this.graph.step();
         break;
       }
       case "delta": {
         const changeCount = Object.keys(message.changes).length;
-        console.log(`[WebSocketClient] Received delta with ${changeCount} changes for streams: ${Object.keys(message.changes).join(", ")}`);
+        console.log(
+          `[WebSocketClient] Received delta with ${changeCount} changes for streams: ${Object.keys(message.changes).join(", ")}`
+        );
         for (const [idStr, change] of Object.entries(message.changes)) {
           const id = Number(idStr);
-          const sink = this.activeStreams.get(id);
-          if (sink && change && typeof change === "object") {
-            sink(change);
-          } else if (!sink) {
-            console.log(`\u{1F9F9} Sink ${id} GC'd \u2014 auto-unsubscribing`);
+          const active = this.activeStreams.get(id);
+          if (!active) {
+            this.sendMessage(ClientMessage.unsubscribe(id));
+            continue;
+          }
+          const input = active.input.deref();
+          if (!input) {
             this.sendMessage(ClientMessage.unsubscribe(id));
             this.activeStreams.delete(id);
+            continue;
+          }
+          if (change && typeof change === "object") {
+            active.sink.apply(change, input);
           }
         }
         console.log("[WebSocketClient] Stepping graph");
@@ -135,19 +292,23 @@ var WebSocketClient = class {
         break;
       }
       case "result": {
-        console.log(`[WebSocketClient] Received mutation result for call ${message.id}:`, message.success ? "success" : "error");
+        console.log(
+          `[WebSocketClient] Received mutation result for call ${message.id}:`,
+          message.success ? "success" : "error"
+        );
         const resolve = this.pendingMutations.get(message.id);
-        if (resolve) {
-          if (message.success) {
-            resolve({ success: true, value: message.value });
-          } else {
-            resolve({
-              success: false,
-              error: message.error || "Unknown error"
-            });
-          }
-          this.pendingMutations.delete(message.id);
+        if (!resolve) {
+          break;
+        }
+        if (message.success) {
+          resolve({ success: true, value: message.value });
+        } else {
+          resolve({
+            success: false,
+            error: message.error || "Unknown error"
+          });
         }
+        this.pendingMutations.delete(message.id);
         break;
       }
       case "heartbeat":
@@ -155,52 +316,105 @@ var WebSocketClient = class {
         break;
     }
   }
+  isSocketOpen() {
+    return this.ws.readyState === this.ws.OPEN;
+  }
   sendMessage(message) {
+    if (this.closed) {
+      return false;
+    }
+    if (!this.isSocketOpen()) {
+      return false;
+    }
     this.resetHeartbeat();
-    this.ws.send(JSON.stringify(message));
+    try {
+      this.ws.send(JSON.stringify(message));
+      return true;
+    } catch (e) {
+      this.handleTransportClose();
+      return false;
+    }
   }
   async run(key, args) {
     console.log(
       `Running stream ${String(key)} with args ${JSON.stringify(args)}`
     );
     const id = this.nextId++;
-    this.sendMessage(ClientMessage.subscribe(id, String(key), args));
-    const snapshot = await new Promise((resolve) => {
-      this.pendingStreams.set(id, resolve);
+    const name = String(key);
+    const typedArgs = args;
+    const snapshot = await new Promise((resolve, reject) => {
+      this.pendingStreams.set(id, { name, args: typedArgs, resolve, reject });
+      this.sendMessage(ClientMessage.subscribe(id, name, typedArgs));
     });
     const endpoint = this.sinks[key];
     const sinkAdapter = endpoint(snapshot);
     const { stream, input } = sinkAdapter.build();
     const inputRef = new WeakRef(input);
-    this.activeStreams.set(id, changer(sinkAdapter, inputRef));
-    this.registry.register(input, [id, String(key)]);
+    this.activeStreams.set(id, {
+      name,
+      args: typedArgs,
+      sink: sinkAdapter,
+      input: inputRef
+    });
+    this.registry.register(input, [id, name]);
     return stream;
   }
   async call(key, args) {
     console.log(
       `Calling mutation ${String(key)} with args ${JSON.stringify(args)}`
     );
+    if (!this.connected || !this.isSocketOpen()) {
+      return {
+        success: false,
+        error: DISCONNECTED_ERROR
+      };
+    }
     const id = this.nextId++;
-    this.sendMessage(
-      ClientMessage.call(id, String(key), args)
-    );
-    const result = await new Promise((resolve) => {
+    const resultPromise = new Promise((resolve) => {
       this.pendingMutations.set(
         id,
         resolve
       );
     });
-    return result;
+    const sent = this.sendMessage(
+      ClientMessage.call(id, String(key), args)
+    );
+    if (!sent) {
+      this.pendingMutations.delete(id);
+      return {
+        success: false,
+        error: DISCONNECTED_ERROR
+      };
+    }
+    return resultPromise;
   }
   close() {
-    clearTimeout(this.heartbeatTimeout);
-    clearTimeout(this.inactivityTimeout);
+    if (this.closed) {
+      return;
+    }
+    this.closed = true;
+    this.connected = false;
+    if (this.heartbeatTimeout) {
+      clearTimeout(this.heartbeatTimeout);
+      this.heartbeatTimeout = void 0;
+    }
+    if (this.inactivityTimeout) {
+      clearTimeout(this.inactivityTimeout);
+      this.inactivityTimeout = void 0;
+    }
+    if (this.reconnectTimeout) {
+      clearTimeout(this.reconnectTimeout);
+      this.reconnectTimeout = void 0;
+    }
+    this.failPendingStreams(CONNECTION_CLOSED_ERROR);
+    this.failPendingMutations(CONNECTION_CLOSED_ERROR);
     try {
       this.ws.close();
     } catch (e) {
     }
   }
   setPresence(value) {
+    this.currentPresence = value;
     this.sendMessage(ClientMessage.presence(value));
   }
 };