fix: recover stale IM session bindings

IM adapters could keep in-process bridge state after desktop session deletion removed the persisted mapping or after the server no longer had the mapped session. The restore path now validates the stored mapping against bridge memory and the server session before reusing it, then clears stale transient state before creating a replacement session.

Constraint: Issue #574 is intermittent and tied to adapter-created sessions; live reporter Telegram credentials are not available locally.
Rejected: Keep separate recovery checks in each IM adapter | duplicates the same stale-state logic across Telegram, Feishu, WeChat, and DingTalk.
Confidence: high
Scope-risk: moderate
Directive: Do not use bridge.hasSession() as the sole restore check; verify adapter-sessions state and server session existence first.
Tested: bun run check:adapters (364 pass, 0 fail)
Tested: cd adapters && bunx tsc --noEmit
Tested: bun test src/server/__tests__/sessions.test.ts src/server/__tests__/websocket-handler.test.ts --timeout 30000 (110 pass, 0 fail)
Tested: git diff --check
Not-tested: Live Telegram bot/provider reproduction on the reporter machine
This commit is contained in:
程序员阿江(Relakkes) 2026-05-24 15:51:36 +08:00
parent acedb2b5b5
commit 49481123f5
10 changed files with 338 additions and 76 deletions

View File

@ -106,6 +106,20 @@ describe('AdapterHttpClient', () => {
expect(client.createSession('')).rejects.toThrow()
})
it('sessionExists returns false for deleted sessions', async () => {
globalThis.fetch = mock(() =>
Promise.resolve(new Response(JSON.stringify({ error: 'NOT_FOUND' }), {
status: 404,
headers: { 'Content-Type': 'application/json' },
}))
) as any
await expect(client.sessionExists('deleted-session')).resolves.toBe(false)
expect((globalThis.fetch as any).mock.calls[0][0]).toBe(
'http://127.0.0.1:3456/api/sessions/deleted-session',
)
})
it('getGitInfo calls GET /api/sessions/:id/git-info', async () => {
globalThis.fetch = mock(() =>
Promise.resolve(new Response(JSON.stringify({

View File

@ -0,0 +1,163 @@
import { describe, expect, it } from 'bun:test'
import { restoreStoredSessionBinding } from '../session-recovery.js'
import type { SessionEntry } from '../session-store.js'
function makeStore(initial: SessionEntry | null) {
let entry = initial
return {
get() {
return entry
},
delete() {
entry = null
},
current() {
return entry
},
}
}
function makeBridge(options?: { currentSessionId?: string | null; open?: boolean; hasSession?: boolean }) {
const calls: string[] = []
const bridge = {
calls,
connectSession(_chatId: string, sessionId: string) {
calls.push(`connect:${sessionId}`)
return true
},
getSessionId() {
return options?.currentSessionId ?? null
},
hasSession() {
return options?.hasSession ?? Boolean(options?.currentSessionId)
},
isSessionOpen(_chatId: string, sessionId?: string) {
return Boolean(options?.open && (!sessionId || sessionId === options?.currentSessionId))
},
onServerMessage() {
calls.push('handler')
},
resetSession() {
calls.push('reset')
},
waitForOpen() {
calls.push('wait')
return Promise.resolve(true)
},
}
return bridge
}
describe('restoreStoredSessionBinding', () => {
it('resets stale bridge memory when server-side delete removed the stored mapping', async () => {
const store = makeStore(null)
const bridge = makeBridge({ currentSessionId: 'deleted-session', hasSession: true })
let cleared = 0
const restored = await restoreStoredSessionBinding({
chatId: 'chat-1',
bridge,
sessionStore: store,
httpClient: { sessionExists: async () => true },
onServerMessage: () => {},
logPrefix: '[Test]',
clearTransientState: () => {
cleared += 1
},
})
expect(restored).toBeNull()
expect(bridge.calls).toEqual(['reset'])
expect(cleared).toBe(1)
})
it('drops a persisted mapping when the server no longer has that session', async () => {
const entry = { sessionId: 'deleted-session', workDir: '/repo', updatedAt: 1 }
const store = makeStore(entry)
const bridge = makeBridge()
let cleared = 0
const restored = await restoreStoredSessionBinding({
chatId: 'chat-1',
bridge,
sessionStore: store,
httpClient: { sessionExists: async () => false },
onServerMessage: () => {},
logPrefix: '[Test]',
clearTransientState: () => {
cleared += 1
},
})
expect(restored).toBeNull()
expect(store.current()).toBeNull()
expect(bridge.calls).toEqual([])
expect(cleared).toBe(1)
})
it('resets bridge memory when it points at a different session than the store', async () => {
const entry = { sessionId: 'stored-session', workDir: '/repo', updatedAt: 1 }
const store = makeStore(entry)
const bridge = makeBridge({ currentSessionId: 'stale-session', hasSession: true })
let cleared = 0
const restored = await restoreStoredSessionBinding({
chatId: 'chat-1',
bridge,
sessionStore: store,
httpClient: { sessionExists: async () => true },
onServerMessage: () => {},
logPrefix: '[Test]',
clearTransientState: () => {
cleared += 1
},
})
expect(restored).toEqual(entry)
expect(bridge.calls).toEqual(['reset', 'connect:stored-session', 'handler', 'wait'])
expect(cleared).toBe(1)
})
it('reuses an already-open bridge for the stored session', async () => {
const entry = { sessionId: 'live-session', workDir: '/repo', updatedAt: 1 }
const store = makeStore(entry)
const bridge = makeBridge({ currentSessionId: 'live-session', open: true })
let checked = 0
const restored = await restoreStoredSessionBinding({
chatId: 'chat-1',
bridge,
sessionStore: store,
httpClient: {
sessionExists: async () => {
checked += 1
return true
},
},
onServerMessage: () => {},
logPrefix: '[Test]',
})
expect(restored).toEqual(entry)
expect(checked).toBe(0)
expect(bridge.calls).toEqual([])
})
it('connects to an existing stored session after validating it still exists', async () => {
const entry = { sessionId: 'stored-session', workDir: '/repo', updatedAt: 1 }
const store = makeStore(entry)
const bridge = makeBridge()
const restored = await restoreStoredSessionBinding({
chatId: 'chat-1',
bridge,
sessionStore: store,
httpClient: { sessionExists: async () => true },
onServerMessage: () => {},
logPrefix: '[Test]',
})
expect(restored).toEqual(entry)
expect(bridge.calls).toEqual(['connect:stored-session', 'handler', 'wait'])
})
})

View File

@ -17,6 +17,8 @@ describe('WsBridge', () => {
const result = bridge.connectSession('chat-1', 'my-uuid-session-id')
expect(result).toBe(true)
expect(bridge.hasSession('chat-1')).toBe(true)
expect(bridge.getSessionId('chat-1')).toBe('my-uuid-session-id')
expect(bridge.isSessionOpen('chat-1', 'my-uuid-session-id')).toBe(false)
})
it('connectSession for different chatIds creates separate sessions', () => {

View File

@ -72,6 +72,23 @@ export class AdapterHttpClient {
}
}
async sessionExists(sessionId: string): Promise<boolean> {
const { controller, timer } = this.createTimeoutController()
try {
const res = await fetch(`${this.httpBaseUrl}/api/sessions/${encodeURIComponent(sessionId)}`, {
signal: controller.signal,
})
if (res.status === 404) return false
if (!res.ok) {
const err = await res.json().catch(() => ({ message: res.statusText }))
throw new Error(`Failed to check session: ${(err as any).message}`)
}
return true
} finally {
clearTimeout(timer)
}
}
async listRecentProjects(): Promise<RecentProject[]> {
const { controller, timer } = this.createTimeoutController()
try {

View File

@ -0,0 +1,83 @@
import type { AdapterHttpClient } from './http-client.js'
import type { SessionEntry, SessionStore } from './session-store.js'
import type { ServerMessage, WsBridge } from './ws-bridge.js'
type BridgeSessionOps = Pick<
WsBridge,
| 'connectSession'
| 'getSessionId'
| 'hasSession'
| 'isSessionOpen'
| 'onServerMessage'
| 'resetSession'
| 'waitForOpen'
>
type RestoreStoredSessionBindingOptions = {
chatId: string
bridge: BridgeSessionOps
sessionStore: Pick<SessionStore, 'delete' | 'get'>
httpClient: Pick<AdapterHttpClient, 'sessionExists'>
onServerMessage: (msg: ServerMessage) => void | Promise<void>
logPrefix: string
clearTransientState?: () => void
}
function resetStaleBridge(
chatId: string,
bridge: BridgeSessionOps,
clearTransientState?: () => void,
): void {
if (!bridge.hasSession(chatId)) return
bridge.resetSession(chatId)
clearTransientState?.()
}
export async function restoreStoredSessionBinding({
chatId,
bridge,
sessionStore,
httpClient,
onServerMessage,
logPrefix,
clearTransientState,
}: RestoreStoredSessionBindingOptions): Promise<SessionEntry | null> {
const stored = sessionStore.get(chatId)
if (!stored) {
resetStaleBridge(chatId, bridge, clearTransientState)
return null
}
const currentSessionId = bridge.getSessionId(chatId)
if (currentSessionId && currentSessionId !== stored.sessionId) {
resetStaleBridge(chatId, bridge, clearTransientState)
}
if (bridge.isSessionOpen(chatId, stored.sessionId)) {
return stored
}
let exists = true
try {
exists = await httpClient.sessionExists(stored.sessionId)
} catch (err) {
console.warn(
`${logPrefix} Failed to verify stored session ${stored.sessionId}: ${
err instanceof Error ? err.message : String(err)
}`,
)
}
if (!exists) {
sessionStore.delete(chatId)
const hadBridgeSession = bridge.hasSession(chatId)
resetStaleBridge(chatId, bridge, clearTransientState)
if (!hadBridgeSession) clearTransientState?.()
return null
}
bridge.connectSession(chatId, stored.sessionId)
bridge.onServerMessage(chatId, onServerMessage)
const opened = await bridge.waitForOpen(chatId)
return opened ? stored : null
}

View File

@ -113,6 +113,17 @@ export class WsBridge {
this.handlers.set(chatId, handler)
}
getSessionId(chatId: string): string | null {
return this.sessions.get(chatId)?.sessionId ?? null
}
isSessionOpen(chatId: string, sessionId?: string): boolean {
const session = this.sessions.get(chatId)
if (!session) return false
if (sessionId && session.sessionId !== sessionId) return false
return session.ws.readyState === WebSocket.OPEN
}
/** Reset session for a chatId (e.g. /new command). */
resetSession(chatId: string): void {
const session = this.sessions.get(chatId)

View File

@ -21,6 +21,7 @@ import {
} from '../common/permission.js'
import { SessionStore } from '../common/session-store.js'
import { AdapterHttpClient, type RecentProject } from '../common/http-client.js'
import { restoreStoredSessionBinding } from '../common/session-recovery.js'
import { isAllowedUser, tryPair } from '../common/pairing.js'
import { AttachmentStore } from '../common/attachment/attachment-store.js'
import { checkAttachmentLimit } from '../common/attachment/attachment-limits.js'
@ -222,17 +223,15 @@ function clearPendingPermissions(chatId: string): void {
}
async function ensureExistingSession(chatId: string): Promise<{ sessionId: string; workDir: string } | null> {
const stored = sessionStore.get(chatId)
if (!stored) return null
if (!bridge.hasSession(chatId)) {
bridge.connectSession(chatId, stored.sessionId)
bridge.onServerMessage(chatId, (msg) => handleServerMessage(chatId, msg))
const opened = await bridge.waitForOpen(chatId)
if (!opened) return null
}
return stored
return await restoreStoredSessionBinding({
chatId,
bridge,
sessionStore,
httpClient,
onServerMessage: (msg) => handleServerMessage(chatId, msg),
logPrefix: '[DingTalk]',
clearTransientState: () => clearTransientChatState(chatId),
})
}
async function buildStatusText(chatId: string): Promise<string> {
@ -287,14 +286,8 @@ async function buildStatusText(chatId: string): Promise<string> {
}
async function ensureSession(chatId: string): Promise<boolean> {
if (bridge.hasSession(chatId)) return true
const stored = sessionStore.get(chatId)
if (stored) {
bridge.connectSession(chatId, stored.sessionId)
bridge.onServerMessage(chatId, (msg) => handleServerMessage(chatId, msg))
return await bridge.waitForOpen(chatId)
}
const stored = await ensureExistingSession(chatId)
if (stored) return true
return await createSessionForChat(chatId, defaultWorkDir)
}

View File

@ -29,6 +29,7 @@ import {
} from '../common/permission.js'
import { SessionStore } from '../common/session-store.js'
import { AdapterHttpClient, type RecentProject } from '../common/http-client.js'
import { restoreStoredSessionBinding } from '../common/session-recovery.js'
import { isAllowedUser, tryPair } from '../common/pairing.js'
import { optimizeMarkdownForFeishu } from './markdown-style.js'
import { extractInboundPayload } from './extract-payload.js'
@ -218,17 +219,15 @@ function clearTransientChatState(chatId: string): void {
}
async function ensureExistingSession(chatId: string): Promise<{ sessionId: string; workDir: string } | null> {
const stored = sessionStore.get(chatId)
if (!stored) return null
if (!bridge.hasSession(chatId)) {
bridge.connectSession(chatId, stored.sessionId)
bridge.onServerMessage(chatId, (msg) => handleServerMessage(chatId, msg))
const opened = await bridge.waitForOpen(chatId)
if (!opened) return null
}
return stored
return await restoreStoredSessionBinding({
chatId,
bridge,
sessionStore,
httpClient,
onServerMessage: (msg) => handleServerMessage(chatId, msg),
logPrefix: '[Feishu]',
clearTransientState: () => clearTransientChatState(chatId),
})
}
async function buildStatusText(chatId: string): Promise<string> {
@ -646,14 +645,8 @@ function buildPermissionCard(
// ---------- session management ----------
async function ensureSession(chatId: string): Promise<boolean> {
if (bridge.hasSession(chatId)) return true
const stored = sessionStore.get(chatId)
if (stored) {
bridge.connectSession(chatId, stored.sessionId)
bridge.onServerMessage(chatId, (msg) => handleServerMessage(chatId, msg))
return await bridge.waitForOpen(chatId)
}
const stored = await ensureExistingSession(chatId)
if (stored) return true
const workDir = defaultWorkDir
if (workDir) {

View File

@ -33,6 +33,7 @@ import {
} from '../common/permission.js'
import { SessionStore } from '../common/session-store.js'
import { AdapterHttpClient } from '../common/http-client.js'
import { restoreStoredSessionBinding } from '../common/session-recovery.js'
import { isAllowedUser, tryPair } from '../common/pairing.js'
import { TelegramMediaService } from './media.js'
import { AttachmentStore } from '../common/attachment/attachment-store.js'
@ -150,17 +151,15 @@ async function handlePermissionDecision(chatId: string, decision: PermissionDeci
}
async function ensureExistingSession(chatId: string): Promise<{ sessionId: string; workDir: string } | null> {
const stored = sessionStore.get(chatId)
if (!stored) return null
if (!bridge.hasSession(chatId)) {
bridge.connectSession(chatId, stored.sessionId)
bridge.onServerMessage(chatId, (msg) => handleServerMessage(chatId, msg))
const opened = await bridge.waitForOpen(chatId)
if (!opened) return null
}
return stored
return await restoreStoredSessionBinding({
chatId,
bridge,
sessionStore,
httpClient,
onServerMessage: (msg) => handleServerMessage(chatId, msg),
logPrefix: '[Telegram]',
clearTransientState: () => clearTransientChatState(chatId),
})
}
async function buildStatusText(chatId: string): Promise<string> {
@ -283,14 +282,8 @@ async function flushToTelegram(chatId: string, newText: string, isComplete: bool
// ---------- session management ----------
async function ensureSession(chatId: string): Promise<boolean> {
if (bridge.hasSession(chatId)) return true
const stored = sessionStore.get(chatId)
if (stored) {
bridge.connectSession(chatId, stored.sessionId)
bridge.onServerMessage(chatId, (msg) => handleServerMessage(chatId, msg))
return await bridge.waitForOpen(chatId)
}
const stored = await ensureExistingSession(chatId)
if (stored) return true
const workDir = defaultWorkDir
if (workDir) {

View File

@ -17,6 +17,7 @@ import {
} from '../common/permission.js'
import { SessionStore } from '../common/session-store.js'
import { AdapterHttpClient } from '../common/http-client.js'
import { restoreStoredSessionBinding } from '../common/session-recovery.js'
import { isAllowedUser, tryPair } from '../common/pairing.js'
import { AttachmentStore } from '../common/attachment/attachment-store.js'
import { checkAttachmentLimit } from '../common/attachment/attachment-limits.js'
@ -203,17 +204,15 @@ function enqueueWechat(chatId: string, task: () => Promise<void>): void {
}
async function ensureExistingSession(chatId: string): Promise<{ sessionId: string; workDir: string } | null> {
const stored = sessionStore.get(chatId)
if (!stored) return null
if (!bridge.hasSession(chatId)) {
bridge.connectSession(chatId, stored.sessionId)
bridge.onServerMessage(chatId, (msg) => handleServerMessage(chatId, msg))
const opened = await bridge.waitForOpen(chatId)
if (!opened) return null
}
return stored
return await restoreStoredSessionBinding({
chatId,
bridge,
sessionStore,
httpClient,
onServerMessage: (msg) => handleServerMessage(chatId, msg),
logPrefix: '[WeChat]',
clearTransientState: () => clearTransientChatState(chatId),
})
}
async function buildStatusText(chatId: string): Promise<string> {
@ -268,14 +267,8 @@ async function buildStatusText(chatId: string): Promise<string> {
}
async function ensureSession(chatId: string): Promise<boolean> {
if (bridge.hasSession(chatId)) return true
const stored = sessionStore.get(chatId)
if (stored) {
bridge.connectSession(chatId, stored.sessionId)
bridge.onServerMessage(chatId, (msg) => handleServerMessage(chatId, msg))
return await bridge.waitForOpen(chatId)
}
const stored = await ensureExistingSession(chatId)
if (stored) return true
const workDir = defaultWorkDir
if (workDir) return await createSessionForChat(chatId, workDir)