Carlos Florêncio
81 lines
2.1 KiB
TypeScript
81 lines
2.1 KiB
TypeScript
"use client";
|
|
|
|
import type { FileInfo } from "@/lib/state/store";
|
|
import { getWebSocketClient } from "@/lib/ws/connection";
|
|
|
|
type SessionStatusResponse = {
|
|
is_agent_running: boolean;
|
|
};
|
|
|
|
export type CommitDiffResponse = {
|
|
success: boolean;
|
|
files?: Record<string, FileInfo>;
|
|
ready?: boolean;
|
|
reason?: string;
|
|
retry_after_ms?: number;
|
|
};
|
|
|
|
const notReadyProbeCooldownMs = 1000;
|
|
const nextProbeBySession = new Map<string, number>();
|
|
|
|
function notReadyResult(retryAfterMs = 500): CommitDiffResponse {
|
|
return {
|
|
success: false,
|
|
ready: false,
|
|
reason: "agent_starting",
|
|
retry_after_ms: retryAfterMs,
|
|
};
|
|
}
|
|
|
|
export async function requestCommitDiff(params: {
|
|
sessionId: string;
|
|
taskId: string | null;
|
|
commitSha: string;
|
|
agentctlReady: boolean;
|
|
}): Promise<CommitDiffResponse | null> {
|
|
const { sessionId, taskId, commitSha, agentctlReady } = params;
|
|
const ws = getWebSocketClient();
|
|
if (!ws) return null;
|
|
|
|
if (!agentctlReady) {
|
|
const now = Date.now();
|
|
const nextProbeAt = nextProbeBySession.get(sessionId) ?? 0;
|
|
if (now < nextProbeAt) {
|
|
return notReadyResult();
|
|
}
|
|
if (!taskId) {
|
|
nextProbeBySession.set(sessionId, now + notReadyProbeCooldownMs);
|
|
return notReadyResult();
|
|
}
|
|
try {
|
|
const status = await ws.request<SessionStatusResponse>(
|
|
"task.session.status",
|
|
{ task_id: taskId, session_id: sessionId },
|
|
5000,
|
|
);
|
|
if (!status?.is_agent_running) {
|
|
nextProbeBySession.set(sessionId, now + notReadyProbeCooldownMs);
|
|
return notReadyResult();
|
|
}
|
|
} catch {
|
|
nextProbeBySession.set(sessionId, now + notReadyProbeCooldownMs);
|
|
return notReadyResult();
|
|
}
|
|
}
|
|
|
|
const response = await ws.request<CommitDiffResponse>(
|
|
"session.commit_diff",
|
|
{ session_id: sessionId, commit_sha: commitSha },
|
|
10000,
|
|
);
|
|
if (response?.success) {
|
|
nextProbeBySession.delete(sessionId);
|
|
return response;
|
|
}
|
|
if (response?.ready === false) {
|
|
const retryAfter = response.retry_after_ms ?? notReadyProbeCooldownMs;
|
|
nextProbeBySession.set(sessionId, Date.now() + retryAfter);
|
|
}
|
|
return response;
|
|
}
|