Function useAdminTaskStream

src/ui/src/lib/hooks/useAdminTaskStream.ts:35–192

(taskId: string | null | undefined)

}

export function useAdminTaskStream(taskId: string | null | undefined) {
  const [task, setTask] = useState<AdminTask | null>(null)
  const [events, setEvents] = useState<AdminTaskEvent[]>([])
  const [connectionState, setConnectionState] = useState<'idle' | 'connecting' | 'connected' | 'reconnecting' | 'error'>(
    taskId ? 'connecting' : 'idle'
  )
  const [error, setError] = useState<string | null>(null)
  const abortRef = useRef<AbortController | null>(null)
  const reconnectRef = useRef<number | null>(null)
  const lastEventIdRef = useRef<string | null>(null)
  const latestTaskRef = useRef<AdminTask | null>(null)

  useEffect(() => {
    latestTaskRef.current = task
  }, [task])

  useEffect(() => {
    return () => {
      abortRef.current?.abort()
      if (reconnectRef.current != null) {
        window.clearTimeout(reconnectRef.current)
      }
    }
  }, [])

  useEffect(() => {
    if (!taskId) {
      abortRef.current?.abort()
      if (reconnectRef.current != null) {
        window.clearTimeout(reconnectRef.current)
      }
      setTask(null)
      setEvents([])
      setConnectionState('idle')
      setError(null)
      lastEventIdRef.current = null
      return
    }

    let cancelled = false

    const fetchSnapshot = async (signal: AbortSignal) => {
      const response = await fetch(`/api/system/tasks/${encodeURIComponent(taskId)}`, {
        method: 'GET',
        headers: {
          Accept: 'application/json',
          ...authHeaders(),
        },
        signal,
      })
      if (!response.ok) {
        throw new Error(`HTTP ${response.status}`)
      }
      const payload = (await response.json()) as { task?: AdminTask | null }
      if (payload?.task && typeof payload.task === 'object') {
        setTask(payload.task)
      }
    }