Jyotisha/frontend/src/hooks/use-rectification-live-activity.ts

"use client";

import { useCallback, useEffect, useRef, type Dispatch, type SetStateAction } from "react";
import { rectificationLiveProgressLabel } from "@/lib/rectification-activity-labels";
import type { RenderMessage } from "@/components/rectification-message-entry";

/**
 * The live working row's label clock: when the run and the current step
 * started, which tool is running, and the 4-second tick that refreshes the
 * elapsed-time label while a turn is busy. Moved out of
 * RectificationAgenticChat unchanged (TASK-rectification-code-split-20260926).
 */
export function useRectificationLiveActivity({
  busy,
  setMessages,
}: Readonly<{
  busy: boolean;
  setMessages: Dispatch<SetStateAction<RenderMessage[]>>;
}>) {
  const runStartedAtRef = useRef(0);
  const stepStartedAtRef = useRef(0);
  const liveToolRef = useRef<string | null>(null);
  const liveBaseLabelRef = useRef("正在处理…");

  const beginLiveRun = useCallback((baseLabel: string) => {
    const now = Date.now();
    runStartedAtRef.current = now;
    stepStartedAtRef.current = now;
    liveToolRef.current = null;
    liveBaseLabelRef.current = baseLabel;
  }, []);

  const rememberLiveActivity = useCallback((baseLabel: string, tool: string | null = liveToolRef.current) => {
    if (liveBaseLabelRef.current !== baseLabel) {
      liveBaseLabelRef.current = baseLabel;
      stepStartedAtRef.current = Date.now();
    }
    liveToolRef.current = tool;
    return rectificationLiveProgressLabel({
      tool,
      baseLabel,
      stepStartedAt: stepStartedAtRef.current,
      runStartedAt: runStartedAtRef.current,
    });
  }, []);

  useEffect(() => {
    if (!busy) return;
    const id = window.setInterval(() => {
      const label = rectificationLiveProgressLabel({
        tool: liveToolRef.current,
        baseLabel: liveBaseLabelRef.current,
        stepStartedAt: stepStartedAtRef.current,
        runStartedAt: runStartedAtRef.current,
      });
      setMessages((current) => current.map((message) => {
        if (message.state !== "thinking" && message.state !== "streaming") return message;
        if (!message.activity || message.activity.label === label) return message;
        return { ...message, activity: { ...message.activity, label } };
      }));
    }, 4000);
    return () => window.clearInterval(id);
  }, [busy]);

  return { beginLiveRun, rememberLiveActivity };
}