Carlos Florêncio
55 lines
1.8 KiB
TypeScript
55 lines
1.8 KiB
TypeScript
"use client";
|
|
|
|
import { useCallback, useEffect, useRef, useState } from "react";
|
|
|
|
import { listAgentUpdateStatuses, type AgentUpdateJob, type AgentUpdateStatus } from "@/lib/api";
|
|
|
|
type StatusByAgent = Record<string, AgentUpdateStatus>;
|
|
|
|
function indexStatuses(statuses: AgentUpdateStatus[]): StatusByAgent {
|
|
return Object.fromEntries(statuses.map((status) => [status.agent_name, status]));
|
|
}
|
|
|
|
export function useAgentRuntimeUpdateStatuses(updateJobs: Record<string, AgentUpdateJob>) {
|
|
const [statusByAgent, setStatusByAgent] = useState<StatusByAgent>({});
|
|
const observedSuccessfulJobs = useRef(new Set<string>());
|
|
const pendingSuccessfulJobs = useRef(new Set<string>());
|
|
|
|
const refresh = useCallback(async (): Promise<boolean> => {
|
|
try {
|
|
const response = await listAgentUpdateStatuses({ cache: "no-store" });
|
|
setStatusByAgent(indexStatuses(response.statuses));
|
|
return true;
|
|
} catch {
|
|
// A status hint is advisory. Keep the last good map so a transient
|
|
// registry failure does not remove an already visible hint or disable
|
|
// the authoritative update control.
|
|
return false;
|
|
}
|
|
}, []);
|
|
|
|
useEffect(() => {
|
|
void refresh();
|
|
}, [refresh]);
|
|
|
|
useEffect(() => {
|
|
for (const job of Object.values(updateJobs)) {
|
|
if (job.status !== "succeeded" || observedSuccessfulJobs.current.has(job.job_id)) {
|
|
continue;
|
|
}
|
|
if (pendingSuccessfulJobs.current.has(job.job_id)) {
|
|
continue;
|
|
}
|
|
pendingSuccessfulJobs.current.add(job.job_id);
|
|
void refresh().then((succeeded) => {
|
|
pendingSuccessfulJobs.current.delete(job.job_id);
|
|
if (succeeded) {
|
|
observedSuccessfulJobs.current.add(job.job_id);
|
|
}
|
|
});
|
|
}
|
|
}, [refresh, updateJobs]);
|
|
|
|
return { refresh, statusByAgent };
|
|
}
|