From 003c76941bab3e4e2b0448310bb6c8e6a3277824 Mon Sep 17 00:00:00 2001
From: Bill Min <nmin@ebay.com>
Date: Thu, 2 May 2024 18:29:21 -0400
Subject: [PATCH] feat: add fetch imp as third param in custom fetch

---
 README.md                                     |  4 +-
 .../renders.expected/loading.0.html           |  5 ++
 .../renders.expected/loading.1.html           | 13 ++++
 .../renders.expected/loading.2.html           |  9 +++
 .../isomorphic-fetch-imp/components/app.marko | 12 ++++
 .../fixtures/isomorphic-fetch-imp/index.marko | 13 ++++
 .../gql-query/__tests__/server.test.ts        |  4 ++
 src/node_modules/@internal/client/browser.ts  | 12 +++-
 src/node_modules/@internal/client/node.ts     | 65 ++++++++++++-------
 9 files changed, 112 insertions(+), 25 deletions(-)
 create mode 100644 src/components/gql-query/__tests__/__snapshots__/isomorphic-fetch-imp/renders.expected/loading.0.html
 create mode 100644 src/components/gql-query/__tests__/__snapshots__/isomorphic-fetch-imp/renders.expected/loading.1.html
 create mode 100644 src/components/gql-query/__tests__/__snapshots__/isomorphic-fetch-imp/renders.expected/loading.2.html
 create mode 100644 src/components/gql-query/__tests__/fixtures/isomorphic-fetch-imp/components/app.marko
 create mode 100644 src/components/gql-query/__tests__/fixtures/isomorphic-fetch-imp/index.marko

diff --git a/README.md b/README.md
index b800342..daaccec 100644
--- a/README.md
+++ b/README.md
@@ -108,10 +108,12 @@ This attribute allows you to pass a custom `fetch` implementation.
 
 In the following example we'll add a token to each fetch request that our Client sends to our GraphQL API.
 
+> Note: fetchImp as third parameter is the default implementation based on environment. It is `fetch` API in browser and `make-fetch-happen` in node.
+
 ```marko
 <gql-client
   url="http://localhost:3000/graphql"
-  fetch=((resource, options) => {
+  fetch=((resource, options, fetchImp) => {
     const token = getToken();
     return fetch(resource, {
       ...options,
diff --git a/src/components/gql-query/__tests__/__snapshots__/isomorphic-fetch-imp/renders.expected/loading.0.html b/src/components/gql-query/__tests__/__snapshots__/isomorphic-fetch-imp/renders.expected/loading.0.html
new file mode 100644
index 0000000..d9eed76
--- /dev/null
+++ b/src/components/gql-query/__tests__/__snapshots__/isomorphic-fetch-imp/renders.expected/loading.0.html
@@ -0,0 +1,5 @@
+<div>
+  <noscript
+    id="GENERATED-0"
+  />
+</div>
\ No newline at end of file
diff --git a/src/components/gql-query/__tests__/__snapshots__/isomorphic-fetch-imp/renders.expected/loading.1.html b/src/components/gql-query/__tests__/__snapshots__/isomorphic-fetch-imp/renders.expected/loading.1.html
new file mode 100644
index 0000000..8a37fa4
--- /dev/null
+++ b/src/components/gql-query/__tests__/__snapshots__/isomorphic-fetch-imp/renders.expected/loading.1.html
@@ -0,0 +1,13 @@
+<div>
+  <noscript
+    id="GENERATED-0"
+  />
+</div>
+<div
+  id="GENERATED-1"
+  style="display:none"
+>
+  <span>
+    Hello world!
+  </span>
+</div>
\ No newline at end of file
diff --git a/src/components/gql-query/__tests__/__snapshots__/isomorphic-fetch-imp/renders.expected/loading.2.html b/src/components/gql-query/__tests__/__snapshots__/isomorphic-fetch-imp/renders.expected/loading.2.html
new file mode 100644
index 0000000..6f301af
--- /dev/null
+++ b/src/components/gql-query/__tests__/__snapshots__/isomorphic-fetch-imp/renders.expected/loading.2.html
@@ -0,0 +1,9 @@
+<div>
+  <span>
+    Hello world!
+  </span>
+</div>
+<div
+  id="GENERATED-0"
+  style="display:none"
+/>
\ No newline at end of file
diff --git a/src/components/gql-query/__tests__/fixtures/isomorphic-fetch-imp/components/app.marko b/src/components/gql-query/__tests__/fixtures/isomorphic-fetch-imp/components/app.marko
new file mode 100644
index 0000000..254af2b
--- /dev/null
+++ b/src/components/gql-query/__tests__/fixtures/isomorphic-fetch-imp/components/app.marko
@@ -0,0 +1,12 @@
+import { gql } from "../../../../../../index";
+static const QUERY = gql`
+  {
+    hello
+  }
+`;
+
+class {}
+
+<gql-query query=QUERY>
+  <@then|{ data }|><span>${data.hello}</span></@then>
+</gql-query>
diff --git a/src/components/gql-query/__tests__/fixtures/isomorphic-fetch-imp/index.marko b/src/components/gql-query/__tests__/fixtures/isomorphic-fetch-imp/index.marko
new file mode 100644
index 0000000..8242a20
--- /dev/null
+++ b/src/components/gql-query/__tests__/fixtures/isomorphic-fetch-imp/index.marko
@@ -0,0 +1,13 @@
+<!DOCTYPE html>
+<html lang="en">
+<head>
+  <meta charset="UTF-8">
+  <meta name="viewport" content="width=device-width, initial-scale=1.0">
+  <title>Simple Example</title>
+  <esbuild-assets/>
+</head>
+<body>
+  <gql-client url=input.graphqlURL fetch=((resource, options, fetchImp) => fetchImp(resource, options)) />
+  <app />
+</body>
+</html>
\ No newline at end of file
diff --git a/src/components/gql-query/__tests__/server.test.ts b/src/components/gql-query/__tests__/server.test.ts
index ecf72fd..6f747cb 100644
--- a/src/components/gql-query/__tests__/server.test.ts
+++ b/src/components/gql-query/__tests__/server.test.ts
@@ -17,6 +17,10 @@ describe(
 );
 
 describe("isomorphic", fixture(path.join(__dirname, "fixtures/isomorphic")));
+describe(
+  "isomorphic-fetch-imp",
+  fixture(path.join(__dirname, "fixtures/isomorphic-fetch-imp")),
+);
 describe(
   "isomorphic-placeholder",
   fixture(path.join(__dirname, "fixtures/isomorphic-placeholder")),
diff --git a/src/node_modules/@internal/client/browser.ts b/src/node_modules/@internal/client/browser.ts
index 1a05370..c16afd2 100644
--- a/src/node_modules/@internal/client/browser.ts
+++ b/src/node_modules/@internal/client/browser.ts
@@ -20,7 +20,13 @@ export function getClient() {
   return client;
 }
 
-export function configureClient(config: ClientOptions) {
+export function configureClient(
+  config: ClientOptions & {
+    fetch?: (
+      ...params: [...Parameters<typeof fetch>, typeof fetch]
+    ) => ReturnType<typeof fetch>;
+  },
+) {
   ssr = ssrExchange({ isClient: true });
   let exchanges = [dedupExchange, cacheExchange, ssr, fetchExchange];
   if (process.env.NODE_ENV !== "production" && devtoolsExchange) {
@@ -34,7 +40,9 @@ export function configureClient(config: ClientOptions) {
   client = createClient({
     ...config,
     exchanges,
-    fetch: config.fetch || fetch,
+    fetch: config.fetch
+      ? (...params: Parameters<typeof fetch>) => config.fetch!(...params, fetch)
+      : fetch,
   });
 }
 
diff --git a/src/node_modules/@internal/client/node.ts b/src/node_modules/@internal/client/node.ts
index 6591547..f7d3709 100644
--- a/src/node_modules/@internal/client/node.ts
+++ b/src/node_modules/@internal/client/node.ts
@@ -1,4 +1,5 @@
-import fetchImpl from "make-fetch-happen";
+import fetchImpl, { FetchOptions } from "make-fetch-happen";
+import https from "https";
 import {
   createClient,
   dedupExchange,
@@ -9,8 +10,17 @@ import {
   type Client,
 } from "@urql/core";
 
+type FetchFn = (
+  uriOrRequest: string | Request,
+  opts?: FetchOptions,
+) => ReturnType<typeof fetchImpl>;
+
 const kClient = Symbol("client");
+const { ca } = https.globalAgent.options;
 const strictSSL = process.env.NODE_TLS_REJECT_UNAUTHORIZED !== "0";
+function internalFetch(url, options) {
+  return fetchImpl(url, { ...options, ca, strictSSL });
+}
 
 export const readyLookup = {};
 
@@ -20,7 +30,14 @@ export function getClient(out: any): Client {
   return out.global[kClient];
 }
 
-export function configureClient(config: ClientOptions, out: any) {
+export function configureClient(
+  config: ClientOptions & {
+    fetch?: (
+      ...params: [...Parameters<FetchFn>, FetchFn]
+    ) => ReturnType<FetchFn>;
+  },
+  out: any,
+) {
   let exchanges = [
     dedupExchange,
     cacheExchange,
@@ -35,30 +52,34 @@ export function configureClient(config: ClientOptions, out: any) {
   out.global[kClient] = createClient({
     ...config,
     exchanges,
-    fetch:
-      config.fetch ||
-      (async (url, options) => {
-        const incomingMessage =
-          (out.stream && (out.stream.req || out.stream.request)) ||
-          out.global.req ||
-          out.global.request;
-        if (typeof url !== "string") throw new Error("Expected a string.");
-        if (!incomingMessage) return fetchImpl(url, options as any) as any;
-        const protocol =
-          incomingMessage.headers["x-forwarded-proto"] ||
-          incomingMessage.protocol;
-        const host =
-          incomingMessage.headers["x-forwarded-host"] ||
-          incomingMessage.headers.host;
-        return fetchImpl(new URL(url, `${protocol}://${host}`).toString(), {
-          ...(options as any),
-          strictSSL,
+    fetch: async (url, options: any) => {
+      const fetchImplOverride = config.fetch
+        ? (...params: Parameters<FetchFn>) =>
+            config.fetch!(...params, internalFetch)
+        : internalFetch;
+      const incomingMessage =
+        (out.stream && (out.stream.req || out.stream.request)) ||
+        out.global.req ||
+        out.global.request;
+      if (typeof url !== "string") throw new Error("Expected a string.");
+      if (!incomingMessage) return fetchImplOverride(url, options) as any;
+      const protocol =
+        incomingMessage.headers["x-forwarded-proto"] ||
+        incomingMessage.protocol;
+      const host =
+        incomingMessage.headers["x-forwarded-host"] ||
+        incomingMessage.headers.host;
+      return fetchImplOverride(
+        new URL(url, `${protocol}://${host}`).toString(),
+        {
+          ...options,
           headers: {
             ...incomingMessage.headers,
             ...options?.headers,
           },
-        }) as any;
-      }),
+        },
+      ) as any;
+    },
   });
 }