@cueloop/client 0.1.0-alpha.96 → 0.1.0-alpha.98

package/src/thread/use-thread-agent.ts

@@ -0,0 +1,101 @@
+import { useEffect, useRef, useState } from "react";
+import { DaemonClient } from "@cueloop/daemon/client";
+import type { ThreadAgentState } from "@cueloop/schema";
+
+/** Agent transport is injectable for frame tests; production uses the daemon socket. */
+export type ThreadAgentClient = Pick<
+  DaemonClient,
+  "agentGet" | "agentPrompt" | "agentCancel" | "agentComment" | "agentPermission"
+> &
+  Partial<Pick<DaemonClient, "agentConfigure">>;
+
+/** Subscribe to durable agent state; unmount closes only this client connection. */
+export function useThreadAgent(id: string, home?: string, injected?: ThreadAgentClient) {
+  const [client, setClient] = useState<ThreadAgentClient | null>(injected ?? null);
+  const [state, setState] = useState<ThreadAgentState>({
+    threadId: id,
+    phase: { kind: "idle" },
+    messages: [],
+    tools: [],
+    comments: [],
+  });
+  const revision = useRef(0);
+  const [error, setError] = useState("");
+
+  useEffect(() => {
+    let cancelled = false;
+    let connection: DaemonClient | undefined;
+    let unsubscribe: (() => void) | undefined;
+    revision.current++;
+    const refresh = async (api: ThreadAgentClient) => {
+      const requested = ++revision.current;
+      const result = await api.agentGet(id);
+
+      if (!cancelled && requested === revision.current) setState(result);
+    };
+
+    const configure = async (api: ThreadAgentClient): Promise<void> => {
+      if (!api.agentConfigure) return;
+      const requested = ++revision.current;
+      const configured = await api.agentConfigure({ id });
+
+      if (!cancelled && requested === revision.current) setState(configured);
+      await refresh(api);
+    };
+
+    void (async () => {
+      try {
+        if (injected) {
+          await refresh(injected);
+          await configure(injected);
+
+          return;
+        }
+        connection = await DaemonClient.connect({ home, autostart: true });
+        if (cancelled) {
+          connection.close();
+          return;
+        }
+        setClient(connection);
+        unsubscribe = connection.onEvent((event) => {
+          if (event.event === "agent.updated" && event.sessionId === id)
+            void refresh(connection!).catch((error) => {
+              if (!cancelled) setError(String(error));
+            });
+        });
+        await connection.subscribe();
+        await refresh(connection);
+        await configure(connection);
+      } catch (error) {
+        if (!cancelled) setError(error instanceof Error ? error.message : String(error));
+      }
+    })();
+
+    return () => {
+      cancelled = true;
+      unsubscribe?.();
+      connection?.close();
+    };
+  }, [id, home, injected]);
+
+  const act = async (
+    request: (client: ThreadAgentClient) => Promise<ThreadAgentState>,
+  ): Promise<boolean> => {
+    if (!client) return false;
+    try {
+      const requested = ++revision.current;
+      const result = await request(client);
+
+      if (requested === revision.current) setState(result);
+      setError("");
+
+      return true;
+    } catch (error) {
+      setError(error instanceof Error ? error.message : String(error));
+
+      return false;
+    }
+  };
+
+  return { state, client, error, setError, act };
+}

package/src/workbench/components/GridTabContent.tsx

@@ -34,6 +34,9 @@ export type DiffSurfaceProps = Pick<
   | "onAnnotate"
   | "onReply"
   | "onUpdateAnnotation"
+  | "isAnnotationReadOnly"
+  | "annotationAction"
+  | "onInvoke"
   | "resolveAuthorLabel"
   | "onNavCommand"
   | "onExit"
@@ -128,6 +131,9 @@ export function GridTabContent(props: {
         onAddComment={(anchor, body) => props.onAddFileComment(filePath, anchor, body)}
         onReply={props.surface.onReply}
         onUpdateAnnotation={props.surface.onUpdateAnnotation}
+        isAnnotationReadOnly={props.surface.isAnnotationReadOnly}
+        annotationAction={props.surface.annotationAction}
+        onInvoke={props.surface.onInvoke}
         resolveAuthorLabel={props.surface.resolveAuthorLabel}
         onExit={props.surface.onExit}
         theme={props.theme}