nshop-front/src/lib/apollo-client.ts

import { cache as reactCache } from "react";
import { GRAPHQL_URL } from "@/utils/constants";
import {
  ApolloClient,
  HttpLink,
  InMemoryCache,
  from,
} from "@apollo/client";
import { setContext } from "@apollo/client/link/context";
import { getSession } from "next-auth/react";
import { getCartToken } from "@/utils/getCartToken";
import { BagistoSession } from "@/types/types";



let sessionCache: { session: BagistoSession | null; timestamp: number } | null = null;
const SESSION_CACHE_TTL = 5000;

const getSessionForRequest = reactCache(async () => {
  return (await getSession()) as BagistoSession | null;
});

async function getCachedSession(): Promise<BagistoSession | null> {
  if (typeof window === "undefined") {
    return getSessionForRequest();
  }

  const now = Date.now();

  if (sessionCache && now - sessionCache.timestamp < SESSION_CACHE_TTL) {
    return sessionCache.session;
  }
  const session = (await getSession()) as BagistoSession | null;
  sessionCache = { session, timestamp: now };
  return session;
}

function createApolloClient() {
  const ssrMode = typeof window === "undefined";
  const cache = new InMemoryCache();

  // 服务端渲染的话请求的地址是bagisto后端地址，否则的话请求的是nextjs的代理接口
  const httpLink = new HttpLink({
    uri: ssrMode ? GRAPHQL_URL : "/api/graphql",
    credentials: "include",
  });

  // 如果是服务端渲染，因为直接调用bagisto后端接口，需要设置storefront key
  const authLink = setContext(async (_, { headers }) => {
    if (ssrMode) {
      const storefrontKey =
        process.env.BAGISTO_STOREFRONT_KEY ||
        process.env.NEXT_PUBLIC_BAGISTO_STOREFRONT_KEY ||
        "";

      return {
        headers: {
          ...headers,
          "X-STOREFRONT-KEY": storefrontKey,
        },
      };
    }

    const session = await getCachedSession();
    const userToken = session?.user?.accessToken;
    const guestToken = !userToken ? getCartToken() : null;
    const token = userToken || guestToken;

    return {
      headers: {
        ...headers,
        ...(token && { Authorization: `Bearer ${token}` }),
        "Content-Type": "application/json",
      },
    };
  });

  const link = from([authLink, httpLink]);

  return new ApolloClient({
    ssrMode,
    link,
    cache,
    defaultOptions: {
      watchQuery: {
        fetchPolicy: ssrMode ? "network-only" : "cache-first",
        nextFetchPolicy: ssrMode ? "network-only" : "cache-first",
      },
      query: {
        fetchPolicy: ssrMode ? "network-only" : "cache-first",
      },
    },
  });
}

const getClient = reactCache(createApolloClient);

export default function initializeApollo() {
  if (typeof window === "undefined") {
    return getClient();
  }
  return createApolloClient();
}