fix: only forward required cookie header (t3-oss#1644)

Co-authored-by: Josip Igrec <[email protected]>
mittalsam98 · Nov 2, 2023 · e6649f1 · e6649f1
1 parent 2052334
commit e6649f1
Show file tree

Hide file tree

Showing 5 changed files with 23 additions and 12 deletions.
diff --git a/.changeset/smooth-radios-sparkle.md b/.changeset/smooth-radios-sparkle.md
@@ -0,0 +1,5 @@
+---
+"create-t3-app": patch
+---
+
+fix: only forward required cookie header
diff --git a/cli/template/extras/src/app/layout/with-trpc-tw.tsx b/cli/template/extras/src/app/layout/with-trpc-tw.tsx
@@ -1,7 +1,7 @@
 import "~/styles/globals.css";
 
 import { Inter } from "next/font/google";
-import { headers } from "next/headers";
+import { cookies } from "next/headers";
 
 import { TRPCReactProvider } from "~/trpc/react";
 
@@ -24,7 +24,9 @@ export default function RootLayout({
  return (
  <html lang="en">
  <body className={`font-sans ${inter.variable}`}>
- <TRPCReactProvider headers={headers()}>{children}</TRPCReactProvider>
+ <TRPCReactProvider cookies={cookies().toString()}>
+ {children}
+ </TRPCReactProvider>
  </body>
  </html>
  );

diff --git a/cli/template/extras/src/app/layout/with-trpc.tsx b/cli/template/extras/src/app/layout/with-trpc.tsx
@@ -1,7 +1,7 @@
 import "~/styles/globals.css";
 
 import { Inter } from "next/font/google";
-import { headers } from "next/headers";
+import { cookies } from "next/headers";
 
 import { TRPCReactProvider } from "~/trpc/react";
 
@@ -23,7 +23,9 @@ export default function RootLayout({
  return (
  <html lang="en">
  <body className={inter.className}>
- <TRPCReactProvider headers={headers()}>{children}</TRPCReactProvider>
+ <TRPCReactProvider cookies={cookies().toString()}>
+ {children}
+ </TRPCReactProvider>
  </body>
  </html>
  );

diff --git a/cli/template/extras/src/trpc/react.tsx b/cli/template/extras/src/trpc/react.tsx
@@ -12,7 +12,7 @@ export const api = createTRPCReact<AppRouter>();
 
 export function TRPCReactProvider(props: {
  children: React.ReactNode;
- headers: Headers;
+ cookies: string;
 }) {
  const [queryClient] = useState(() => new QueryClient());
 
@@ -28,9 +28,10 @@ export function TRPCReactProvider(props: {
  unstable_httpBatchStreamLink({
  url: getUrl(),
  headers() {
- const heads = new Map(props.headers);
- heads.set("x-trpc-source", "react");
- return Object.fromEntries(heads);
+ return {
+ cookie: props.cookies,
+ "x-trpc-source": "react",
+ };
  },
  }),
  ],

diff --git a/cli/template/extras/src/trpc/server.ts b/cli/template/extras/src/trpc/server.ts
@@ -3,7 +3,7 @@ import {
  loggerLink,
  unstable_httpBatchStreamLink,
 } from "@trpc/client";
-import { headers } from "next/headers";
+import { cookies } from "next/headers";
 
 import { type AppRouter } from "~/server/api/root";
 import { getUrl, transformer } from "./shared";
@@ -19,9 +19,10 @@ export const api = createTRPCProxyClient<AppRouter>({
  unstable_httpBatchStreamLink({
  url: getUrl(),
  headers() {
- const heads = new Map(headers());
- heads.set("x-trpc-source", "rsc");
- return Object.fromEntries(heads);
+ return {
+ cookie: cookies().toString(),
+ "x-trpc-source": "rsc",
+ };
  },
  }),
  ],