From c012f3b5c454121a547fe54e6ecbd6e11b465730 Mon Sep 17 00:00:00 2001
From: Prasanna721 <106952318+Prasanna721@users.noreply.github.com>
Date: Fri, 3 Apr 2026 01:55:24 +0000
Subject: [PATCH] fix stale session cookie (#823)

- redirect to login when session is gone instead of blank screen
- show cached username while session restores so header doesn't flicker
- cleaned up redundant type casts and unused vars
---
 apps/web/components/ensure-workspace.tsx | 10 ++++++++--
 apps/web/components/header.tsx           | 14 ++++++++------
 packages/lib/auth-context.tsx            |  8 +++-----
 3 files changed, 19 insertions(+), 13 deletions(-)

diff --git a/apps/web/components/ensure-workspace.tsx b/apps/web/components/ensure-workspace.tsx
index a9bd8823..370d5ca0 100644
--- a/apps/web/components/ensure-workspace.tsx
+++ b/apps/web/components/ensure-workspace.tsx
@@ -10,8 +10,14 @@ export function EnsureWorkspace({ children }: { children: React.ReactNode }) {
 	const { session, organizations, isRestoring } = useAuth()
 
 	useEffect(() => {
-		if (!session) return
-		if (isRestoring || organizations === null) return
+		if (isRestoring) return
+		if (!session) {
+			router.replace(
+				`/login?redirect=${encodeURIComponent(window.location.href)}`,
+			)
+			return
+		}
+		if (organizations === null) return
 		if (organizations.length > 0) return
 		if (pathname.startsWith("/onboarding")) return
 		router.replace("/onboarding/welcome?step=input")
diff --git a/apps/web/components/header.tsx b/apps/web/components/header.tsx
index 38c5c68c..a66626b0 100644
--- a/apps/web/components/header.tsx
+++ b/apps/web/components/header.tsx
@@ -34,7 +34,7 @@ import { useIsMobile } from "@hooks/use-mobile"
 import { useLocalStorageUsername } from "@hooks/use-local-storage-username"
 import { UserProfileMenu } from "@/components/user-profile-menu"
 import { FeedbackModal } from "./feedback-modal"
-import { useViewMode } from "@/lib/view-mode-context"
+import { useViewMode, type ViewMode } from "@/lib/view-mode-context"
 import { useQueryState } from "nuqs"
 import { feedbackParam } from "@/lib/search-params"
 
@@ -45,7 +45,7 @@ interface HeaderProps {
 }
 
 export function Header({ onAddMemory, onOpenChat, onOpenSearch }: HeaderProps) {
-	const { user } = useAuth()
+	const { user, isRestoring } = useAuth()
 	const { selectedProjects, setSelectedProjects } = useProject()
 	const router = useRouter()
 	const isMobile = useIsMobile()
@@ -53,14 +53,16 @@ export function Header({ onAddMemory, onOpenChat, onOpenSearch }: HeaderProps) {
 		"feedback",
 		feedbackParam,
 	)
-	const isFeedbackOpen = feedbackOpen ?? false
 	const { viewMode, setViewMode } = useViewMode()
 
 	const handleFeedback = () => setFeedbackOpen(true)
 
 	const localStorageUsername = useLocalStorageUsername()
 	const displayName =
-		user?.displayUsername || localStorageUsername || user?.name || ""
+		user?.displayUsername ||
+		(isRestoring ? localStorageUsername : "") ||
+		user?.name ||
+		""
 	const userName = displayName ? `${displayName.split(" ")[0]}'s` : "My"
 	return (
 		<div className="flex p-3 md:p-4 justify-between items-center gap-2">
@@ -141,7 +143,7 @@ export function Header({ onAddMemory, onOpenChat, onOpenSearch }: HeaderProps) {
 				<Tabs
 					value={viewMode === "list" ? "grid" : viewMode}
 					onValueChange={(v) =>
-						setViewMode(v === "grid" ? "list" : (v as "graph" | "integrations"))
+						setViewMode(v === "grid" ? "list" : (v as ViewMode))
 					}
 				>
 					<TabsList className="rounded-full border border-[#161F2C] h-11! z-10!">
@@ -307,7 +309,7 @@ export function Header({ onAddMemory, onOpenChat, onOpenSearch }: HeaderProps) {
 				<UserProfileMenu />
 			</div>
 			<FeedbackModal
-				isOpen={isFeedbackOpen}
+				isOpen={feedbackOpen}
 				onClose={() => setFeedbackOpen(false)}
 			/>
 		</div>
diff --git a/packages/lib/auth-context.tsx b/packages/lib/auth-context.tsx
index cfb9b9ab..83f65f46 100644
--- a/packages/lib/auth-context.tsx
+++ b/packages/lib/auth-context.tsx
@@ -42,7 +42,7 @@ export function AuthProvider({ children }: { children: ReactNode }) {
 		isPending: orgsPending,
 	} = authClient.useListOrganizations()
 
-	const organizations: OrganizationListItem[] | null =
+	const organizations =
 		session?.session == null ? null : orgsPending ? null : (orgsData ?? [])
 
 	const refetchOrganizations = useCallback(
@@ -74,9 +74,7 @@ export function AuthProvider({ children }: { children: ReactNode }) {
 	}, [])
 
 	useEffect(() => {
-		if (isSessionPending) {
-			return
-		}
+		if (isSessionPending) return
 
 		if (!session?.session) {
 			setIsRestoring(false)
@@ -150,7 +148,7 @@ export function AuthProvider({ children }: { children: ReactNode }) {
 		return () => {
 			cancelled = true
 		}
-	}, [session, isSessionPending, orgsData, orgsPending, setActiveOrg])
+	}, [isSessionPending, session, orgsData, orgsPending, setActiveOrg])
 
 	useEffect(() => {
 		if (typeof window === "undefined") return