Fix active session replay state

frontend/app/app/assistant.tsx

@@ -19,6 +19,10 @@ import {
 	SidebarProvider,
 	SidebarTrigger,
 } from "@/components/ui/sidebar";
+import {
+	readActiveSessionId,
+	writeActiveSessionId,
+} from "@/lib/claw-active-session";
 import { ClawSessionReplayProvider } from "@/lib/claw-session-replay";
 import { ClawAccountGate, useClawAccount } from "./claw-account-gate";
 
@@ -31,21 +35,13 @@ export const Assistant = () => {
 				prepareSendMessagesRequest: async (options) => {
 					const body = options.body as Record<string, unknown>;
 					const messages = options.messages as UIMessage[];
-					const selectedSessionId =
-						typeof window !== "undefined"
-							? window.localStorage.getItem("claw.activeSessionId")
-							: null;
+					const selectedSessionId = readActiveSessionId();
 					const lastSessionId = getLastSessionId(messages);
 					const selectedResumeSessionId =
 						messages.length > 1 ? selectedSessionId : null;
 					const outgoingSessionId =
 						lastSessionId ?? selectedResumeSessionId ?? options.id;
-					if (typeof window !== "undefined" && outgoingSessionId) {
-						window.localStorage.setItem(
-							"claw.activeSessionId",
-							outgoingSessionId,
-						);
-					}
+					writeActiveSessionId(outgoingSessionId);
 
 					return {
 						body: {
@@ -65,7 +61,7 @@ export const Assistant = () => {
 	});
 	const replaySession = useCallback(
 		(sessionId: string, repository: ExportedMessageRepository) => {
-			window.localStorage.setItem("claw.activeSessionId", sessionId);
+			writeActiveSessionId(sessionId);
 			runtime.thread.import(repository);
 		},
 		[runtime],