fix: align session status parsing and vscode reconnect reconcile (#1125)
* fix: align session status parsing and vscode reconnect reconcile * fix vscode session status fallback and reconcile cleanup safety --------- Co-authored-by: vhqtvn <8930337+vhqtvn@users.noreply.github.com> Co-authored-by: Bohdan Triapitsyn <artmore@protonmail.com>
This commit is contained in:
committed by
GitHub
co-authored by
vhqtvn
Bohdan Triapitsyn
parent
3a98987f71
commit
82b36e5080
@@ -16,6 +16,38 @@ const SESSION_COOLDOWN_DURATION_MS = 2000;
|
|||||||
let globalEventWatcherAbortController: AbortController | null = null;
|
let globalEventWatcherAbortController: AbortController | null = null;
|
||||||
let chatViewProvider: { postMessage: (message: unknown) => void } | null = null;
|
let chatViewProvider: { postMessage: (message: unknown) => void } | null = null;
|
||||||
|
|
||||||
|
const reconcileSessionActivityFromStatus = async (manager: OpenCodeManager): Promise<void> => {
|
||||||
|
const baseUrl = manager.getApiUrl();
|
||||||
|
if (!baseUrl) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
const url = new URL('/session/status', baseUrl);
|
||||||
|
const response = await fetch(url.toString(), {
|
||||||
|
headers: manager.getOpenCodeAuthHeaders(),
|
||||||
|
});
|
||||||
|
|
||||||
|
if (!response.ok) {
|
||||||
|
throw new Error(`session status fetch failed (${response.status})`);
|
||||||
|
}
|
||||||
|
|
||||||
|
const statuses = await response.json() as Record<string, { type?: string }>;
|
||||||
|
const knownSessionIds = new Set(Object.keys(statuses || {}));
|
||||||
|
|
||||||
|
for (const [sessionId, data] of Object.entries(statuses || {})) {
|
||||||
|
const type = typeof data?.type === 'string' ? data.type : 'idle';
|
||||||
|
const phase: ActivityPhase = type === 'busy' || type === 'retry' ? 'busy' : 'idle';
|
||||||
|
setSessionActivityPhase(sessionId, phase);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Drop stale in-memory activity entries not present in authoritative status.
|
||||||
|
for (const sessionId of Array.from(sessionActivityPhases.keys())) {
|
||||||
|
if (!knownSessionIds.has(sessionId)) {
|
||||||
|
setSessionActivityPhase(sessionId, 'idle');
|
||||||
|
}
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
const setSessionActivityPhase = (sessionId: string, phase: ActivityPhase): void => {
|
const setSessionActivityPhase = (sessionId: string, phase: ActivityPhase): void => {
|
||||||
if (!sessionId) return;
|
if (!sessionId) return;
|
||||||
|
|
||||||
@@ -82,8 +114,9 @@ const deriveSessionActivity = (payload: Record<string, unknown>): SessionActivit
|
|||||||
|
|
||||||
if (type === 'session.status') {
|
if (type === 'session.status') {
|
||||||
const status = properties?.status as Record<string, unknown> | undefined;
|
const status = properties?.status as Record<string, unknown> | undefined;
|
||||||
|
const info = properties?.info as Record<string, unknown> | undefined;
|
||||||
const sessionId = (properties?.sessionID ?? properties?.sessionId) as string;
|
const sessionId = (properties?.sessionID ?? properties?.sessionId) as string;
|
||||||
const statusType = status?.type as string;
|
const statusType = (status?.type ?? info?.type) as string;
|
||||||
|
|
||||||
if (typeof sessionId === 'string' && sessionId.length > 0 && typeof statusType === 'string') {
|
if (typeof sessionId === 'string' && sessionId.length > 0 && typeof statusType === 'string') {
|
||||||
const phase = statusType === 'busy' || statusType === 'retry' ? 'busy' : 'idle';
|
const phase = statusType === 'busy' || statusType === 'retry' ? 'busy' : 'idle';
|
||||||
@@ -166,6 +199,14 @@ export const startGlobalEventWatcher = async (
|
|||||||
baseUrl,
|
baseUrl,
|
||||||
headers: manager.getOpenCodeAuthHeaders(),
|
headers: manager.getOpenCodeAuthHeaders(),
|
||||||
});
|
});
|
||||||
|
try {
|
||||||
|
await reconcileSessionActivityFromStatus(manager);
|
||||||
|
} catch (error) {
|
||||||
|
console.warn(
|
||||||
|
'[VSCode:Activity] session status reconcile failed',
|
||||||
|
error instanceof Error ? error.message : error,
|
||||||
|
);
|
||||||
|
}
|
||||||
const result = await client.global.event({
|
const result = await client.global.event({
|
||||||
signal,
|
signal,
|
||||||
sseMaxRetryAttempts: 0,
|
sseMaxRetryAttempts: 0,
|
||||||
|
|||||||
@@ -716,9 +716,12 @@ const processForwardedEventPayload = (payload, emitSyntheticEvent) => {
|
|||||||
}
|
}
|
||||||
|
|
||||||
const properties = payload.properties && typeof payload.properties === 'object' ? payload.properties : {};
|
const properties = payload.properties && typeof payload.properties === 'object' ? payload.properties : {};
|
||||||
|
const statusInfo = properties.status && typeof properties.status === 'object' ? properties.status : {};
|
||||||
const info = properties.info && typeof properties.info === 'object' ? properties.info : {};
|
const info = properties.info && typeof properties.info === 'object' ? properties.info : {};
|
||||||
const sessionId = typeof properties.sessionID === 'string' ? properties.sessionID.trim() : '';
|
const sessionId = typeof properties.sessionID === 'string' ? properties.sessionID.trim() : '';
|
||||||
const status = typeof info.type === 'string' ? info.type.trim() : '';
|
const status = typeof statusInfo.type === 'string'
|
||||||
|
? statusInfo.type.trim()
|
||||||
|
: (typeof info.type === 'string' ? info.type.trim() : '');
|
||||||
|
|
||||||
if (!sessionId || !status) {
|
if (!sessionId || !status) {
|
||||||
return;
|
return;
|
||||||
@@ -731,9 +734,15 @@ const processForwardedEventPayload = (payload, emitSyntheticEvent) => {
|
|||||||
status,
|
status,
|
||||||
timestamp: Date.now(),
|
timestamp: Date.now(),
|
||||||
metadata: {
|
metadata: {
|
||||||
attempt: typeof info.attempt === 'number' ? info.attempt : undefined,
|
attempt: typeof statusInfo.attempt === 'number'
|
||||||
message: typeof info.message === 'string' ? info.message : undefined,
|
? statusInfo.attempt
|
||||||
next: typeof info.next === 'number' ? info.next : undefined,
|
: (typeof info.attempt === 'number' ? info.attempt : undefined),
|
||||||
|
message: typeof statusInfo.message === 'string'
|
||||||
|
? statusInfo.message
|
||||||
|
: (typeof info.message === 'string' ? info.message : undefined),
|
||||||
|
next: typeof statusInfo.next === 'number'
|
||||||
|
? statusInfo.next
|
||||||
|
: (typeof info.next === 'number' ? info.next : undefined),
|
||||||
},
|
},
|
||||||
needsAttention: false,
|
needsAttention: false,
|
||||||
},
|
},
|
||||||
|
|||||||
@@ -9,9 +9,13 @@ const extractSessionStatusUpdate = (payload) => {
|
|||||||
}
|
}
|
||||||
|
|
||||||
const properties = payload.properties && typeof payload.properties === 'object' ? payload.properties : {};
|
const properties = payload.properties && typeof payload.properties === 'object' ? payload.properties : {};
|
||||||
|
const status = properties.status && typeof properties.status === 'object' ? properties.status : {};
|
||||||
const info = properties.info && typeof properties.info === 'object' ? properties.info : {};
|
const info = properties.info && typeof properties.info === 'object' ? properties.info : {};
|
||||||
const sessionId = typeof properties.sessionID === 'string' ? properties.sessionID.trim() : '';
|
const sessionId = typeof properties.sessionID === 'string' ? properties.sessionID.trim() : '';
|
||||||
const type = typeof info.type === 'string' ? info.type.trim() : '';
|
// Canonical OpenCode schema uses properties.status.type. Keep legacy info.type fallback for compatibility.
|
||||||
|
const type = typeof status.type === 'string'
|
||||||
|
? status.type.trim()
|
||||||
|
: (typeof info.type === 'string' ? info.type.trim() : '');
|
||||||
|
|
||||||
if (!sessionId || !type) {
|
if (!sessionId || !type) {
|
||||||
return null;
|
return null;
|
||||||
@@ -21,9 +25,15 @@ const extractSessionStatusUpdate = (payload) => {
|
|||||||
sessionId,
|
sessionId,
|
||||||
type,
|
type,
|
||||||
eventId: typeof payload.id === 'string' ? payload.id : '',
|
eventId: typeof payload.id === 'string' ? payload.id : '',
|
||||||
attempt: typeof info.attempt === 'number' ? info.attempt : undefined,
|
attempt: typeof status.attempt === 'number'
|
||||||
message: typeof info.message === 'string' ? info.message : undefined,
|
? status.attempt
|
||||||
next: typeof info.next === 'number' ? info.next : undefined,
|
: (typeof info.attempt === 'number' ? info.attempt : undefined),
|
||||||
|
message: typeof status.message === 'string'
|
||||||
|
? status.message
|
||||||
|
: (typeof info.message === 'string' ? info.message : undefined),
|
||||||
|
next: typeof status.next === 'number'
|
||||||
|
? status.next
|
||||||
|
: (typeof info.next === 'number' ? info.next : undefined),
|
||||||
};
|
};
|
||||||
};
|
};
|
||||||
|
|
||||||
|
|||||||
@@ -29,7 +29,7 @@ describe('session runtime', () => {
|
|||||||
type: 'session.status',
|
type: 'session.status',
|
||||||
properties: {
|
properties: {
|
||||||
sessionID: 'session-1',
|
sessionID: 'session-1',
|
||||||
info: {
|
status: {
|
||||||
type: 'busy',
|
type: 'busy',
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
@@ -39,7 +39,7 @@ describe('session runtime', () => {
|
|||||||
type: 'session.status',
|
type: 'session.status',
|
||||||
properties: {
|
properties: {
|
||||||
sessionID: 'session-1',
|
sessionID: 'session-1',
|
||||||
info: {
|
status: {
|
||||||
type: 'idle',
|
type: 'idle',
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
@@ -65,4 +65,36 @@ describe('session runtime', () => {
|
|||||||
},
|
},
|
||||||
});
|
});
|
||||||
});
|
});
|
||||||
|
|
||||||
|
it('accepts legacy session.status info.type payloads', () => {
|
||||||
|
const events = [];
|
||||||
|
const runtime = createSessionRuntime({
|
||||||
|
writeSseEvent() {
|
||||||
|
throw new Error('SSE fallback should not be used when broadcastEvent is provided');
|
||||||
|
},
|
||||||
|
getNotificationClients: () => new Set(),
|
||||||
|
broadcastEvent: (payload) => {
|
||||||
|
events.push(payload);
|
||||||
|
},
|
||||||
|
});
|
||||||
|
runtimes.push(runtime);
|
||||||
|
|
||||||
|
runtime.processOpenCodeSsePayload({
|
||||||
|
type: 'session.status',
|
||||||
|
properties: {
|
||||||
|
sessionID: 'legacy-session-1',
|
||||||
|
info: {
|
||||||
|
type: 'busy',
|
||||||
|
},
|
||||||
|
},
|
||||||
|
});
|
||||||
|
|
||||||
|
expect(events).toContainEqual({
|
||||||
|
type: 'openchamber:session-status',
|
||||||
|
properties: expect.objectContaining({
|
||||||
|
sessionId: 'legacy-session-1',
|
||||||
|
status: 'busy',
|
||||||
|
}),
|
||||||
|
});
|
||||||
|
});
|
||||||
});
|
});
|
||||||
|
|||||||
Reference in New Issue
Block a user