feat: update session management functions and improve partition handling

apps/desktop-client/src/main/account-health.ts

@@ -19,7 +19,7 @@ import {
   type PlatformFailureInput,
 } from './platform-auth-adapters'
 import { emitRuntimeInvalidated } from './runtime-events'
-import { getPersistedPartition, getSessionHandle } from './session-registry'
+import { createSessionHandleForPartition, resolveStoredSessionPartition } from './session-registry'
 
 const PROBE_TICK_MS = 1_000
 const STALE_AFTER_MS = 45 * 60_000
@@ -117,15 +117,16 @@ function persistedHealthPath(): string {
 }
 
 function resolvePartition(accountId: string): string {
-  return (
-    getSessionHandle(accountId)?.partition ??
-    getPersistedPartition(accountId) ??
-    `persist:acc-${accountId}`
-  )
+  return resolveStoredSessionPartition(accountId) ?? `persist:acc-${accountId}`
 }
 
 function resolveKnownPartition(accountId: string): string | null {
-  return getSessionHandle(accountId)?.partition ?? getPersistedPartition(accountId) ?? null
+  const partition = resolveStoredSessionPartition(accountId)
+  if (!partition) {
+    return null
+  }
+  createSessionHandleForPartition(accountId, partition)
+  return partition
 }
 
 function nextInitialProbeAt(now = Date.now()): number {