From 0b8bea393e98559f1a6cbb6e332ba2336879353b Mon Sep 17 00:00:00 2001
From: Eli Ribble <eli@gleipnir.technology>
Date: Sun, 22 Mar 2026 08:04:28 +0000
Subject: [PATCH] Fix updates to notification counts

---
 ts/components/layout/Sidebar.vue | 42 +++-----------------------------
 ts/store/user.ts                 |  8 ++++++
 2 files changed, 12 insertions(+), 38 deletions(-)

diff --git a/ts/components/layout/Sidebar.vue b/ts/components/layout/Sidebar.vue
index 26f06686..a12c0d3c 100644
--- a/ts/components/layout/Sidebar.vue
+++ b/ts/components/layout/Sidebar.vue
@@ -22,7 +22,7 @@
 					to="/communication"
 					icon="messaging"
 					label="Communication"
-					:notificationCount="notificationCounts.communication"
+					:notificationCount="user?.notification_counts?.communication ?? 0"
 				/>
 			</li>
 			<li>
@@ -36,7 +36,7 @@
 					to="/review"
 					icon="review"
 					label="Review"
-					:notificationCount="notificationCounts.review"
+					:notificationCount="user?.notification_counts?.review ?? 0"
 				/>
 			</li>
 			<li>
@@ -57,15 +57,12 @@
 import { ref, reactive, onMounted, onBeforeUnmount, nextTick } from "vue";
 import { Tooltip, Popover } from "bootstrap";
 import NavigationLink from "../common/NavigationLink.vue";
+import { useUserStore } from "../../store/user";
 
 // Reactive state
 const isCollapsed = ref(false);
-const notificationCounts = reactive({
-	communication: 0,
-	review: 0,
-});
 
-const userData = reactive({});
+const user = useUserStore();
 
 // Bootstrap tooltip instances
 let tooltipInstances = [];
@@ -128,25 +125,6 @@ const setTooltipsForSidebar = () => {
 	});
 };
 
-// Fetch user state from API
-const updateUserState = async () => {
-	try {
-		const response = await fetch("/api/user/self");
-		const data = await response.json();
-
-		// Update reactive data
-		Object.keys(data).forEach((key) => {
-			if (key === "notification_counts") {
-				Object.assign(notificationCounts, data[key]);
-			} else {
-				userData[key] = data[key];
-			}
-		});
-	} catch (error) {
-		console.error("Failed to update user state:", error);
-	}
-};
-
 // Lifecycle hooks
 onMounted(async () => {
 	restoreLocalStorage();
@@ -155,18 +133,6 @@ onMounted(async () => {
 
 	initializeBootstrap();
 	setTooltipsForSidebar();
-
-	// Subscribe to SSE events (assuming SSEManager is globally available)
-	if (window.SSEManager) {
-		sseUnsubscribe = window.SSEManager.subscribe("*", (e) => {
-			if (e.type !== "heartbeat") {
-				updateUserState();
-			}
-		});
-	}
-
-	// Initial user state fetch
-	updateUserState();
 });
 
 onBeforeUnmount(() => {
diff --git a/ts/store/user.ts b/ts/store/user.ts
index 1bc2d883..86015ee3 100644
--- a/ts/store/user.ts
+++ b/ts/store/user.ts
@@ -1,5 +1,6 @@
 import { defineStore } from "pinia";
 import { ref, computed } from "vue";
+import { SSEManager } from "../SSEManager";
 
 // Define interfaces matching your Go structs
 interface URLsAPI {
@@ -44,6 +45,13 @@ export const useUserStore = defineStore("user", () => {
 	const urls = ref<URLs | null>(null);
 	const username = ref<string | null>(null);
 
+	// Subscription
+	SSEManager.subscribe("*", (e) => {
+		if (e.type !== "heartbeat") {
+			fetchUser();
+		}
+	});
+
 	// Actions
 	async function fetchUser() {
 		loading.value = true;