51 lines
1.6 KiB
TypeScript
51 lines
1.6 KiB
TypeScript
|
import type { SessionStatus } from "@opencode-ai/sdk/v2/client"
|
||
|
|
|
||
|
|
export const LIVE_STATUS_TTL_MS = 15_000
|
||
|
|
|
||
|
|
type RuntimeLiveStatus = {
|
||
|
|
runtimeKey: string
|
||
|
|
directory: string
|
||
|
|
sessionId: string
|
||
|
|
status: SessionStatus
|
||
|
|
expiresAt: number
|
||
|
|
}
|
||
|
|
|
||
|
|
const liveStatusByRuntime = new Map<string, RuntimeLiveStatus>()
|
||
|
|
|
||
|
|
const keyFor = (runtimeKey: string, directory: string) => `${runtimeKey}\n${directory}`
|
||
|
|
|
||
|
|
export function rememberRuntimeLiveStatus(params: {
|
||
|
|
runtimeKey: string
|
||
|
|
directory: string | null | undefined
|
||
|
|
sessionId: string | null | undefined
|
||
|
|
status: SessionStatus | null | undefined
|
||
|
|
}) {
|
||
|
|
if (!params.runtimeKey || !params.directory || !params.sessionId || !params.status) return
|
||
|
|
if (params.status.type === "idle") return
|
||
|
|
|
||
|
|
// Evict expired entries on write so keys that are never read again don't
|
||
|
|
// accumulate (reads are lazy and only prune their own key).
|
||
|
|
const now = Date.now()
|
||
|
|
for (const [key, entry] of liveStatusByRuntime) {
|
||
|
|
if (entry.expiresAt <= now) liveStatusByRuntime.delete(key)
|
||
|
|
}
|
||
|
|
|
||
|
|
liveStatusByRuntime.set(keyFor(params.runtimeKey, params.directory), {
|
||
|
|
runtimeKey: params.runtimeKey,
|
||
|
|
directory: params.directory,
|
||
|
|
sessionId: params.sessionId,
|
||
|
|
status: params.status,
|
||
|
|
expiresAt: Date.now() + LIVE_STATUS_TTL_MS,
|
||
|
|
})
|
||
|
|
}
|
||
|
|
|
||
|
|
export function getRuntimeLiveStatusSeed(runtimeKey: string, directory: string): RuntimeLiveStatus | null {
|
||
|
|
const entry = liveStatusByRuntime.get(keyFor(runtimeKey, directory))
|
||
|
|
if (!entry) return null
|
||
|
|
if (entry.expiresAt <= Date.now()) {
|
||
|
|
liveStatusByRuntime.delete(keyFor(runtimeKey, directory))
|
||
|
|
return null
|
||
|
|
}
|
||
|
|
return entry
|
||
|
|
}
|