From 1daf35406abfcb7a984388291c3de19a87c8c028 Mon Sep 17 00:00:00 2001
From: Henry Dollman <hank@henrygd.me>
Date: Wed, 5 Feb 2025 20:24:39 -0500
Subject: [PATCH] auth: ensure realtime subscriptions are cleared on logout

---
 beszel/site/src/components/navbar.tsx | 4 ++--
 beszel/site/src/lib/utils.ts          | 7 ++++++-
 2 files changed, 8 insertions(+), 3 deletions(-)

diff --git a/beszel/site/src/components/navbar.tsx b/beszel/site/src/components/navbar.tsx
index 034ae9d..89ae1c5 100644
--- a/beszel/site/src/components/navbar.tsx
+++ b/beszel/site/src/components/navbar.tsx
@@ -15,7 +15,7 @@ import { LangToggle } from "./lang-toggle"
 import { ModeToggle } from "./mode-toggle"
 import { Logo } from "./logo"
 import { pb } from "@/lib/stores"
-import { cn, isReadOnlyUser, isAdmin } from "@/lib/utils"
+import { cn, isReadOnlyUser, isAdmin, logOut } from "@/lib/utils"
 import {
 	DropdownMenu,
 	DropdownMenuTrigger,
@@ -99,7 +99,7 @@ export default function Navbar() {
 								</>
 							)}
 						</DropdownMenuGroup>
-						<DropdownMenuItem onSelect={() => pb.authStore.clear()}>
+						<DropdownMenuItem onSelect={logOut}>
 							<LogOutIcon className="me-2.5 h-4 w-4" />
 							<span>
 								<Trans>Log Out</Trans>
diff --git a/beszel/site/src/lib/utils.ts b/beszel/site/src/lib/utils.ts
index 1586b0e..8f8a5e6 100644
--- a/beszel/site/src/lib/utils.ts
+++ b/beszel/site/src/lib/utils.ts
@@ -33,7 +33,7 @@ const verifyAuth = () => {
 	pb.collection("users")
 		.authRefresh()
 		.catch(() => {
-			pb.authStore.clear()
+			logOut()
 			toast({
 				title: t`Failed to authenticate`,
 				description: t`Please log in again`,
@@ -65,6 +65,11 @@ export const updateSystemList = (() => {
 	}
 })()
 
+/** Logs the user out by clearing the auth store and unsubscribing from realtime updates. */
+export async function logOut() {
+	pb.authStore.clear()
+	pb.realtime.unsubscribe()
+}
 
 export const updateAlerts = () => {
 	pb.collection("alerts")