José Almeida
67 lines
1.9 KiB
TypeScript
67 lines
1.9 KiB
TypeScript
"use client";
|
|
|
|
import { useCallback, useEffect, useMemo, useRef, useState } from "react";
|
|
import { ApiError } from "@/lib/api/client";
|
|
import {
|
|
fetchTaskEnvironmentLive,
|
|
type ContainerLiveStatus,
|
|
type TaskEnvironment,
|
|
} from "@/lib/api/domains/task-environment-api";
|
|
import {
|
|
resolveExecutorEnvironmentStatus,
|
|
type ExecutorEnvironmentStatus,
|
|
} from "@/components/task/executor-environment-status";
|
|
|
|
const POLL_INTERVAL_MS = 3000;
|
|
|
|
export function isExecutorEnvironmentUnavailable(
|
|
status: ExecutorEnvironmentStatus | null,
|
|
): boolean {
|
|
if (!status) return false;
|
|
if (status.tone === "running" || status.tone === "neutral") return false;
|
|
return status.label !== "starting";
|
|
}
|
|
|
|
export function useExecutorEnvironmentAvailability(taskId: string | null, enabled: boolean) {
|
|
const [env, setEnv] = useState<TaskEnvironment | null>(null);
|
|
const [container, setContainer] = useState<ContainerLiveStatus | null>(null);
|
|
const inFlight = useRef(false);
|
|
|
|
useEffect(() => {
|
|
setEnv(null);
|
|
setContainer(null);
|
|
}, [taskId]);
|
|
|
|
const load = useCallback(async () => {
|
|
if (!taskId || inFlight.current) return;
|
|
inFlight.current = true;
|
|
try {
|
|
const data = await fetchTaskEnvironmentLive(taskId);
|
|
setEnv(data.environment);
|
|
setContainer(data.container ?? null);
|
|
} catch (err) {
|
|
if (err instanceof ApiError && err.status === 404) {
|
|
setEnv(null);
|
|
setContainer(null);
|
|
}
|
|
} finally {
|
|
inFlight.current = false;
|
|
}
|
|
}, [taskId]);
|
|
|
|
useEffect(() => {
|
|
if (!enabled || !taskId) return;
|
|
void load();
|
|
const interval = window.setInterval(() => void load(), POLL_INTERVAL_MS);
|
|
return () => window.clearInterval(interval);
|
|
}, [enabled, taskId, load]);
|
|
|
|
const status = useMemo(
|
|
() => (env ? resolveExecutorEnvironmentStatus(env, container) : null),
|
|
[env, container],
|
|
);
|
|
const unavailable = isExecutorEnvironmentUnavailable(status);
|
|
|
|
return { status, unavailable };
|
|
}
|