Files
kandev/apps/web/components/task/chat/chat-input-area.tsx
T

661 lines
22 KiB
TypeScript

"use client";
import { useCallback, useState, type ReactNode } from "react";
import { IconArrowRight } from "@tabler/icons-react";
import { Button } from "@kandev/ui/button";
import { Tooltip, TooltipContent, TooltipTrigger } from "@kandev/ui/tooltip";
import { TodoIndicator } from "./todo-indicator";
import { AutoScrollToggleButton } from "./auto-scroll-toggle-button";
import { PRMergedBanner, PRClosedBanner } from "./pr-archive-banners";
import { PRStatusChip } from "@/components/github/pr-status-chip";
import { MRStatusChip } from "@/components/gitlab/mr-status-chip";
import { AzureDevOpsTaskPullRequestChip } from "@/components/azure-devops/azure-devops-task-pull-request-chip";
import { RegisteredChangeRequestStatus } from "@/components/integrations/registered-change-request-status";
import { shareableSessionStateClient } from "@/components/task/share/share-button";
import { TranscriptNavGroup } from "@/components/task/chat/transcript-nav-group";
import { getWebSocketClient } from "@/lib/ws/connection";
import { useKeyboardShortcut } from "@/hooks/use-keyboard-shortcut";
import { useMessageHandler, buildTaskMentionsContext } from "@/hooks/use-message-handler";
import { useAppStore, useAppStoreApi } from "@/components/state-provider";
import { getShortcut } from "@/lib/keyboard/shortcut-overrides";
import {
ChatInputContainer,
type ChatSubmitPayload,
type ChatSubmitResult,
type ChatInputContainerHandle,
} from "@/components/task/chat/chat-input-container";
import { QueueAffordance } from "@/components/task/chat/queued-ghost-list";
import {
formatReviewCommentsAsMarkdown,
formatPRFeedbackAsMarkdown,
formatPlanCommentsAsMarkdown,
formatWalkthroughCommentsAsMarkdown,
formatAgentMessageCommentsAsMarkdown,
} from "@/lib/state/slices/comments/format";
import { usePlanActions } from "@/hooks/domains/kanban/use-plan-actions";
import { useExecutorEnvironmentAvailability } from "@/hooks/domains/session/use-executor-environment-availability";
import { useToast } from "@/components/toast-provider";
import { isMessageSendError } from "@/lib/chat/message-send-error";
import type { DiffComment } from "@/lib/diff/types";
import type { AgentMessageComment } from "@/lib/state/slices/comments";
import type { ChatPanelState } from "./use-chat-panel-state";
import { useComposerProps } from "./use-composer-props";
import { cn } from "@/lib/utils";
import { resolveComposerWorkspaceId } from "./composer-workspace";
import { useTranslation } from "react-i18next";
import { t } from "@/lib/i18n";
import { AutopilotChatChip, useTaskAutopilot } from "./task-autopilot-chat-chip";
const PLAN_CONTEXT_PATH = "plan:context";
/**
* Prepends any pending review/walkthrough/PR-feedback/plan/message comments
* to the composer text as Markdown, in a fixed stacking order, before send.
*/
export function buildSubmitMessage({
message,
reviewComments,
pendingPRFeedback,
planComments,
walkthroughComments = [],
messageComments = [],
}: {
message: string;
reviewComments?: DiffComment[];
pendingPRFeedback: import("@/lib/state/slices/comments").PRFeedbackComment[];
planComments: import("@/lib/state/slices/comments").PlanComment[];
walkthroughComments?: import("@/lib/state/slices/comments").WalkthroughComment[];
messageComments?: AgentMessageComment[];
}): string {
let finalMessage = message;
if (reviewComments && reviewComments.length > 0) {
finalMessage = formatReviewCommentsAsMarkdown(reviewComments) + (message || "");
}
if (walkthroughComments.length > 0) {
finalMessage = formatWalkthroughCommentsAsMarkdown(walkthroughComments) + finalMessage;
}
if (pendingPRFeedback.length > 0) {
finalMessage = formatPRFeedbackAsMarkdown(pendingPRFeedback) + finalMessage;
}
if (planComments.length > 0) {
const planMarkdown = formatPlanCommentsAsMarkdown(planComments);
finalMessage = finalMessage ? `${planMarkdown}${finalMessage}` : planMarkdown;
}
if (messageComments.length > 0) {
const messageCommentsMarkdown = formatAgentMessageCommentsAsMarkdown(messageComments);
finalMessage = finalMessage
? `${messageCommentsMarkdown}${finalMessage}`
: messageCommentsMarkdown;
}
return finalMessage;
}
/** Resolves the composer placeholder text for the current agent/document/plan state. */
export function resolveInputPlaceholder(
isAgentBusy: boolean,
activeDocumentType: string | undefined,
planModeEnabled: boolean,
hasClarification: boolean,
needsRecovery: boolean,
): string {
if (needsRecovery) return "Choose a recovery option above to continue...";
if (hasClarification) return "Queue instructions while the question is pending...";
if (isAgentBusy) return "Queue instructions to the agent...";
if (activeDocumentType === "file") return "Continue working on the file...";
if (planModeEnabled) return "Continue working on the plan...";
return "Continue working on the task...";
}
type PlaceholderArgs = {
override: string | undefined;
isMoving: boolean;
isAgentBusy: boolean;
activeDocumentType: string | undefined;
planModeEnabled: boolean;
hasClarification: boolean;
needsRecovery: boolean;
};
/** Picks the composer placeholder: an explicit override wins, then the
* "switching agent" state, then {@link resolveInputPlaceholder}. */
function pickInputPlaceholder(a: PlaceholderArgs): string {
if (a.isMoving) return "Switching agent...";
// Preserve the prior `??` semantics: an explicit "" override (caller wants
// no placeholder text) must NOT fall through to the resolver default.
if (a.override !== undefined) return a.override;
return resolveInputPlaceholder(
a.isAgentBusy,
a.activeDocumentType,
a.planModeEnabled,
a.hasClarification,
a.needsRecovery,
);
}
/** Shows an error toast for a failed message send, distinguishing a known
* send error from an ambiguous connection drop/timeout. */
function showMessageSendToast(error: unknown, toast: ReturnType<typeof useToast>["toast"]) {
console.error("Failed to send message:", error);
if (isMessageSendError(error)) {
toast({
title: t("task:messageNotSent"),
description: error.message,
variant: "error",
});
return;
}
toast({
title: t("task:messageSendStatusUnknown"),
description: t("task:theConnectionDroppedOrTimedOut"),
variant: "error",
});
}
/** Adapts {@link useChatPanelState}'s fields into {@link useMessageHandler}'s params. */
function usePanelMessageHandler(panelState: ChatPanelState) {
const {
resolvedSessionId,
sessionModel,
activeModel,
pendingClarification,
activeDocument,
planComments,
contextFiles,
prompts,
} = panelState;
return useMessageHandler({
resolvedSessionId,
taskId: panelState.taskId,
sessionModel,
activeModel,
planModeEnabled: panelState.planModeEnabled,
hasPendingClarification: !!pendingClarification,
activeDocument,
planComments,
contextFiles,
prompts,
});
}
/** Builds the composer's submit handler, tracking in-flight sends and
* routing errors to a toast. */
export function useSubmitHandler(
panelState: ChatPanelState,
onSend?: (payload: ChatSubmitPayload) => ChatSubmitResult,
) {
const [isSending, setIsSending] = useState(false);
const storeApi = useAppStoreApi();
const { toast } = useToast();
const {
resolvedSessionId,
planComments,
pendingPRFeedback,
walkthroughComments,
messageComments,
markCommentsSent,
clearSessionPlanComments,
handleClearPRFeedback,
handleClearWalkthroughComments,
clearEphemeral,
addContextFile,
planModeEnabled,
pendingClarification,
} = panelState;
const { handleSendMessage } = usePanelMessageHandler(panelState);
const handleSubmit = useCallback(
async (payload: ChatSubmitPayload) => {
if (isSending) return;
setIsSending(true);
try {
const finalMessage = buildSubmitMessage({
message: payload.message,
reviewComments: payload.reviewComments,
pendingPRFeedback,
planComments,
walkthroughComments,
messageComments,
});
const outbound = { ...payload, message: finalMessage };
if (onSend && !pendingClarification) {
// Expand task mentions because onSend bypasses useMessageHandler.buildFinalMessage.
const taskCtx = payload.inlineTaskMentions?.length
? buildTaskMentionsContext(payload.inlineTaskMentions, storeApi.getState())
: "";
await onSend({ ...outbound, message: finalMessage + taskCtx });
} else {
await handleSendMessage(outbound);
}
if (payload.reviewComments && payload.reviewComments.length > 0)
markCommentsSent(payload.reviewComments.map((c) => c.id));
if (messageComments.length > 0) markCommentsSent(messageComments.map((c) => c.id));
if (pendingPRFeedback.length > 0) handleClearPRFeedback();
if (walkthroughComments.length > 0) handleClearWalkthroughComments();
if (planComments.length > 0) clearSessionPlanComments();
if (resolvedSessionId) {
clearEphemeral(resolvedSessionId);
// Re-add plan context if plan mode is still active (clearEphemeral removes unpinned files)
if (planModeEnabled) {
addContextFile(resolvedSessionId, { path: PLAN_CONTEXT_PATH, name: "Plan" });
}
}
} catch (error) {
showMessageSendToast(error, toast);
return false;
} finally {
setIsSending(false);
}
},
[
isSending,
onSend,
pendingClarification,
storeApi,
handleSendMessage,
markCommentsSent,
planComments,
clearSessionPlanComments,
walkthroughComments,
messageComments,
handleClearWalkthroughComments,
pendingPRFeedback,
handleClearPRFeedback,
resolvedSessionId,
clearEphemeral,
planModeEnabled,
addContextFile,
toast,
],
);
return { isSending, handleSubmit };
}
/** Builds the cancel-turn handler and registers the global focus-input keyboard shortcut. */
export function useChatPanelHandlers(
resolvedSessionId: string | null,
chatInputRef: React.RefObject<ChatInputContainerHandle | null>,
) {
const handleCancelTurn = useCallback(async () => {
if (!resolvedSessionId) return;
const client = getWebSocketClient();
if (!client) return;
try {
await client.request("agent.cancel", { session_id: resolvedSessionId }, 15000);
} catch (error) {
console.error("Failed to cancel agent turn:", error);
}
}, [resolvedSessionId]);
const keyboardShortcuts = useAppStore((s) => s.userSettings.keyboardShortcuts);
useKeyboardShortcut(
getShortcut("FOCUS_INPUT", keyboardShortcuts),
useCallback(
(event: KeyboardEvent) => {
const el = document.activeElement;
const isTyping =
el instanceof HTMLInputElement ||
el instanceof HTMLTextAreaElement ||
(el instanceof HTMLElement && el.isContentEditable);
if (isTyping) return;
const inputHandle = chatInputRef.current;
if (inputHandle) {
event.preventDefault();
inputHandle.focusInput();
}
},
[chatInputRef],
),
{ enabled: true, preventDefault: false },
);
return { handleCancelTurn };
}
type TodoDisplayItem = {
text: string;
done?: boolean;
status?: "pending" | "in_progress" | "completed" | "failed";
};
export function shouldRenderChatStatusBar({
hasTask,
hasTodos,
hasQueueChip,
showRightControls,
showProceed,
}: {
hasTask: boolean;
hasTodos: boolean;
hasQueueChip: boolean;
showRightControls: boolean;
showProceed: boolean;
}): boolean {
return hasTask || hasTodos || hasQueueChip || showRightControls || showProceed;
}
function getRightControlVisibility({
taskId,
sessionId,
sessionState,
showAutoScrollControl,
showScrollToLastPrompt,
showScrollToStart,
}: {
taskId: string | null;
sessionId: string | null;
sessionState: string | null;
showAutoScrollControl: boolean;
showScrollToLastPrompt: boolean | undefined;
showScrollToStart: boolean | undefined;
}) {
const canShare = !!taskId && !!sessionId && shareableSessionStateClient(sessionState);
const showRightControls =
(showAutoScrollControl && !!sessionId) ||
canShare ||
!!showScrollToLastPrompt ||
!!showScrollToStart;
return { canShare, showRightControls };
}
/**
* Row above the composer showing todo progress, PR/CI status chips,
* merged/closed PR banners, the auto-scroll toggle + Share (right-aligned),
* and a "move to next step" action when the workflow allows it.
*/
type ChatStatusBarProps = {
todoItems: TodoDisplayItem[];
taskId: string | null;
sessionId: string | null;
sessionState: string | null;
nextStepName: string | null;
onProceed: () => void;
isAgentBusy: boolean;
isMoving: boolean;
queueChip?: ReactNode;
showScrollToLastPrompt?: boolean;
onScrollToLastPrompt?: () => void;
lastPromptScrollDirection?: "up" | "down";
showScrollToStart?: boolean;
onScrollToStart?: () => void;
};
function ChatStatusBar({
todoItems,
taskId,
sessionId,
sessionState,
nextStepName,
onProceed,
isAgentBusy,
isMoving,
queueChip,
showScrollToLastPrompt,
onScrollToLastPrompt,
lastPromptScrollDirection,
showScrollToStart,
onScrollToStart,
}: ChatStatusBarProps) {
const { t } = useTranslation();
const showTodos = todoItems.length > 0;
const showProceed = !!nextStepName && !isAgentBusy;
const autopilot = useTaskAutopilot(taskId);
const showAutoScrollControl = useAppStore(
(state) => state.userSettings.showTranscriptAutoScrollControl,
);
const { canShare, showRightControls } = getRightControlVisibility({
taskId,
sessionId,
sessionState,
showAutoScrollControl,
showScrollToLastPrompt,
showScrollToStart,
});
if (
!shouldRenderChatStatusBar({
hasTask: !!taskId,
hasTodos: showTodos,
hasQueueChip: !!queueChip,
showRightControls,
showProceed,
})
) {
return null;
}
// PRMergedBanner returns null internally when not applicable
return (
<div
data-testid="chat-status-bar"
className="flex min-w-0 flex-wrap items-center gap-1.5 py-1 text-xs text-muted-foreground"
>
{showTodos && <TodoIndicator todos={todoItems} />}
{autopilot && <AutopilotChatChip />}
<PRStatusChip taskId={taskId} />
<MRStatusChip taskId={taskId} />
<AzureDevOpsTaskPullRequestChip taskId={taskId} />
<RegisteredChangeRequestStatus taskId={taskId} sessionId={sessionId} surface="composer" />
{queueChip}
{/* Distinct per-banner keys: the key remounts the banner on task switch
so its dismissed state re-initialises, and keeping the two suffixes
different avoids a duplicate-sibling-key collision. */}
{taskId && <PRMergedBanner key={`${taskId}-merged`} taskId={taskId} />}
{taskId && <PRClosedBanner key={`${taskId}-closed`} taskId={taskId} />}
{showRightControls && (
<div className="ml-auto flex shrink-0 items-center gap-1.5">
{sessionId && <AutoScrollToggleButton sessionId={sessionId} />}
<TranscriptNavGroup
canShare={canShare}
taskId={taskId}
sessionId={sessionId}
showScrollToLastPrompt={showScrollToLastPrompt}
onScrollToLastPrompt={onScrollToLastPrompt}
lastPromptScrollDirection={lastPromptScrollDirection}
showScrollToStart={showScrollToStart}
onScrollToStart={onScrollToStart}
/>
</div>
)}
{showProceed && (
<Tooltip>
<TooltipTrigger asChild>
<Button
type="button"
variant="outline"
size="sm"
className={`${showRightControls ? "" : "ml-auto "}h-6 gap-1 px-2.5 text-xs cursor-pointer text-primary`}
onClick={onProceed}
disabled={isMoving}
data-testid="proceed-next-step"
>
{nextStepName}
<IconArrowRight className="h-3.5 w-3.5" />
</Button>
</TooltipTrigger>
<TooltipContent>{t("task:moveTaskToTheNextWorkflow")}</TooltipContent>
</Tooltip>
)}
</div>
);
}
type ChatInputAreaProps = {
chatInputRef: React.RefObject<ChatInputContainerHandle | null>;
clarificationKey: number;
onClarificationResolved: () => void;
handleSubmit: (payload: ChatSubmitPayload) => ChatSubmitResult;
handleCancelTurn: () => Promise<void>;
showRequestChangesTooltip: boolean;
onRequestChangesTooltipDismiss?: () => void;
panelState: ChatPanelState;
isSending: boolean;
hideSessionsDropdown?: boolean;
minimalToolbar?: boolean;
/** Hide ACP/session-specific controls (model picker, mode, MCP, reset context,
* sessions, enhance, plugin actions) while keeping Plan, attachment and
* context controls. Surfaces whose agent only exists for the length of a turn
* — an automation run — flip this off the moment the turn ends, so a control
* that talks to a live ACP session is never left on screen without one. */
hideAgentControls?: boolean;
/** Hide the plan mode toggle button (for ephemeral/quick chat sessions) */
hidePlanMode?: boolean;
placeholderOverride?: string;
surfaceClassName?: string;
/** Always-on affordance: scrolls the transcript to the top of the last
* user prompt. Omitted callers (e.g. quick chat) render no button. */
showScrollToLastPrompt?: boolean;
onScrollToLastPrompt?: () => void;
/** Direction the last prompt actually sits in, driving the scroll
* button's icon. Ignored while `showScrollToLastPrompt` is falsy. */
lastPromptScrollDirection?: "up" | "down";
/** Shown once the first message scrolls out of view: jumps back to the
* start of the transcript. */
showScrollToStart?: boolean;
onScrollToStart?: () => void;
};
/** Resolves whether this session's executor environment is unavailable, and why. */
function useExecutorUnavailable(taskId: string | null, sessionId: string | null) {
const availability = useExecutorEnvironmentAvailability(taskId, Boolean(sessionId && taskId));
return {
unavailable: availability.unavailable,
reason: availability.status?.label,
};
}
/** Resolves the workspace ID the composer should attribute a new message to. */
function useComposerWorkspaceId(sessionId: string | null, taskId: string | null) {
return useAppStore((state) =>
resolveComposerWorkspaceId({
sessionId,
taskId,
quickChatSessions: state.quickChat.sessions,
activeWorkflowId: state.kanban.workflowId,
activeTasks: state.kanban.tasks,
snapshots: Object.values(state.kanbanMulti.snapshots),
workflows: state.workflows.items,
}),
);
}
/** Derives the composer's plan actions, executor-availability state, and placeholder text. */
function useChatInputDerived(
panelState: ChatPanelState,
chatInputRef: React.RefObject<ChatInputContainerHandle | null>,
placeholderOverride: string | undefined,
) {
const { resolvedSessionId, taskId, isAgentBusy, needsRecovery, planModeEnabled, activeDocument } =
panelState;
const planActions = usePlanActions({
resolvedSessionId,
taskId,
planModeEnabled,
handlePlanModeChange: panelState.handlePlanModeChange,
chatInputRef,
});
const hasClarification = !!panelState.pendingClarification;
const executor = useExecutorUnavailable(taskId, resolvedSessionId);
const placeholder = pickInputPlaceholder({
override: placeholderOverride,
isMoving: planActions.isMoving,
isAgentBusy,
activeDocumentType: activeDocument?.type,
planModeEnabled,
hasClarification,
needsRecovery,
});
return { planActions, executor, placeholder };
}
/** Whether the user can manually drain the queued-message backlog right now
* (no pending clarification, and the session is idle/waiting for input). */
function canManuallyDrainQueue(pendingClarification: unknown, sessionState: string | null) {
return !pendingClarification && (sessionState === "WAITING_FOR_INPUT" || sessionState === "IDLE");
}
/**
* The chat composer: input box, submit/cancel handling, plan-mode toggle,
* clarification banner, and the {@link ChatStatusBar} above it.
*/
export function ChatInputArea({
chatInputRef,
clarificationKey,
onClarificationResolved,
handleSubmit,
handleCancelTurn,
showRequestChangesTooltip,
onRequestChangesTooltipDismiss,
panelState,
isSending,
hideSessionsDropdown,
minimalToolbar,
hideAgentControls,
hidePlanMode,
placeholderOverride,
surfaceClassName,
showScrollToLastPrompt,
onScrollToLastPrompt,
lastPromptScrollDirection,
showScrollToStart,
onScrollToStart,
}: ChatInputAreaProps) {
const { resolvedSessionId, taskId, isAgentBusy } = panelState;
const composerWorkspaceId = useComposerWorkspaceId(resolvedSessionId, taskId);
const sessionState = panelState.session?.state ?? null;
const canDrainQueue = canManuallyDrainQueue(panelState.pendingClarification, sessionState);
const { planActions, executor, placeholder } = useChatInputDerived(
panelState,
chatInputRef,
placeholderOverride,
);
const { implementPlanHandler, proceedStepName, proceed, isMoving } = planActions;
const composerProps = useComposerProps({
panelState,
composerWorkspaceId,
isMoving,
implementPlanHandler,
executor,
placeholder,
handleSubmit,
handleCancelTurn,
isSending,
showRequestChangesTooltip,
onRequestChangesTooltipDismiss,
onClarificationResolved,
hideSessionsDropdown,
minimalToolbar,
hideAgentControls,
hidePlanMode,
});
return (
<div
data-testid="chat-input-area"
className={cn("bg-card flex-shrink-0 px-2 pb-2 pt-1", surfaceClassName)}
>
<QueueAffordance
sessionId={resolvedSessionId}
canDrain={canDrainQueue}
renderStatusBar={(queueChip) => (
<ChatStatusBar
todoItems={panelState.todoItems}
taskId={taskId}
sessionId={resolvedSessionId}
sessionState={sessionState}
nextStepName={proceedStepName}
onProceed={proceed}
isAgentBusy={isAgentBusy}
isMoving={isMoving}
queueChip={queueChip}
showScrollToLastPrompt={showScrollToLastPrompt}
onScrollToLastPrompt={onScrollToLastPrompt}
lastPromptScrollDirection={lastPromptScrollDirection}
showScrollToStart={showScrollToStart}
onScrollToStart={onScrollToStart}
/>
)}
>
<ChatInputContainer ref={chatInputRef} key={clarificationKey} {...composerProps} />
</QueueAffordance>
</div>
);
}