YeonGyu-Kim
e395eadfa4
🤖 Generated with assistance of [OhMyOpenCode](https://github.com/code-yeongyu/oh-my-opencode)
34 lines
824 B
TypeScript
34 lines
824 B
TypeScript
type SessionState = {
|
|
isRecovering?: boolean
|
|
}
|
|
|
|
export function createLoopSessionRecovery(options?: { recoveryWindowMs?: number }) {
|
|
const recoveryWindowMs = options?.recoveryWindowMs ?? 5000
|
|
const sessions = new Map<string, SessionState>()
|
|
|
|
function getSessionState(sessionID: string): SessionState {
|
|
let state = sessions.get(sessionID)
|
|
if (!state) {
|
|
state = {}
|
|
sessions.set(sessionID, state)
|
|
}
|
|
return state
|
|
}
|
|
|
|
return {
|
|
isRecovering(sessionID: string): boolean {
|
|
return getSessionState(sessionID).isRecovering === true
|
|
},
|
|
markRecovering(sessionID: string): void {
|
|
const state = getSessionState(sessionID)
|
|
state.isRecovering = true
|
|
setTimeout(() => {
|
|
state.isRecovering = false
|
|
}, recoveryWindowMs)
|
|
},
|
|
clear(sessionID: string): void {
|
|
sessions.delete(sessionID)
|
|
},
|
|
}
|
|
}
|