301 lines
8.8 KiB
TypeScript
301 lines
8.8 KiB
TypeScript
import { Box, Text, useInput } from "ink";
|
||
import { memo, useMemo, useState } from "react";
|
||
import { useProgressIndicator } from "../hooks/useProgressIndicator";
|
||
import { useTerminalWidth } from "../hooks/useTerminalWidth";
|
||
import { useTextInputCursor } from "../hooks/useTextInputCursor";
|
||
import { colors } from "./colors";
|
||
|
||
type Props = {
|
||
taskInfo: {
|
||
subagentType: string;
|
||
description: string;
|
||
prompt: string;
|
||
model?: string;
|
||
};
|
||
onApprove: () => void;
|
||
onApproveAlways: (scope: "project" | "session") => void;
|
||
onDeny: (reason: string) => void;
|
||
onCancel?: () => void;
|
||
isFocused?: boolean;
|
||
approveAlwaysText?: string;
|
||
allowPersistence?: boolean;
|
||
};
|
||
|
||
// Horizontal line character for Claude Code style
|
||
const SOLID_LINE = "─";
|
||
|
||
/**
|
||
* Truncate text to max length with ellipsis
|
||
*/
|
||
function truncate(text: string, maxLength: number): string {
|
||
if (text.length <= maxLength) return text;
|
||
return `${text.slice(0, maxLength - 3)}...`;
|
||
}
|
||
|
||
/**
|
||
* InlineTaskApproval - Renders Task tool approval UI inline with pretty formatting
|
||
*
|
||
* Shows subagent type, description, and prompt in a readable format.
|
||
*/
|
||
export const InlineTaskApproval = memo(
|
||
({
|
||
taskInfo,
|
||
onApprove,
|
||
onApproveAlways,
|
||
onDeny,
|
||
onCancel,
|
||
isFocused = true,
|
||
approveAlwaysText,
|
||
allowPersistence = true,
|
||
}: Props) => {
|
||
const [selectedOption, setSelectedOption] = useState(0);
|
||
const {
|
||
text: customReason,
|
||
cursorPos,
|
||
handleKey,
|
||
clear,
|
||
} = useTextInputCursor();
|
||
const columns = useTerminalWidth();
|
||
useProgressIndicator();
|
||
|
||
// Custom option index depends on whether "always" option is shown
|
||
const customOptionIndex = allowPersistence ? 2 : 1;
|
||
const maxOptionIndex = customOptionIndex;
|
||
const isOnCustomOption = selectedOption === customOptionIndex;
|
||
const customOptionPlaceholder =
|
||
"No, and tell Letta Code what to do differently";
|
||
|
||
useInput(
|
||
(input, key) => {
|
||
if (!isFocused) return;
|
||
|
||
// CTRL-C: cancel (queue denial, return to input)
|
||
if (key.ctrl && input === "c") {
|
||
onCancel?.();
|
||
return;
|
||
}
|
||
|
||
// Arrow navigation always works
|
||
if (key.upArrow) {
|
||
setSelectedOption((prev) => Math.max(0, prev - 1));
|
||
return;
|
||
}
|
||
if (key.downArrow) {
|
||
setSelectedOption((prev) => Math.min(maxOptionIndex, prev + 1));
|
||
return;
|
||
}
|
||
|
||
// When on custom input option
|
||
if (isOnCustomOption) {
|
||
if (key.return) {
|
||
if (customReason.trim()) {
|
||
onDeny(customReason.trim());
|
||
}
|
||
return;
|
||
}
|
||
if (key.escape) {
|
||
if (customReason) {
|
||
clear();
|
||
} else {
|
||
onCancel?.();
|
||
}
|
||
return;
|
||
}
|
||
// Handle text input (arrows, backspace, typing)
|
||
if (handleKey(input, key)) return;
|
||
}
|
||
|
||
// When on regular options
|
||
if (key.return) {
|
||
if (selectedOption === 0) {
|
||
onApprove();
|
||
} else if (selectedOption === 1 && allowPersistence) {
|
||
onApproveAlways("session");
|
||
}
|
||
return;
|
||
}
|
||
if (key.escape) {
|
||
onCancel?.();
|
||
}
|
||
},
|
||
{ isActive: isFocused },
|
||
);
|
||
|
||
// Generate horizontal line
|
||
const solidLine = SOLID_LINE.repeat(Math.max(columns - 2, 10));
|
||
const contentWidth = Math.max(0, columns - 4); // 2 padding on each side
|
||
|
||
// Memoize the static task content so it doesn't re-render on keystroke
|
||
const memoizedTaskContent = useMemo(() => {
|
||
const { subagentType, description, prompt, model } = taskInfo;
|
||
|
||
// Truncate prompt if too long (show first ~200 chars)
|
||
const truncatedPrompt = truncate(prompt, 300);
|
||
|
||
return (
|
||
<>
|
||
{/* Top solid line */}
|
||
<Text dimColor>{solidLine}</Text>
|
||
|
||
{/* Header */}
|
||
<Text bold color={colors.approval.header}>
|
||
Run Task?
|
||
</Text>
|
||
|
||
{/* Task details */}
|
||
<Box paddingLeft={2} flexDirection="column" marginTop={1}>
|
||
{/* Subagent type */}
|
||
<Box flexDirection="row">
|
||
<Box width={12} flexShrink={0}>
|
||
<Text dimColor>Type:</Text>
|
||
</Box>
|
||
<Box flexGrow={1} width={contentWidth - 12}>
|
||
<Text wrap="wrap" color={colors.subagent.header}>
|
||
{subagentType}
|
||
</Text>
|
||
</Box>
|
||
</Box>
|
||
|
||
{/* Model (if specified) */}
|
||
{model && (
|
||
<Box flexDirection="row">
|
||
<Box width={12} flexShrink={0}>
|
||
<Text dimColor>Model:</Text>
|
||
</Box>
|
||
<Box flexGrow={1} width={contentWidth - 12}>
|
||
<Text wrap="wrap" dimColor>
|
||
{model}
|
||
</Text>
|
||
</Box>
|
||
</Box>
|
||
)}
|
||
|
||
{/* Description */}
|
||
<Box flexDirection="row">
|
||
<Box width={12} flexShrink={0}>
|
||
<Text dimColor>Description:</Text>
|
||
</Box>
|
||
<Box flexGrow={1} width={contentWidth - 12}>
|
||
<Text wrap="wrap">{description}</Text>
|
||
</Box>
|
||
</Box>
|
||
|
||
{/* Prompt */}
|
||
<Box flexDirection="row" marginTop={1}>
|
||
<Box width={12} flexShrink={0}>
|
||
<Text dimColor>Prompt:</Text>
|
||
</Box>
|
||
<Box flexGrow={1} width={contentWidth - 12}>
|
||
<Text wrap="wrap" dimColor>
|
||
{truncatedPrompt}
|
||
</Text>
|
||
</Box>
|
||
</Box>
|
||
</Box>
|
||
</>
|
||
);
|
||
}, [taskInfo, solidLine, contentWidth]);
|
||
|
||
// Hint text based on state
|
||
const hintText = isOnCustomOption
|
||
? customReason
|
||
? "Enter to submit · Esc to clear"
|
||
: "Type reason · Esc to cancel"
|
||
: "Enter to select · Esc to cancel";
|
||
|
||
// Generate "always" text for Task tool
|
||
const alwaysText =
|
||
approveAlwaysText || "Yes, allow Task operations during this session";
|
||
|
||
return (
|
||
<Box flexDirection="column">
|
||
{/* Static task content - memoized to prevent re-render on keystroke */}
|
||
{memoizedTaskContent}
|
||
|
||
{/* Options */}
|
||
<Box marginTop={1} flexDirection="column">
|
||
{/* Option 1: Yes */}
|
||
<Box flexDirection="row">
|
||
<Box width={5} flexShrink={0}>
|
||
<Text
|
||
color={
|
||
selectedOption === 0 ? colors.approval.header : undefined
|
||
}
|
||
>
|
||
{selectedOption === 0 ? "❯" : " "} 1.
|
||
</Text>
|
||
</Box>
|
||
<Box flexGrow={1} width={Math.max(0, columns - 5)}>
|
||
<Text
|
||
wrap="wrap"
|
||
color={
|
||
selectedOption === 0 ? colors.approval.header : undefined
|
||
}
|
||
>
|
||
Yes
|
||
</Text>
|
||
</Box>
|
||
</Box>
|
||
|
||
{/* Option 2: Yes, always (only if persistence allowed) */}
|
||
{allowPersistence && (
|
||
<Box flexDirection="row">
|
||
<Box width={5} flexShrink={0}>
|
||
<Text
|
||
color={
|
||
selectedOption === 1 ? colors.approval.header : undefined
|
||
}
|
||
>
|
||
{selectedOption === 1 ? "❯" : " "} 2.
|
||
</Text>
|
||
</Box>
|
||
<Box flexGrow={1} width={Math.max(0, columns - 5)}>
|
||
<Text
|
||
wrap="wrap"
|
||
color={
|
||
selectedOption === 1 ? colors.approval.header : undefined
|
||
}
|
||
>
|
||
{alwaysText}
|
||
</Text>
|
||
</Box>
|
||
</Box>
|
||
)}
|
||
|
||
{/* Custom input option */}
|
||
<Box flexDirection="row">
|
||
<Box width={5} flexShrink={0}>
|
||
<Text
|
||
color={isOnCustomOption ? colors.approval.header : undefined}
|
||
>
|
||
{isOnCustomOption ? "❯" : " "} {customOptionIndex + 1}.
|
||
</Text>
|
||
</Box>
|
||
<Box flexGrow={1} width={Math.max(0, columns - 5)}>
|
||
{customReason ? (
|
||
<Text wrap="wrap">
|
||
{customReason.slice(0, cursorPos)}
|
||
{isOnCustomOption && "█"}
|
||
{customReason.slice(cursorPos)}
|
||
</Text>
|
||
) : (
|
||
<Text wrap="wrap" dimColor>
|
||
{customOptionPlaceholder}
|
||
{isOnCustomOption && "█"}
|
||
</Text>
|
||
)}
|
||
</Box>
|
||
</Box>
|
||
</Box>
|
||
|
||
{/* Hint */}
|
||
<Box marginTop={1}>
|
||
<Text dimColor>{hintText}</Text>
|
||
</Box>
|
||
</Box>
|
||
);
|
||
},
|
||
);
|
||
|
||
InlineTaskApproval.displayName = "InlineTaskApproval";
|