chore: add missing headers

Author: dahal

apps/captable/app/(authenticated)/(dashboard)/[publicId]/documents/[bucketId]/page.tsx

@@ -9,14 +9,15 @@ import { RiArrowLeftSLine } from "@remixicon/react";
 import Link from "next/link";
 import { notFound } from "next/navigation";
 import { Fragment } from "react";
+import { headers } from "next/headers";
 
 const DocumentPreview = async ({
   params,
 }: {
   params: Promise<{ publicId: string; bucketId: string }>;
 }) => {
   const { publicId, bucketId } = await params;
-  const session = await useServerSideSession();
+  const session = await useServerSideSession({ headers: await headers() });
   const companyId = session?.user?.companyId;
   const document = await db
     .select({

apps/captable/app/(authenticated)/(dashboard)/[publicId]/documents/esign/[templatePublicId]/page.tsx

@@ -5,14 +5,15 @@ import { Badge } from "@/components/ui/badge";
 import { TemplateFieldProvider } from "@/providers/template-field-provider";
 import { useServerSideSession } from "@/hooks/use-server-side-session";
 import { api } from "@/trpc/server";
+import { headers } from "next/headers";
 
 const EsignTemplateDetailPage = async ({
   params,
 }: {
   params: Promise<{ templatePublicId: string }>;
 }) => {
   const { templatePublicId } = await params;
-  const session = await useServerSideSession();
+  const session = await useServerSideSession({ headers: await headers() });
 
   const { name, status, url, fields, recipients } =
     await api.template.get.query({
@@ -23,7 +24,7 @@ const EsignTemplateDetailPage = async ({
   return (
     <TemplateFieldProvider recipients={recipients} fields={fields}>
       <TemplateFieldForm
-        companyPublicId={session.user.companyPublicId}
+        companyPublicId={session?.user?.companyPublicId ?? ""}
         templatePublicId={templatePublicId}
       >
         <div className="grid grid-cols-12">

apps/captable/app/(authenticated)/(dashboard)/[publicId]/documents/esign/page.tsx

@@ -7,13 +7,14 @@ import { RiUploadCloudLine } from "@remixicon/react";
 import type { Metadata } from "next";
 import { AddEsignDocumentButton } from "./components/add-esign-doc-button";
 import { ESignTable } from "./components/table";
+import { headers } from "next/headers";
 
 export const metadata: Metadata = {
   title: "Documents",
 };
 
 const EsignDocumentPage = async () => {
-  const session = await useServerSideSession();
+  const session = await useServerSideSession({ headers: await headers() });
   const { documents } = await api.template.all.query();
 
   if (documents.length === 0) {
@@ -26,7 +27,7 @@ const EsignDocumentPage = async () => {
         <AddEsignDocumentButton
           title="esign a Document"
           subtitle=""
-          companyPublicId={session.user.companyPublicId}
+          companyPublicId={session?.user?.companyPublicId ?? ""}
         />
       </EmptyState>
     );
@@ -40,14 +41,14 @@ const EsignDocumentPage = async () => {
         <AddEsignDocumentButton
           title="esign a Document"
           subtitle=""
-          companyPublicId={session.user.companyPublicId}
+          companyPublicId={session?.user?.companyPublicId ?? ""}
         />
       }
     >
       <Card className="mt-3">
         <div className="p-6">
           <ESignTable
-            companyPublicId={session.user.companyPublicId}
+            companyPublicId={session?.user?.companyPublicId ?? ""}
             documents={documents}
           />
         </div>

apps/captable/app/(authenticated)/(dashboard)/[publicId]/documents/share/_page.tsx

@@ -8,14 +8,15 @@ import { RiAddFill, RiUploadCloudLine } from "@remixicon/react";
 import type { Metadata } from "next";
 import DocumentUploadModal from "../components/modal";
 import DocumentsTable from "../components/table";
+import { headers } from "next/headers";
 
 export const metadata: Metadata = {
   title: "Documents",
 };
 
 const DocumentsPage = async () => {
   const documents = await api.document.getAll.query();
-  const session = await useServerSideSession();
+  const session = await useServerSideSession({ headers: await headers() });
 
   if (documents.length === 0) {
     return (
@@ -25,7 +26,7 @@ const DocumentsPage = async () => {
         subtitle="Please click the button below to upload a new document."
       >
         <DocumentUploadModal
-          companyPublicId={session.user.companyPublicId}
+          companyPublicId={session?.user?.companyPublicId ?? ""}
           trigger={
             <Button>
               <RiAddFill className="mr-2 h-5 w-5" />
@@ -44,7 +45,7 @@ const DocumentsPage = async () => {
         description="Share pitch decks, financials, and any other important documents."
         action={
           <DocumentUploadModal
-            companyPublicId={session.user.companyPublicId}
+            companyPublicId={session?.user?.companyPublicId ?? ""}
             trigger={
               <Button>
                 <RiAddFill className="mr-2 h-5 w-5" />
@@ -58,7 +59,7 @@ const DocumentsPage = async () => {
       <Card className="mt-3">
         <div className="p-6">
           <DocumentsTable
-            companyPublicId={session.user.companyPublicId}
+            companyPublicId={session?.user?.companyPublicId ?? ""}
             documents={documents}
           />
         </div>

apps/captable/app/(authenticated)/(dashboard)/[publicId]/equity-plans/page.tsx

@@ -9,6 +9,7 @@ import { RiPieChart2Line } from "@remixicon/react";
 import type { Metadata } from "next";
 import { CreateEquityPlanButton } from "./create-equity-plan-button";
 import EquityPlanTable from "./table";
+import { headers } from "next/headers";
 
 export const metadata: Metadata = {
   title: "Equity plans",
@@ -29,7 +30,7 @@ const getShareClasses = async (companyId: string) => {
 };
 
 const EquityPlanPage = async () => {
-  const session = await useServerSideSession();
+  const session = await useServerSideSession({ headers: await headers() });
   const companyId = session?.user?.companyId;
   let equityPlans: EquityPlanMutationType[] = [];
 
@@ -40,7 +41,7 @@ const EquityPlanPage = async () => {
   }
 
   const shareClasses: ShareClassMutationType[] = (await getShareClasses(
-    companyId,
+    companyId as string,
   )) as unknown as ShareClassMutationType[];
 
   if (equityPlans.length === 0) {

apps/captable/app/(authenticated)/(dashboard)/[publicId]/share-classes/page.tsx

@@ -7,6 +7,7 @@ import { RiPieChart2Line } from "@remixicon/react";
 import type { Metadata } from "next";
 import { CreateShareButton } from "./create-share-class-button";
 import ShareClassTable from "./table";
+import { headers } from "next/headers";
 
 export const metadata: Metadata = {
   title: "Share classes",
@@ -20,7 +21,7 @@ const getShareClasses = async (companyId: string) => {
 };
 
 const SharesPage = async () => {
-  const session = await useServerSideSession();
+  const session = await useServerSideSession({ headers: await headers() });
   const companyId = session?.user?.companyId;
   let shareClasses: ShareClassMutationType[] = [];
 

apps/captable/components/security/SettingHeader.tsx

@@ -3,6 +3,7 @@ import { useServerSideSession } from "@/hooks/use-server-side-session";
 import { RiArrowLeftLine } from "@remixicon/react";
 import Link from "next/link";
 import type React from "react";
+import { headers } from "next/headers";
 
 type SettingsHeaderProps = {
   title: string;
@@ -19,7 +20,7 @@ export const SettingsHeader = async ({
   showBackArrow = true,
   className,
 }: SettingsHeaderProps) => {
-  const session = await useServerSideSession();
+  const session = await useServerSideSession({ headers: await headers() });
 
   const href = `/${session?.user.companyPublicId}/settings/security`;
   return (