mirror of
https://github.com/pacnpal/Roo-Code.git
synced 2025-12-20 04:11:10 -05:00
825 lines
24 KiB
TypeScript
825 lines
24 KiB
TypeScript
import { VSCodeBadge, VSCodeProgressRing } from "@vscode/webview-ui-toolkit/react"
|
|
import deepEqual from "fast-deep-equal"
|
|
import React, { memo, useEffect, useMemo, useRef } from "react"
|
|
import { ClaudeApiReqInfo, ClaudeMessage, ClaudeSayTool } from "../../../../src/shared/ExtensionMessage"
|
|
import { COMMAND_OUTPUT_STRING } from "../../../../src/shared/combineCommandSequences"
|
|
import { vscode } from "../../utils/vscode"
|
|
import CodeAccordian, { removeLeadingNonAlphanumeric } from "../common/CodeAccordian"
|
|
import CodeBlock, { CODE_BLOCK_BG_COLOR } from "../common/CodeBlock"
|
|
import MarkdownBlock from "../common/MarkdownBlock"
|
|
import Thumbnails from "../common/Thumbnails"
|
|
import { highlightMentions } from "./TaskHeader"
|
|
import { useSize } from "react-use"
|
|
|
|
interface ChatRowProps {
|
|
message: ClaudeMessage
|
|
isExpanded: boolean
|
|
onToggleExpand: () => void
|
|
lastModifiedMessage?: ClaudeMessage
|
|
isLast: boolean
|
|
onHeightChange: (height: number) => void
|
|
}
|
|
|
|
interface ChatRowContentProps extends Omit<ChatRowProps, "onHeightChange"> {}
|
|
|
|
const ChatRow = memo(
|
|
(props: ChatRowProps) => {
|
|
const { isLast, onHeightChange, message } = props
|
|
// Store the previous height to compare with the current height
|
|
// This allows us to detect changes without causing re-renders
|
|
const prevHeightRef = useRef(0)
|
|
|
|
const [chatrow, { height }] = useSize(
|
|
<div
|
|
style={{
|
|
padding: "10px 6px 10px 15px",
|
|
}}>
|
|
<ChatRowContent {...props} />
|
|
</div>
|
|
)
|
|
|
|
useEffect(() => {
|
|
// used for partials, command output, etc.
|
|
const isInitialRender = prevHeightRef.current === 0 // prevents scrolling when new element is added since we already scroll for that
|
|
// height starts off at Infinity
|
|
if (isLast && height !== 0 && height !== Infinity && height !== prevHeightRef.current) {
|
|
prevHeightRef.current = height
|
|
if (!isInitialRender) {
|
|
onHeightChange(height)
|
|
}
|
|
}
|
|
}, [height, isLast, onHeightChange, message])
|
|
|
|
// we cannot return null as virtuoso does not support it, so we use a separate visibleMessages array to filter out messages that should not be rendered
|
|
return chatrow
|
|
},
|
|
// memo does shallow comparison of props, so we need to do deep comparison of arrays/objects whose properties might change
|
|
deepEqual
|
|
)
|
|
|
|
export default ChatRow
|
|
|
|
const ChatRowContent = ({ message, isExpanded, onToggleExpand, lastModifiedMessage, isLast }: ChatRowContentProps) => {
|
|
const [cost, apiReqCancelReason, apiReqStreamingFailedMessage] = useMemo(() => {
|
|
if (message.text != null && message.say === "api_req_started") {
|
|
const info: ClaudeApiReqInfo = JSON.parse(message.text)
|
|
return [info.cost, info.cancelReason, info.streamingFailedMessage]
|
|
}
|
|
return [undefined, undefined, undefined]
|
|
}, [message.text, message.say])
|
|
const apiRequestFailedMessage =
|
|
isLast && lastModifiedMessage?.ask === "api_req_failed" // if request is retried then the latest message is a api_req_retried
|
|
? lastModifiedMessage?.text
|
|
: undefined
|
|
const isCommandExecuting =
|
|
isLast && lastModifiedMessage?.ask === "command" && lastModifiedMessage?.text?.includes(COMMAND_OUTPUT_STRING)
|
|
const type = message.type === "ask" ? message.ask : message.say
|
|
|
|
const normalColor = "var(--vscode-foreground)"
|
|
const errorColor = "var(--vscode-errorForeground)"
|
|
const successColor = "var(--vscode-charts-green)"
|
|
const cancelledColor = "var(--vscode-descriptionForeground)"
|
|
|
|
const [icon, title] = useMemo(() => {
|
|
switch (type) {
|
|
case "error":
|
|
return [
|
|
<span
|
|
className="codicon codicon-error"
|
|
style={{ color: errorColor, marginBottom: "-1.5px" }}></span>,
|
|
<span style={{ color: errorColor, fontWeight: "bold" }}>Error</span>,
|
|
]
|
|
case "mistake_limit_reached":
|
|
return [
|
|
<span
|
|
className="codicon codicon-error"
|
|
style={{ color: errorColor, marginBottom: "-1.5px" }}></span>,
|
|
<span style={{ color: errorColor, fontWeight: "bold" }}>Claude is having trouble...</span>,
|
|
]
|
|
case "command":
|
|
return [
|
|
isCommandExecuting ? (
|
|
<ProgressIndicator />
|
|
) : (
|
|
<span
|
|
className="codicon codicon-terminal"
|
|
style={{ color: normalColor, marginBottom: "-1.5px" }}></span>
|
|
),
|
|
<span style={{ color: normalColor, fontWeight: "bold" }}>
|
|
Claude wants to execute this command:
|
|
</span>,
|
|
]
|
|
case "completion_result":
|
|
return [
|
|
<span
|
|
className="codicon codicon-check"
|
|
style={{ color: successColor, marginBottom: "-1.5px" }}></span>,
|
|
<span style={{ color: successColor, fontWeight: "bold" }}>Task Completed</span>,
|
|
]
|
|
case "api_req_started":
|
|
const getIconSpan = (iconName: string, color: string) => (
|
|
<div
|
|
style={{
|
|
width: 16,
|
|
height: 16,
|
|
display: "flex",
|
|
alignItems: "center",
|
|
justifyContent: "center",
|
|
}}>
|
|
<span
|
|
className={`codicon codicon-${iconName}`}
|
|
style={{
|
|
color,
|
|
fontSize: 16,
|
|
marginBottom: "-1.5px",
|
|
}}></span>
|
|
</div>
|
|
)
|
|
return [
|
|
cost != null ? (
|
|
apiReqCancelReason != null ? (
|
|
apiReqCancelReason === "user_cancelled" ? (
|
|
getIconSpan("error", cancelledColor)
|
|
) : (
|
|
getIconSpan("error", errorColor)
|
|
)
|
|
) : (
|
|
getIconSpan("check", successColor)
|
|
)
|
|
) : apiRequestFailedMessage ? (
|
|
getIconSpan("error", errorColor)
|
|
) : (
|
|
<ProgressIndicator />
|
|
),
|
|
cost != null ? (
|
|
apiReqCancelReason != null ? (
|
|
apiReqCancelReason === "user_cancelled" ? (
|
|
<span style={{ color: normalColor, fontWeight: "bold" }}>API Request Cancelled</span>
|
|
) : (
|
|
<span style={{ color: errorColor, fontWeight: "bold" }}>API Streaming Failed</span>
|
|
)
|
|
) : (
|
|
<span style={{ color: normalColor, fontWeight: "bold" }}>API Request</span>
|
|
)
|
|
) : apiRequestFailedMessage ? (
|
|
<span style={{ color: errorColor, fontWeight: "bold" }}>API Request Failed</span>
|
|
) : (
|
|
<span style={{ color: normalColor, fontWeight: "bold" }}>API Request...</span>
|
|
),
|
|
]
|
|
case "followup":
|
|
return [
|
|
<span
|
|
className="codicon codicon-question"
|
|
style={{ color: normalColor, marginBottom: "-1.5px" }}></span>,
|
|
<span style={{ color: normalColor, fontWeight: "bold" }}>Claude has a question:</span>,
|
|
]
|
|
default:
|
|
return [null, null]
|
|
}
|
|
}, [type, cost, apiRequestFailedMessage, isCommandExecuting, apiReqCancelReason])
|
|
|
|
const headerStyle: React.CSSProperties = {
|
|
display: "flex",
|
|
alignItems: "center",
|
|
gap: "10px",
|
|
marginBottom: "10px",
|
|
}
|
|
|
|
const pStyle: React.CSSProperties = {
|
|
margin: 0,
|
|
whiteSpace: "pre-wrap",
|
|
wordBreak: "break-word",
|
|
overflowWrap: "anywhere",
|
|
}
|
|
|
|
const tool = useMemo(() => {
|
|
if (message.ask === "tool" || message.say === "tool") {
|
|
return JSON.parse(message.text || "{}") as ClaudeSayTool
|
|
}
|
|
return null
|
|
}, [message.ask, message.say, message.text])
|
|
|
|
if (tool) {
|
|
const toolIcon = (name: string) => (
|
|
<span
|
|
className={`codicon codicon-${name}`}
|
|
style={{ color: "var(--vscode-foreground)", marginBottom: "-1.5px" }}></span>
|
|
)
|
|
|
|
switch (tool.tool) {
|
|
case "editedExistingFile":
|
|
return (
|
|
<>
|
|
<div style={headerStyle}>
|
|
{toolIcon("edit")}
|
|
<span style={{ fontWeight: "bold" }}>Claude wants to edit this file:</span>
|
|
</div>
|
|
<CodeAccordian
|
|
diff={tool.diff!}
|
|
path={tool.path!}
|
|
isExpanded={isExpanded}
|
|
onToggleExpand={onToggleExpand}
|
|
/>
|
|
</>
|
|
)
|
|
case "newFileCreated":
|
|
return (
|
|
<>
|
|
<div style={headerStyle}>
|
|
{toolIcon("new-file")}
|
|
<span style={{ fontWeight: "bold" }}>Claude wants to create a new file:</span>
|
|
</div>
|
|
<CodeAccordian
|
|
code={tool.content!}
|
|
path={tool.path!}
|
|
isExpanded={isExpanded}
|
|
onToggleExpand={onToggleExpand}
|
|
/>
|
|
</>
|
|
)
|
|
case "readFile":
|
|
return (
|
|
<>
|
|
<div style={headerStyle}>
|
|
{toolIcon("file-code")}
|
|
<span style={{ fontWeight: "bold" }}>
|
|
{message.type === "ask" ? "Claude wants to read this file:" : "Claude read this file:"}
|
|
</span>
|
|
</div>
|
|
{/* <CodeAccordian
|
|
code={tool.content!}
|
|
path={tool.path!}
|
|
isExpanded={isExpanded}
|
|
onToggleExpand={onToggleExpand}
|
|
/> */}
|
|
<div
|
|
style={{
|
|
borderRadius: 3,
|
|
backgroundColor: CODE_BLOCK_BG_COLOR,
|
|
overflow: "hidden",
|
|
border: "1px solid var(--vscode-editorGroup-border)",
|
|
}}>
|
|
<div
|
|
style={{
|
|
color: "var(--vscode-descriptionForeground)",
|
|
display: "flex",
|
|
alignItems: "center",
|
|
padding: "9px 10px",
|
|
cursor: "pointer",
|
|
userSelect: "none",
|
|
WebkitUserSelect: "none",
|
|
MozUserSelect: "none",
|
|
msUserSelect: "none",
|
|
}}
|
|
onClick={() => {
|
|
vscode.postMessage({ type: "openFile", text: tool.content })
|
|
}}>
|
|
{tool.path?.startsWith(".") && <span>.</span>}
|
|
<span
|
|
style={{
|
|
whiteSpace: "nowrap",
|
|
overflow: "hidden",
|
|
textOverflow: "ellipsis",
|
|
marginRight: "8px",
|
|
direction: "rtl",
|
|
textAlign: "left",
|
|
}}>
|
|
{removeLeadingNonAlphanumeric(tool.path ?? "") + "\u200E"}
|
|
</span>
|
|
<div style={{ flexGrow: 1 }}></div>
|
|
<span
|
|
className={`codicon codicon-link-external`}
|
|
style={{ fontSize: 13.5, margin: "1px 0" }}></span>
|
|
</div>
|
|
</div>
|
|
</>
|
|
)
|
|
case "listFilesTopLevel":
|
|
return (
|
|
<>
|
|
<div style={headerStyle}>
|
|
{toolIcon("folder-opened")}
|
|
<span style={{ fontWeight: "bold" }}>
|
|
{message.type === "ask"
|
|
? "Claude wants to view the top level files in this directory:"
|
|
: "Claude viewed the top level files in this directory:"}
|
|
</span>
|
|
</div>
|
|
<CodeAccordian
|
|
code={tool.content!}
|
|
path={tool.path!}
|
|
language="shell-session"
|
|
isExpanded={isExpanded}
|
|
onToggleExpand={onToggleExpand}
|
|
/>
|
|
</>
|
|
)
|
|
case "listFilesRecursive":
|
|
return (
|
|
<>
|
|
<div style={headerStyle}>
|
|
{toolIcon("folder-opened")}
|
|
<span style={{ fontWeight: "bold" }}>
|
|
{message.type === "ask"
|
|
? "Claude wants to recursively view all files in this directory:"
|
|
: "Claude recursively viewed all files in this directory:"}
|
|
</span>
|
|
</div>
|
|
<CodeAccordian
|
|
code={tool.content!}
|
|
path={tool.path!}
|
|
language="shell-session"
|
|
isExpanded={isExpanded}
|
|
onToggleExpand={onToggleExpand}
|
|
/>
|
|
</>
|
|
)
|
|
case "listCodeDefinitionNames":
|
|
return (
|
|
<>
|
|
<div style={headerStyle}>
|
|
{toolIcon("file-code")}
|
|
<span style={{ fontWeight: "bold" }}>
|
|
{message.type === "ask"
|
|
? "Claude wants to view source code definition names used in this directory:"
|
|
: "Claude viewed source code definition names used in this directory:"}
|
|
</span>
|
|
</div>
|
|
<CodeAccordian
|
|
code={tool.content!}
|
|
path={tool.path!}
|
|
isExpanded={isExpanded}
|
|
onToggleExpand={onToggleExpand}
|
|
/>
|
|
</>
|
|
)
|
|
case "searchFiles":
|
|
return (
|
|
<>
|
|
<div style={headerStyle}>
|
|
{toolIcon("search")}
|
|
<span style={{ fontWeight: "bold" }}>
|
|
{message.type === "ask" ? (
|
|
<>
|
|
Claude wants to search this directory for <code>{tool.regex}</code>:
|
|
</>
|
|
) : (
|
|
<>
|
|
Claude searched this directory for <code>{tool.regex}</code>:
|
|
</>
|
|
)}
|
|
</span>
|
|
</div>
|
|
<CodeAccordian
|
|
code={tool.content!}
|
|
path={tool.path! + (tool.filePattern ? `/(${tool.filePattern})` : "")}
|
|
language="plaintext"
|
|
isExpanded={isExpanded}
|
|
onToggleExpand={onToggleExpand}
|
|
/>
|
|
</>
|
|
)
|
|
case "inspectSite":
|
|
const isInspecting = lastModifiedMessage?.say === "inspect_site_result" && !lastModifiedMessage?.images
|
|
return (
|
|
<>
|
|
<div style={headerStyle}>
|
|
{isInspecting ? <ProgressIndicator /> : toolIcon("inspect")}
|
|
<span style={{ fontWeight: "bold" }}>
|
|
{message.type === "ask" ? (
|
|
<>Claude wants to inspect this website:</>
|
|
) : (
|
|
<>Claude is inspecting this website:</>
|
|
)}
|
|
</span>
|
|
</div>
|
|
<div
|
|
style={{
|
|
borderRadius: 3,
|
|
border: "1px solid var(--vscode-editorGroup-border)",
|
|
overflow: "hidden",
|
|
backgroundColor: CODE_BLOCK_BG_COLOR,
|
|
}}>
|
|
<CodeBlock source={`${"```"}shell\n${tool.path}\n${"```"}`} forceWrap={true} />
|
|
</div>
|
|
</>
|
|
)
|
|
default:
|
|
return null
|
|
}
|
|
}
|
|
|
|
switch (message.type) {
|
|
case "say":
|
|
switch (message.say) {
|
|
case "api_req_started":
|
|
return (
|
|
<>
|
|
<div
|
|
style={{
|
|
...headerStyle,
|
|
marginBottom:
|
|
(cost == null && apiRequestFailedMessage) || apiReqStreamingFailedMessage
|
|
? 10
|
|
: 0,
|
|
justifyContent: "space-between",
|
|
cursor: "pointer",
|
|
userSelect: "none",
|
|
WebkitUserSelect: "none",
|
|
MozUserSelect: "none",
|
|
msUserSelect: "none",
|
|
}}
|
|
onClick={onToggleExpand}>
|
|
<div style={{ display: "flex", alignItems: "center", gap: "10px" }}>
|
|
{icon}
|
|
{title}
|
|
{/* Need to render this everytime since it affects height of row by 2px */}
|
|
<VSCodeBadge style={{ opacity: cost != null && cost > 0 ? 1 : 0 }}>
|
|
${Number(cost || 0)?.toFixed(4)}
|
|
</VSCodeBadge>
|
|
</div>
|
|
<span className={`codicon codicon-chevron-${isExpanded ? "up" : "down"}`}></span>
|
|
</div>
|
|
{((cost == null && apiRequestFailedMessage) || apiReqStreamingFailedMessage) && (
|
|
<>
|
|
<p style={{ ...pStyle, color: "var(--vscode-errorForeground)" }}>
|
|
{apiRequestFailedMessage || apiReqStreamingFailedMessage}
|
|
{apiRequestFailedMessage?.toLowerCase().includes("powershell") && (
|
|
<>
|
|
<br />
|
|
<br />
|
|
It seems like you're having Windows PowerShell issues, please see this{" "}
|
|
<a
|
|
href="https://github.com/saoudrizwan/claude-dev/wiki/TroubleShooting-%E2%80%90-%22PowerShell-is-not-recognized-as-an-internal-or-external-command%22"
|
|
style={{ color: "inherit", textDecoration: "underline" }}>
|
|
troubleshooting guide
|
|
</a>
|
|
.
|
|
</>
|
|
)}
|
|
</p>
|
|
|
|
{/* {apiProvider === "kodu" && (
|
|
<div
|
|
style={{
|
|
display: "flex",
|
|
alignItems: "center",
|
|
backgroundColor:
|
|
"color-mix(in srgb, var(--vscode-errorForeground) 20%, transparent)",
|
|
color: "var(--vscode-editor-foreground)",
|
|
padding: "6px 8px",
|
|
borderRadius: "3px",
|
|
margin: "10px 0 0 0",
|
|
fontSize: "12px",
|
|
}}>
|
|
<i
|
|
className="codicon codicon-warning"
|
|
style={{
|
|
marginRight: 6,
|
|
fontSize: 16,
|
|
color: "var(--vscode-errorForeground)",
|
|
}}></i>
|
|
<span>
|
|
Uh-oh, this could be a problem on Kodu's end. We've been alerted and
|
|
will resolve this ASAP. You can also{" "}
|
|
<a
|
|
href="https://discord.gg/claudedev"
|
|
style={{ color: "inherit", textDecoration: "underline" }}>
|
|
contact us on discord
|
|
</a>
|
|
.
|
|
</span>
|
|
</div>
|
|
)} */}
|
|
</>
|
|
)}
|
|
|
|
{isExpanded && (
|
|
<div style={{ marginTop: "10px" }}>
|
|
<CodeAccordian
|
|
code={JSON.parse(message.text || "{}").request}
|
|
language="markdown"
|
|
isExpanded={true}
|
|
onToggleExpand={onToggleExpand}
|
|
/>
|
|
</div>
|
|
)}
|
|
</>
|
|
)
|
|
case "api_req_finished":
|
|
return null // we should never see this message type
|
|
case "text":
|
|
return (
|
|
<div>
|
|
<Markdown markdown={message.text} />
|
|
</div>
|
|
)
|
|
case "user_feedback":
|
|
return (
|
|
<div
|
|
style={{
|
|
backgroundColor: "var(--vscode-badge-background)",
|
|
color: "var(--vscode-badge-foreground)",
|
|
borderRadius: "3px",
|
|
padding: "9px",
|
|
whiteSpace: "pre-line",
|
|
wordWrap: "break-word",
|
|
}}>
|
|
<span style={{ display: "block" }}>{highlightMentions(message.text)}</span>
|
|
{message.images && message.images.length > 0 && (
|
|
<Thumbnails images={message.images} style={{ marginTop: "8px" }} />
|
|
)}
|
|
</div>
|
|
)
|
|
case "user_feedback_diff":
|
|
const tool = JSON.parse(message.text || "{}") as ClaudeSayTool
|
|
return (
|
|
<div
|
|
style={{
|
|
marginTop: -10,
|
|
width: "100%",
|
|
}}>
|
|
<CodeAccordian
|
|
diff={tool.diff!}
|
|
isFeedback={true}
|
|
isExpanded={isExpanded}
|
|
onToggleExpand={onToggleExpand}
|
|
/>
|
|
</div>
|
|
)
|
|
case "inspect_site_result":
|
|
const logs = message.text || ""
|
|
const screenshot = message.images?.[0]
|
|
return (
|
|
<div
|
|
style={{
|
|
marginTop: -10,
|
|
width: "100%",
|
|
}}>
|
|
{screenshot && (
|
|
<img
|
|
src={screenshot}
|
|
alt="Inspect screenshot"
|
|
style={{
|
|
width: "calc(100% - 2px)",
|
|
height: "auto",
|
|
objectFit: "contain",
|
|
marginBottom: logs ? 7 : 0,
|
|
borderRadius: 3,
|
|
cursor: "pointer",
|
|
marginLeft: "1px",
|
|
}}
|
|
onClick={() => vscode.postMessage({ type: "openImage", text: screenshot })}
|
|
/>
|
|
)}
|
|
{logs && (
|
|
<CodeAccordian
|
|
code={logs}
|
|
language="shell"
|
|
isConsoleLogs={true}
|
|
isExpanded={isExpanded}
|
|
onToggleExpand={onToggleExpand}
|
|
/>
|
|
)}
|
|
</div>
|
|
)
|
|
case "error":
|
|
return (
|
|
<>
|
|
{title && (
|
|
<div style={headerStyle}>
|
|
{icon}
|
|
{title}
|
|
</div>
|
|
)}
|
|
<p style={{ ...pStyle, color: "var(--vscode-errorForeground)" }}>{message.text}</p>
|
|
</>
|
|
)
|
|
case "completion_result":
|
|
return (
|
|
<>
|
|
<div style={headerStyle}>
|
|
{icon}
|
|
{title}
|
|
</div>
|
|
<div style={{ color: "var(--vscode-charts-green)", paddingTop: 10 }}>
|
|
<Markdown markdown={message.text} />
|
|
</div>
|
|
</>
|
|
)
|
|
case "shell_integration_warning":
|
|
return (
|
|
<>
|
|
<div
|
|
style={{
|
|
display: "flex",
|
|
flexDirection: "column",
|
|
backgroundColor: "rgba(255, 191, 0, 0.1)",
|
|
padding: 8,
|
|
borderRadius: 3,
|
|
fontSize: 12,
|
|
}}>
|
|
<div style={{ display: "flex", alignItems: "center", marginBottom: 4 }}>
|
|
<i
|
|
className="codicon codicon-warning"
|
|
style={{
|
|
marginRight: 8,
|
|
fontSize: 18,
|
|
color: "#FFA500",
|
|
}}></i>
|
|
<span style={{ fontWeight: 500, color: "#FFA500" }}>
|
|
Shell Integration Unavailable
|
|
</span>
|
|
</div>
|
|
<div>
|
|
Claude won't be able to view the command's output. Please update VSCode (
|
|
<code>CMD/CTRL + Shift + P</code> → "Update") and make sure you're using a supported
|
|
shell: zsh, bash, fish, or PowerShell (<code>CMD/CTRL + Shift + P</code> →
|
|
"Terminal: Select Default Profile").{" "}
|
|
<a
|
|
href="https://github.com/saoudrizwan/claude-dev/wiki/Troubleshooting-%E2%80%90-Shell-Integration-Unavailable"
|
|
style={{ color: "inherit", textDecoration: "underline" }}>
|
|
Still having trouble?
|
|
</a>
|
|
</div>
|
|
</div>
|
|
</>
|
|
)
|
|
|
|
default:
|
|
return (
|
|
<>
|
|
{title && (
|
|
<div style={headerStyle}>
|
|
{icon}
|
|
{title}
|
|
</div>
|
|
)}
|
|
<div style={{ paddingTop: 10 }}>
|
|
<Markdown markdown={message.text} />
|
|
</div>
|
|
</>
|
|
)
|
|
}
|
|
case "ask":
|
|
switch (message.ask) {
|
|
case "mistake_limit_reached":
|
|
return (
|
|
<>
|
|
<div style={headerStyle}>
|
|
{icon}
|
|
{title}
|
|
</div>
|
|
<p style={{ ...pStyle, color: "var(--vscode-errorForeground)" }}>{message.text}</p>
|
|
</>
|
|
)
|
|
case "command":
|
|
const splitMessage = (text: string) => {
|
|
const outputIndex = text.indexOf(COMMAND_OUTPUT_STRING)
|
|
if (outputIndex === -1) {
|
|
return { command: text, output: "" }
|
|
}
|
|
return {
|
|
command: text.slice(0, outputIndex).trim(),
|
|
output: text
|
|
.slice(outputIndex + COMMAND_OUTPUT_STRING.length)
|
|
.trim()
|
|
.split("")
|
|
.map((char) => {
|
|
switch (char) {
|
|
case "\t":
|
|
return "→ "
|
|
case "\b":
|
|
return "⌫"
|
|
case "\f":
|
|
return "⏏"
|
|
case "\v":
|
|
return "⇳"
|
|
default:
|
|
return char
|
|
}
|
|
})
|
|
.join(""),
|
|
}
|
|
}
|
|
|
|
const { command, output } = splitMessage(message.text || "")
|
|
return (
|
|
<>
|
|
<div style={headerStyle}>
|
|
{icon}
|
|
{title}
|
|
</div>
|
|
{/* <Terminal
|
|
rawOutput={command + (output ? "\n" + output : "")}
|
|
shouldAllowInput={!!isCommandExecuting && output.length > 0}
|
|
/> */}
|
|
<div
|
|
style={{
|
|
borderRadius: 3,
|
|
border: "1px solid var(--vscode-editorGroup-border)",
|
|
overflow: "hidden",
|
|
backgroundColor: CODE_BLOCK_BG_COLOR,
|
|
}}>
|
|
<CodeBlock source={`${"```"}shell\n${command}\n${"```"}`} forceWrap={true} />
|
|
{output.length > 0 && (
|
|
<div style={{ width: "100%" }}>
|
|
<div
|
|
onClick={onToggleExpand}
|
|
style={{
|
|
display: "flex",
|
|
alignItems: "center",
|
|
gap: "4px",
|
|
width: "100%",
|
|
justifyContent: "flex-start",
|
|
cursor: "pointer",
|
|
padding: `2px 8px ${isExpanded ? 0 : 8}px 8px`,
|
|
}}>
|
|
<span
|
|
className={`codicon codicon-chevron-${
|
|
isExpanded ? "down" : "right"
|
|
}`}></span>
|
|
<span style={{ fontSize: "0.8em" }}>Command Output</span>
|
|
</div>
|
|
{isExpanded && <CodeBlock source={`${"```"}shell\n${output}\n${"```"}`} />}
|
|
</div>
|
|
)}
|
|
</div>
|
|
</>
|
|
)
|
|
case "completion_result":
|
|
if (message.text) {
|
|
return (
|
|
<div>
|
|
<div style={headerStyle}>
|
|
{icon}
|
|
{title}
|
|
</div>
|
|
<div style={{ color: "var(--vscode-charts-green)", paddingTop: 10 }}>
|
|
<Markdown markdown={message.text} />
|
|
</div>
|
|
</div>
|
|
)
|
|
} else {
|
|
return null // Don't render anything when we get a completion_result ask without text
|
|
}
|
|
case "followup":
|
|
return (
|
|
<>
|
|
{title && (
|
|
<div style={headerStyle}>
|
|
{icon}
|
|
{title}
|
|
</div>
|
|
)}
|
|
<div style={{ paddingTop: 10 }}>
|
|
<Markdown markdown={message.text} />
|
|
</div>
|
|
</>
|
|
)
|
|
default:
|
|
return null
|
|
}
|
|
}
|
|
}
|
|
|
|
const ProgressIndicator = () => (
|
|
<div
|
|
style={{
|
|
width: "16px",
|
|
height: "16px",
|
|
display: "flex",
|
|
alignItems: "center",
|
|
justifyContent: "center",
|
|
}}>
|
|
<div style={{ transform: "scale(0.55)", transformOrigin: "center" }}>
|
|
<VSCodeProgressRing />
|
|
</div>
|
|
</div>
|
|
)
|
|
|
|
const Markdown = memo(({ markdown }: { markdown?: string }) => {
|
|
const withoutThinkingTags = useMemo(() => {
|
|
if (!markdown) return ""
|
|
|
|
let processed = markdown
|
|
|
|
// Remove end substrings of <thinking or </thinking
|
|
processed = processed.replace(/<\/?t(?:h(?:i(?:n(?:k(?:i(?:n(?:g)?)?)?)?)?)?)?$/, "")
|
|
|
|
// Remove all instances of <thinking> (with optional line break after) and </thinking> (with optional line break before)
|
|
// Needs to be separate since we dont want to remove the line break before the first tag
|
|
processed = processed.replace(/<thinking>\s?/g, "")
|
|
processed = processed.replace(/\s?<\/thinking>/g, "")
|
|
|
|
return processed
|
|
}, [markdown])
|
|
|
|
return (
|
|
<div style={{ wordBreak: "break-word", overflowWrap: "anywhere", marginBottom: -15, marginTop: -15 }}>
|
|
<MarkdownBlock markdown={withoutThinkingTags} />
|
|
</div>
|
|
)
|
|
})
|