tutti/packages/agent/gui/agent-gui/agentGuiNode/view/useAgentGUIDetailSideConversation.ts

import { useCallback, useMemo, useState } from "react";
import type { AgentActivitySlashCommandPolicy } from "@tutti-os/agent-activity-core";
import {
  useAgentSideConversationSnapshot,
  useAgentSideConversationSupport,
  useOptionalAgentSideConversationRuntime
} from "../../../agentSideConversationRuntime";
import type { AgentConversationPromptVM } from "../../../shared/agentConversation/contracts/agentConversationVM";
import type { AgentComposerProps } from "../AgentComposer";
import type { AgentComposerDraft } from "../model/agentGuiNodeTypes";
import {
  agentComposerDraftPrompt,
  appendAgentComposerDraftQuote,
  emptyAgentComposerDraft
} from "../model/agentComposerDraft";
import { useTranslation } from "../../../i18n/index";
import { projectAgentSideConversationViewState } from "../../../agentSideConversationViewProjection";
import type { AgentPromptContentBlock } from "../../../shared/contracts/dto/agentSession";

export function parseAgentSideInvocation(
  content: readonly AgentPromptContentBlock[]
): { prompt: string | null; contentSupported: boolean } | null {
  const text = content
    .filter((block) => block.type === "text")
    .map((block) => block.text ?? "")
    .join("");
  const invocation = text.trim().match(/^\/side(?:\s+([\s\S]*))?$/);
  if (!invocation) return null;
  return {
    prompt: invocation[1]?.trim() || null,
    contentSupported: content.every((block) => block.type === "text")
  };
}

export function appendAgentSidePromptToDraft(
  draft: AgentComposerDraft,
  prompt: string
): AgentComposerDraft {
  const normalizedPrompt = prompt.trim();
  if (!normalizedPrompt) return draft;
  const [textBlock, ...attachmentBlocks] = draft;
  const currentText = textBlock.text.trim();
  return [
    {
      ...textBlock,
      text: currentText
        ? `${textBlock.text}\n${normalizedPrompt}`
        : normalizedPrompt
    },
    ...attachmentBlocks
  ];
}

function projectAgentSideSlashCommandPolicy({
  commands,
  enabled,
  policy
}: {
  commands: AgentComposerProps["availableCommands"];
  enabled: boolean;
  policy: AgentActivitySlashCommandPolicy | null | undefined;
}): AgentActivitySlashCommandPolicy | null | undefined {
  if (!enabled) return policy;
  return {
    ...policy,
    fallbackCommands:
      policy?.fallbackCommands ?? commands.map((command) => command.name),
    commandEffects: [
      ...(policy?.commandEffects ?? []).filter(
        (entry) => entry.command.trim().toLowerCase() !== "side"
      ),
      { command: "side", effect: "submitImmediate" }
    ]
  };
}

interface UseAgentGUIDetailSideConversationInput {
  enabled?: boolean;
  workspaceId: string;
  sourceAgentSessionId: string | null;
  provider: string;
  cwd: string | null;
  capabilityRevision?: string;
  availableCommands: AgentComposerProps["availableCommands"];
  slashCommandPolicy?: AgentActivitySlashCommandPolicy | null;
  clearMainDraft: () => void;
  submitPrompt: NonNullable<AgentComposerProps["onSubmit"]>;
}

export function useAgentGUIDetailSideConversation({
  enabled = true,
  workspaceId,
  sourceAgentSessionId,
  provider,
  cwd,
  capabilityRevision = "",
  availableCommands,
  slashCommandPolicy = null,
  clearMainDraft,
  submitPrompt
}: UseAgentGUIDetailSideConversationInput) {
  const { t } = useTranslation();
  const runtime = useOptionalAgentSideConversationRuntime();
  const [entryErrorState, setEntryErrorState] = useState<{
    identity: string;
    runtime: typeof runtime;
    code: "content_unsupported" | "operation_failed";
  } | null>(null);
  const capabilityIdentity = `${workspaceId}:${sourceAgentSessionId ?? ""}:${provider}:${cwd ?? ""}`;
  const entryError =
    entryErrorState?.identity === capabilityIdentity &&
    entryErrorState.runtime === runtime
      ? entryErrorState.code
      : null;
  const sideSupported = useAgentSideConversationSupport(
    {
      workspaceId,
      sourceAgentSessionId: enabled ? (sourceAgentSessionId ?? "") : "",
      provider,
      cwd
    },
    capabilityRevision
  );
  const runtimeActive = useAgentSideConversationSnapshot(workspaceId).active;
  const active = useMemo(
    () => projectAgentSideConversationViewState(runtimeActive),
    [runtimeActive]
  );
  const activeSideAgentSessionId = active?.sideAgentSessionId ?? null;
  const hasCurrentSourceSide =
    active?.sourceAgentSessionId === sourceAgentSessionId &&
    (active.status === "opening" ||
      active.status === "idle" ||
      active.status === "running");
  const currentSourceSideUnavailable =
    active?.sourceAgentSessionId === sourceAgentSessionId &&
    !hasCurrentSourceSide;
  const sideAvailable =
    !currentSourceSideUnavailable && (sideSupported || hasCurrentSourceSide);
  const [focusedSideAgentSessionId, setFocusedSideAgentSessionId] = useState<
    string | null
  >(null);
  const focused =
    activeSideAgentSessionId !== null &&
    active?.sourceAgentSessionId === sourceAgentSessionId &&
    activeSideAgentSessionId === focusedSideAgentSessionId;
  const [focusRequestSequence, setFocusRequestSequence] = useState<
    number | null
  >(null);
  const emptyDraft = useMemo(emptyAgentComposerDraft, [
    activeSideAgentSessionId
  ]);
  const [draftState, setDraftState] = useState<{
    sideAgentSessionId: string | null;
    content: AgentComposerDraft;
  }>(() => ({
    sideAgentSessionId: null,
    content: emptyAgentComposerDraft()
  }));
  const draftContent =
    draftState.sideAgentSessionId === activeSideAgentSessionId
      ? draftState.content
      : emptyDraft;
  const setDraftContent = useCallback(
    (content: AgentComposerDraft) =>
      setDraftState({ sideAgentSessionId: activeSideAgentSessionId, content }),
    [activeSideAgentSessionId]
  );
  const setFocused = useCallback(
    (nextFocused: boolean) =>
      setFocusedSideAgentSessionId(
        nextFocused ? activeSideAgentSessionId : null
      ),
    [activeSideAgentSessionId]
  );

  const open = useCallback(
    async (initialPrompt?: string | null) => {
      if (!enabled || !runtime || !sourceAgentSessionId) return null;
      setEntryErrorState(null);
      try {
        const existing = runtime.getSnapshot(workspaceId).active;
        if (existing?.sourceAgentSessionId === sourceAgentSessionId) {
          if (
            existing.status === "error" &&
            existing.error === "side_close_failed"
          ) {
            await runtime.close({
              workspaceId,
              sideAgentSessionId: existing.sideAgentSessionId
            });
          } else {
            const prompt = initialPrompt?.trim();
            if (prompt) {
              if (existing.status === "idle") {
                await runtime.send({
                  workspaceId,
                  sideAgentSessionId: existing.sideAgentSessionId,
                  content: [{ type: "text", text: prompt }],
                  displayPrompt: prompt
                });
              } else {
                setDraftState((current) => ({
                  sideAgentSessionId: existing.sideAgentSessionId,
                  content: appendAgentSidePromptToDraft(
                    current.sideAgentSessionId === existing.sideAgentSessionId
                      ? current.content
                      : emptyAgentComposerDraft(),
                    prompt
                  )
                }));
                setFocusedSideAgentSessionId(existing.sideAgentSessionId);
              }
            }
            return existing;
          }
        }
        // Capability discovery already gates the /side command. Enter the
        // local opening state immediately so the Side shell does not wait on
        // another daemon round trip before it becomes visible. The Host and
        // Runtime still perform the authoritative live-source validation as
        // part of open.
        const opened = await runtime.open({
          workspaceId,
          sourceAgentSessionId,
          provider,
          cwd
        });
        if (initialPrompt?.trim()) {
          await runtime.send({
            workspaceId,
            sideAgentSessionId: opened.sideAgentSessionId,
            content: [{ type: "text", text: initialPrompt.trim() }],
            displayPrompt: initialPrompt.trim()
          });
        }
        return opened;
      } catch (error) {
        setEntryErrorState({
          identity: capabilityIdentity,
          runtime,
          code: "operation_failed"
        });
        throw error;
      }
    },
    [
      capabilityIdentity,
      cwd,
      enabled,
      provider,
      runtime,
      sourceAgentSessionId,
      workspaceId
    ]
  );

  const submitMain = useCallback<NonNullable<AgentComposerProps["onSubmit"]>>(
    (content, displayPrompt, options) => {
      if (!enabled) {
        submitPrompt(content, displayPrompt, options);
        return;
      }
      const invocation = parseAgentSideInvocation(content);
      if (!invocation) {
        submitPrompt(content, displayPrompt, options);
        return;
      }
      if (!sideAvailable) {
        // /side is an isolation boundary, not an ordinary provider command.
        // Match Codex App's fail-closed behavior: if the exact live source
        // cannot open Side, never leak the intended Side prompt into main.
        setEntryErrorState({
          identity: capabilityIdentity,
          runtime,
          code: "operation_failed"
        });
        return;
      }
      if (!invocation.contentSupported) {
        setEntryErrorState({
          identity: capabilityIdentity,
          runtime,
          code: "content_unsupported"
        });
        return;
      }
      void open(invocation.prompt)
        .then(() => clearMainDraft())
        .catch(() => {});
    },
    [
      capabilityIdentity,
      clearMainDraft,
      enabled,
      open,
      runtime,
      sideAvailable,
      submitPrompt
    ]
  );

  const submitSide = useCallback<NonNullable<AgentComposerProps["onSubmit"]>>(
    (content, displayPrompt) => {
      if (!runtime || !active || active.status !== "idle") return;
      const draftPrompt = agentComposerDraftPrompt(draftContent);
      void runtime
        .send({
          workspaceId,
          sideAgentSessionId: active.sideAgentSessionId,
          content,
          // Quotes are provider context, not the user's visible message. The
          // Composer's generic display prompt includes quote Markdown so main
          // conversations can preserve their historical representation; Side
          // keeps that full content on the wire while projecting only the
          // typed question into its transient timeline.
          displayPrompt: draftPrompt.trim() ? draftPrompt : displayPrompt
        })
        .then(() => setDraftContent(emptyAgentComposerDraft()))
        .catch(() => {});
    },
    [active, draftContent, runtime, setDraftContent, workspaceId]
  );

  const stageSelection = useCallback(
    async (text: string) => {
      const normalizedText = text.trim();
      if (!normalizedText || !enabled || !runtime || !sourceAgentSessionId) {
        return;
      }
      const existing = runtime.getSnapshot(workspaceId).active;
      if (existing && existing.sourceAgentSessionId !== sourceAgentSessionId) {
        return;
      }
      const target = existing ?? (await open());
      if (!target) return;
      setDraftState((current) => ({
        sideAgentSessionId: target.sideAgentSessionId,
        content: appendAgentComposerDraftQuote(
          current.sideAgentSessionId === target.sideAgentSessionId
            ? current.content
            : emptyAgentComposerDraft(),
          {
            type: "quote",
            id: crypto.randomUUID(),
            text: normalizedText
          }
        )
      }));
      setFocusedSideAgentSessionId(target.sideAgentSessionId);
      setFocusRequestSequence((current) => (current ?? 0) + 1);
    },
    [enabled, open, runtime, sourceAgentSessionId, workspaceId]
  );

  const sideCommandEnabled = Boolean(
    runtime &&
    enabled &&
    sourceAgentSessionId &&
    sideAvailable &&
    (!active || active.sourceAgentSessionId === sourceAgentSessionId)
  );
  const commands = useMemo(() => {
    const commandsWithoutSide = availableCommands.filter(
      (command) => command.name.trim().toLowerCase() !== "side"
    );
    if (!sideCommandEnabled) {
      return commandsWithoutSide;
    }
    return [
      ...commandsWithoutSide,
      {
        name: "side",
        description: t("agentHost.agentGui.sideCommandDescription")
      }
    ];
  }, [availableCommands, sideCommandEnabled, t]);
  const projectedSlashCommandPolicy = useMemo(
    () =>
      projectAgentSideSlashCommandPolicy({
        commands,
        enabled: sideCommandEnabled,
        policy: slashCommandPolicy
      }),
    [commands, sideCommandEnabled, slashCommandPolicy]
  );

  const interrupt = useCallback(() => {
    if (!runtime || !active?.activeTurnId) return;
    void runtime
      .cancel({
        workspaceId,
        sideAgentSessionId: active.sideAgentSessionId,
        turnId: active.activeTurnId
      })
      .catch(() => {});
  }, [active, runtime, workspaceId]);

  const close = useCallback(async () => {
    if (!runtime || !active) return;
    await runtime.close({
      workspaceId,
      sideAgentSessionId: active.sideAgentSessionId
    });
  }, [active, runtime, workspaceId]);

  const [interactionSubmitting, setInteractionSubmitting] = useState(false);
  const interactivePrompt = useMemo<AgentConversationPromptVM | null>(() => {
    const interaction = active?.pendingInteraction;
    if (!interaction) return null;
    if (interaction.kind === "question") {
      const rawQuestions = Array.isArray(interaction.input.questions)
        ? interaction.input.questions
        : [];
      return {
        agentSessionId: active.sideAgentSessionId,
        kind: "ask-user",
        requestId: interaction.requestId,
        title:
          interaction.toolName ?? t("agentHost.agentGui.sideInteractionTitle"),
        turnId: interaction.turnId,
        questions: rawQuestions.flatMap((rawQuestion, index) => {
          if (!rawQuestion || typeof rawQuestion !== "object") return [];
          const question = rawQuestion as Record<string, unknown>;
          const rawOptions = Array.isArray(question.options)
            ? question.options
            : [];
          return [
            {
              id:
                typeof question.id === "string"
                  ? question.id
                  : `question-${index + 1}`,
              header:
                typeof question.header === "string"
                  ? question.header
                  : t("agentHost.agentGui.sideInteractionTitle"),
              question:
                typeof question.question === "string" ? question.question : "",
              options: rawOptions.flatMap((rawOption) => {
                if (!rawOption || typeof rawOption !== "object") return [];
                const option = rawOption as Record<string, unknown>;
                const label =
                  typeof option.label === "string"
                    ? option.label
                    : typeof option.name === "string"
                      ? option.name
                      : "";
                return label
                  ? [
                      {
                        label,
                        description:
                          typeof option.description === "string"
                            ? option.description
                            : ""
                      }
                    ]
                  : [];
              }),
              multiSelect: question.multiSelect === true
            }
          ];
        })
      };
    }
    return {
      kind: "approval",
      id: interaction.requestId,
      requestId: interaction.requestId,
      turnId: interaction.turnId,
      callId: interaction.requestId,
      title:
        interaction.toolName ?? t("agentHost.agentGui.sideInteractionTitle"),
      toolName: interaction.toolName,
      status: "pending",
      input: interaction.input,
      options: interaction.actions.map((action) => ({
        id: action.id,
        label: action.label,
        kind: action.semantic
      })),
      occurredAtUnixMs: null
    };
  }, [active?.pendingInteraction, t]);

  const submitInteraction = useCallback(
    async (input: {
      requestId: string;
      action?: string;
      optionId?: string;
      payload?: Record<string, unknown>;
    }) => {
      const interaction = active?.pendingInteraction;
      if (!runtime || !active || !interaction) return;
      setInteractionSubmitting(true);
      try {
        await runtime.respond({
          workspaceId,
          sideAgentSessionId: active.sideAgentSessionId,
          turnId: interaction.turnId,
          ...input
        });
      } finally {
        setInteractionSubmitting(false);
      }
    },
    [active, runtime, workspaceId]
  );

  return {
    active,
    canOpen: Boolean(
      runtime &&
      sourceAgentSessionId &&
      sideAvailable &&
      (!active || active.sourceAgentSessionId === sourceAgentSessionId)
    ),
    close,
    commands,
    draftContent,
    entryError,
    focused,
    focusRequestSequence,
    interactionSubmitting,
    interactivePrompt,
    interrupt,
    open,
    setFocused,
    setDraftContent,
    slashCommandPolicy: projectedSlashCommandPolicy,
    stageSelection,
    sourceAgentSessionId,
    submitMain,
    submitSide,
    submitInteraction
  };
}