Carlos Florêncio
98 lines
2.9 KiB
TypeScript
98 lines
2.9 KiB
TypeScript
"use client";
|
|
|
|
import { useEffect, useCallback, useRef } from "react";
|
|
import { listTaskPRs } from "@/lib/api/domains/github-api";
|
|
import { getWebSocketClient } from "@/lib/ws/connection";
|
|
import { useAppStore } from "@/components/state-provider";
|
|
import type { TaskPR } from "@/lib/types/github";
|
|
|
|
/** Fetch and cache PR associations for a batch of task IDs. */
|
|
export function useTaskPRs(taskIds: string[]) {
|
|
const byTaskId = useAppStore((state) => state.taskPRs.byTaskId);
|
|
const loading = useAppStore((state) => state.taskPRs.loading);
|
|
const setTaskPRs = useAppStore((state) => state.setTaskPRs);
|
|
const setTaskPRsLoading = useAppStore((state) => state.setTaskPRsLoading);
|
|
|
|
useEffect(() => {
|
|
if (taskIds.length === 0 || loading) return;
|
|
setTaskPRsLoading(true);
|
|
listTaskPRs(taskIds, { cache: "no-store" })
|
|
.then((response) => {
|
|
setTaskPRs(response?.task_prs ?? {});
|
|
})
|
|
.catch(() => {
|
|
// Keep existing data on error
|
|
})
|
|
.finally(() => {
|
|
setTaskPRsLoading(false);
|
|
});
|
|
// eslint-disable-next-line react-hooks/exhaustive-deps
|
|
}, [taskIds.join(",")]);
|
|
|
|
return { byTaskId, loading };
|
|
}
|
|
|
|
const SYNC_RETRY_DELAY = 5_000; // 5 seconds
|
|
const SYNC_MAX_RETRIES = 6; // Up to 30 seconds of retries
|
|
|
|
/** Fetch a single task's PR association, with on-demand sync via WS. */
|
|
export function useTaskPR(taskId: string | null) {
|
|
const pr = useAppStore((state) => (taskId ? (state.taskPRs.byTaskId[taskId] ?? null) : null));
|
|
const setTaskPR = useAppStore((state) => state.setTaskPR);
|
|
const retryRef = useRef(0);
|
|
|
|
const refresh = useCallback(() => {
|
|
if (!taskId) return;
|
|
const client = getWebSocketClient();
|
|
if (!client) return;
|
|
|
|
client
|
|
.request<TaskPR | null>("github.task_pr.sync", { task_id: taskId })
|
|
.then((result) => {
|
|
if (result?.task_id) {
|
|
setTaskPR(taskId, result);
|
|
retryRef.current = 0;
|
|
}
|
|
})
|
|
.catch(() => {
|
|
// Ignore - sync may fail if no watch exists
|
|
});
|
|
}, [taskId, setTaskPR]);
|
|
|
|
// Reset retry count when taskId changes
|
|
useEffect(() => {
|
|
retryRef.current = 0;
|
|
}, [taskId]);
|
|
|
|
useEffect(() => {
|
|
if (!taskId) return;
|
|
|
|
// Always sync once when the task becomes active (freshness)
|
|
refresh();
|
|
|
|
// If no PR in store yet, retry periodically until found
|
|
if (pr) return;
|
|
|
|
const interval = setInterval(() => {
|
|
if (retryRef.current >= SYNC_MAX_RETRIES) {
|
|
clearInterval(interval);
|
|
return;
|
|
}
|
|
retryRef.current++;
|
|
refresh();
|
|
}, SYNC_RETRY_DELAY);
|
|
|
|
return () => clearInterval(interval);
|
|
}, [taskId, pr, refresh]);
|
|
|
|
return { pr, refresh } as { pr: TaskPR | null; refresh: () => void };
|
|
}
|
|
|
|
/** Read the active task's PR from the store (no fetching). */
|
|
export function useActiveTaskPR(): TaskPR | null {
|
|
return useAppStore((s) => {
|
|
const taskId = s.tasks.activeTaskId;
|
|
return taskId ? (s.taskPRs.byTaskId[taskId] ?? null) : null;
|
|
});
|
|
}
|