spacetimedb 2.0.2 → 2.0.4

package/dist/tanstack/index.mjs

@@ -583,6 +583,9 @@ var BinaryWriter = class {
   constructor(init) {
     this.buffer = typeof init === "number" ? new ResizableBuffer(init) : init;
   }
+  clear() {
+    this.offset = 0;
+  }
   reset(buffer) {
     this.buffer = buffer;
     this.offset = 0;
@@ -1400,6 +1403,9 @@ var ConnectionId = class _ConnectionId {
     }
   }
 };
+function toSql(q) {
+  return q.toSql();
+}
 function isLiteralExpr(expr) {
   return expr.type === "literal";
 }
@@ -1476,7 +1482,7 @@ function spacetimeDBQuery(queryOrSkip) {
   const query = queryOrSkip;
   const accessorName = getQueryAccessorName(query);
   const whereExpr = getQueryWhereClause(query);
-  const querySql = query.toSql();
+  const querySql = toSql(query);
   queryRegistry.set(querySql, { accessorName, whereExpr });
   return {
     queryKey: ["spacetimedb", accessorName, querySql],
@@ -1493,6 +1499,7 @@ var SpacetimeDBQueryClient = class {
   setConnection(connection) {
     this.connection = connection;
     this.processPendingQueries();
+    this.hydrateSubscriptions();
   }
   connect(queryClient) {
     this.queryClient = queryClient;
@@ -1634,6 +1641,26 @@ var SpacetimeDBQueryClient = class {
       }
     }
   }
+  // subscribe to queries with SSR cached data but no active subscription
+  hydrateSubscriptions() {
+    if (!this.connection || !this.queryClient) {
+      return;
+    }
+    for (const [querySql, { accessorName, whereExpr }] of queryRegistry) {
+      const queryKey = ["spacetimedb", accessorName, querySql];
+      const keyStr = JSON.stringify(queryKey);
+      if (this.subscriptions.has(keyStr)) {
+        continue;
+      }
+      if (this.queryClient.getQueryData(queryKey) === void 0) {
+        continue;
+      }
+      this.setupSubscription(queryKey, accessorName, querySql, whereExpr).catch(
+        () => {
+        }
+      );
+    }
+  }
   // clean up all subscriptions and disconnect
   disconnect() {
     if (this.cacheUnsubscribe) {