refactor: reduce number of woo token generated

Author: pooriaset

src/graphql/clients/clientSideMakeClient.ts

@@ -8,7 +8,25 @@ import { createErrorLink } from '../utils';
 import { PUBLIC_GATEWAY_URL } from '@/config/app';
 import Cookies from 'js-cookie';
 
-export const WOO_SESSION_KEY = 'woo-session';
+const WOO_SESSION_KEY = 'woo-session';
+
+/**
+ * Apollo Link to manage WooCommerce session tokens.
+ * This link sets the session token in the request headers if available.
+ */
+export const wooSessionSetter = new ApolloLink((operation, forward) => {
+  const token = Cookies.get(WOO_SESSION_KEY);
+  if (token) {
+    // Set session token in the request headers
+    operation.setContext({
+      headers: {
+        'woocommerce-session': `Session ${token}`,
+      },
+    });
+  }
+
+  return forward(operation);
+});
 
 /**
  * Apollo Link to update the WooCommerce session token.
@@ -47,6 +65,7 @@ export const makeClient = () => {
     connectToDevTools: true,
     cache: new NextSSRInMemoryCache(),
     link: from([
+      wooSessionSetter,
       wooSessionUpdater,
       createErrorLink(),
       new SSRMultipartLink({ stripDefer: true }),

src/graphql/clients/serverSideClient.ts

@@ -10,6 +10,7 @@ export const { getClient } = registerApolloClient(() => {
     uri: `${process.env.__NEXT_PRIVATE_ORIGIN}${PUBLIC_GATEWAY_URL}`,
     headers: {
       cookie: cookiesStore.toString(),
+      'X-Server-Side': 'true',
     },
   });
 

src/middleware.ts

@@ -4,7 +4,6 @@ import nextIntlMiddleware from 'next-intl/middleware';
 import type { NextRequest } from 'next/server';
 import { NextResponse } from 'next/server';
 import { PUBLIC_GATEWAY_URL, protectedRoutes } from './config/app';
-import { WOO_SESSION_KEY } from './graphql/clients/clientSideMakeClient';
 
 const intlMiddleware = (request: NextRequest) =>
   Promise.resolve(
@@ -20,11 +19,6 @@ async function middleware(request: NextRequestWithAuth) {
   const { pathname } = request.nextUrl;
   request.headers.set('x-pathname', pathname);
 
-  const wooToken = request.cookies.get(WOO_SESSION_KEY)?.value;
-  if (wooToken) {
-    request.headers.set('woocommerce-session', `Session ${wooToken}`);
-  }
-
   if (pathname.startsWith(PUBLIC_GATEWAY_URL)) {
     request.headers.delete('cookie');