From f9dda49886558f0a3d3e2415c04210bc0d070946 Mon Sep 17 00:00:00 2001
From: Alan Ko <34646302+alanko0511@users.noreply.github.com>
Date: Thu, 12 Sep 2024 12:54:00 -0400
Subject: [PATCH] tmp

---
 packages/api-client-core/src/GadgetConnection.ts | 15 ++++++++++++---
 1 file changed, 12 insertions(+), 3 deletions(-)

diff --git a/packages/api-client-core/src/GadgetConnection.ts b/packages/api-client-core/src/GadgetConnection.ts
index 1a6b834c7..16aa04e1f 100644
--- a/packages/api-client-core/src/GadgetConnection.ts
+++ b/packages/api-client-core/src/GadgetConnection.ts
@@ -1,7 +1,7 @@
 /* eslint-disable @typescript-eslint/no-non-null-assertion */
 import type { DefinitionNode, DirectiveNode, OperationDefinitionNode } from "@0no-co/graphql.web";
-import type { ClientOptions, RequestPolicy } from "@urql/core";
-import { Client, cacheExchange, fetchExchange, subscriptionExchange } from "@urql/core";
+import type { ClientOptions, RequestPolicy, SSRExchange } from "@urql/core";
+import { Client, cacheExchange, fetchExchange, ssrExchange, subscriptionExchange } from "@urql/core";
 import type { ExecutionResult } from "graphql";
 import type { Sink, Client as SubscriptionClient, ClientOptions as SubscriptionClientOptions } from "graphql-ws";
 import { CloseCode, createClient as createSubscriptionClient } from "graphql-ws";
@@ -108,6 +108,8 @@ export class GadgetConnection {
   private requestPolicy: RequestPolicy;
   createSubscriptionClient: typeof createSubscriptionClient;
 
+  ssrExchange?: SSRExchange;
+
   constructor(readonly options: GadgetConnectionOptions) {
     if (!options.endpoint) throw new Error("Must provide an `endpoint` option for a GadgetConnection to connect to");
     this.endpoint = options.endpoint;
@@ -363,7 +365,14 @@ export class GadgetConnection {
   }
 
   private newBaseClient() {
-    const exchanges = [...this.exchanges.beforeAll, operationNameExchange, urlParamExchange];
+    this.ssrExchange = ssrExchange({
+      isClient: typeof window !== "undefined",
+      initialState: typeof window !== "undefined" ? (window as any).__URQL_DATA__ : undefined,
+      staleWhileRevalidate: true,
+    });
+    this.ssrExchange.extractData();
+
+    const exchanges = [...this.exchanges.beforeAll, operationNameExchange, urlParamExchange, this.ssrExchange];
 
     // apply urql's default caching behaviour when client side (but skip it server side)
     if (typeof window != "undefined") {