Function useTaskPolling

frontend/src/hooks/query/use-task-polling.ts:22–78

()

 * Note: This hook does NOT fetch conversation data. It only handles task polling and navigation.
 */
export const useTaskPolling = () => {
  const { conversationId } = useConversationId();
  const navigate = useNavigate();

  // Check if this is a task ID (format: "task-{uuid}")
  const isTask = conversationId.startsWith("task-");
  const taskId = isTask ? conversationId.replace("task-", "") : null;

  // Poll the task if this is a task ID
  const taskQuery = useQuery({
    queryKey: ["start-task", taskId],
    queryFn: async () => {
      if (!taskId) return null;
      return V1ConversationService.getStartTask(taskId);
    },
    enabled: !!taskId,
    refetchInterval: (query) => {
      const task = query.state.data;
      if (!task) return false;

      // Stop polling if ready or error
      if (task.status === "READY" || task.status === "ERROR") {
        return false;
      }

      // Poll every 3 seconds while task is in progress
      return 3000;
    },
    retry: false,
  });

  // Navigate to conversation ID when task is ready
  useEffect(() => {
    const task = taskQuery.data;
    if (task?.status === "READY" && task.app_conversation_id) {
      // Replace the URL with the actual conversation ID
      navigate(`/conversations/${task.app_conversation_id}`, { replace: true });
    }
  }, [taskQuery.data, navigate]);

  return {
    isTask,
    taskId,
    conversationId: isTask ? null : conversationId,
    task: taskQuery.data,
    taskStatus: taskQuery.data?.status,
    taskDetail: taskQuery.data?.detail,
    taskError: taskQuery.error,
    isLoadingTask: taskQuery.isLoading,
    // Repository information from task request
    repositoryInfo: {
      selectedRepository: taskQuery.data?.request?.selected_repository,
      selectedBranch: taskQuery.data?.request?.selected_branch,
      gitProvider: taskQuery.data?.request?.git_provider,
    },
  };
};