ChatView.tsx 48 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522
  1. import { forwardRef, useCallback, useEffect, useImperativeHandle, useMemo, useRef, useState } from "react"
  2. import { useDeepCompareEffect, useEvent, useMount } from "react-use"
  3. import debounce from "debounce"
  4. import { Virtuoso, type VirtuosoHandle } from "react-virtuoso"
  5. import removeMd from "remove-markdown"
  6. import { Trans } from "react-i18next"
  7. import { VSCodeButton } from "@vscode/webview-ui-toolkit/react"
  8. import useSound from "use-sound"
  9. import type { ClineAsk, ClineMessage } from "@roo-code/types"
  10. import { ClineSayBrowserAction, ClineSayTool, ExtensionMessage } from "@roo/ExtensionMessage"
  11. import { McpServer, McpTool } from "@roo/mcp"
  12. import { findLast } from "@roo/array"
  13. import { combineApiRequests } from "@roo/combineApiRequests"
  14. import { combineCommandSequences } from "@roo/combineCommandSequences"
  15. import { getApiMetrics } from "@roo/getApiMetrics"
  16. import { AudioType } from "@roo/WebviewMessage"
  17. import { getAllModes } from "@roo/modes"
  18. import { vscode } from "@src/utils/vscode"
  19. import { validateCommand } from "@src/utils/command-validation"
  20. import { buildDocLink } from "@src/utils/docLinks"
  21. import { useAppTranslation } from "@src/i18n/TranslationContext"
  22. import { useExtensionState } from "@src/context/ExtensionStateContext"
  23. import { useSelectedModel } from "@src/components/ui/hooks/useSelectedModel"
  24. import RooHero from "@src/components/welcome/RooHero"
  25. import RooTips from "@src/components/welcome/RooTips"
  26. import TelemetryBanner from "../common/TelemetryBanner"
  27. import { useTaskSearch } from "../history/useTaskSearch"
  28. import HistoryPreview from "../history/HistoryPreview"
  29. import Announcement from "./Announcement"
  30. import BrowserSessionRow from "./BrowserSessionRow"
  31. import ChatRow from "./ChatRow"
  32. import ChatTextArea from "./ChatTextArea"
  33. import TaskHeader from "./TaskHeader"
  34. import AutoApproveMenu from "./AutoApproveMenu"
  35. import SystemPromptWarning from "./SystemPromptWarning"
  36. import { CheckpointWarning } from "./CheckpointWarning"
  37. import { LRUCache } from "lru-cache"
  38. export interface ChatViewProps {
  39. isHidden: boolean
  40. showAnnouncement: boolean
  41. hideAnnouncement: () => void
  42. }
  43. export interface ChatViewRef {
  44. acceptInput: () => void
  45. }
  46. export const MAX_IMAGES_PER_MESSAGE = 20 // Anthropic limits to 20 images
  47. const isMac = navigator.platform.toUpperCase().indexOf("MAC") >= 0
  48. const ChatViewComponent: React.ForwardRefRenderFunction<ChatViewRef, ChatViewProps> = (
  49. { isHidden, showAnnouncement, hideAnnouncement },
  50. ref,
  51. ) => {
  52. const [audioBaseUri] = useState(() => {
  53. const w = window as any
  54. return w.AUDIO_BASE_URI || ""
  55. })
  56. const { t } = useAppTranslation()
  57. const modeShortcutText = `${isMac ? "⌘" : "Ctrl"} + . ${t("chat:forNextMode")}`
  58. const {
  59. clineMessages: messages,
  60. currentTaskItem,
  61. taskHistory,
  62. apiConfiguration,
  63. mcpServers,
  64. alwaysAllowBrowser,
  65. alwaysAllowReadOnly,
  66. alwaysAllowReadOnlyOutsideWorkspace,
  67. alwaysAllowWrite,
  68. alwaysAllowWriteOutsideWorkspace,
  69. alwaysAllowExecute,
  70. alwaysAllowMcp,
  71. allowedCommands,
  72. writeDelayMs,
  73. mode,
  74. setMode,
  75. autoApprovalEnabled,
  76. alwaysAllowModeSwitch,
  77. alwaysAllowSubtasks,
  78. customModes,
  79. telemetrySetting,
  80. hasSystemPromptOverride,
  81. historyPreviewCollapsed, // Added historyPreviewCollapsed
  82. soundEnabled,
  83. soundVolume,
  84. } = useExtensionState()
  85. const messagesRef = useRef(messages)
  86. useEffect(() => {
  87. messagesRef.current = messages
  88. }, [messages])
  89. const { tasks } = useTaskSearch()
  90. // Initialize expanded state based on the persisted setting (default to expanded if undefined)
  91. const [isExpanded, setIsExpanded] = useState(
  92. historyPreviewCollapsed === undefined ? true : !historyPreviewCollapsed,
  93. )
  94. const toggleExpanded = useCallback(() => {
  95. const newState = !isExpanded
  96. setIsExpanded(newState)
  97. // Send message to extension to persist the new collapsed state
  98. vscode.postMessage({ type: "setHistoryPreviewCollapsed", bool: !newState })
  99. }, [isExpanded])
  100. // Leaving this less safe version here since if the first message is not a
  101. // task, then the extension is in a bad state and needs to be debugged (see
  102. // Cline.abort).
  103. const task = useMemo(() => messages.at(0), [messages])
  104. const modifiedMessages = useMemo(() => combineApiRequests(combineCommandSequences(messages.slice(1))), [messages])
  105. // Has to be after api_req_finished are all reduced into api_req_started messages.
  106. const apiMetrics = useMemo(() => getApiMetrics(modifiedMessages), [modifiedMessages])
  107. const [inputValue, setInputValue] = useState("")
  108. const textAreaRef = useRef<HTMLTextAreaElement>(null)
  109. const [sendingDisabled, setSendingDisabled] = useState(false)
  110. const [selectedImages, setSelectedImages] = useState<string[]>([])
  111. // we need to hold on to the ask because useEffect > lastMessage will always let us know when an ask comes in and handle it, but by the time handleMessage is called, the last message might not be the ask anymore (it could be a say that followed)
  112. const [clineAsk, setClineAsk] = useState<ClineAsk | undefined>(undefined)
  113. const [enableButtons, setEnableButtons] = useState<boolean>(false)
  114. const [primaryButtonText, setPrimaryButtonText] = useState<string | undefined>(undefined)
  115. const [secondaryButtonText, setSecondaryButtonText] = useState<string | undefined>(undefined)
  116. const [didClickCancel, setDidClickCancel] = useState(false)
  117. const virtuosoRef = useRef<VirtuosoHandle>(null)
  118. const [expandedRows, setExpandedRows] = useState<Record<number, boolean>>({})
  119. const prevExpandedRowsRef = useRef<Record<number, boolean>>()
  120. const scrollContainerRef = useRef<HTMLDivElement>(null)
  121. const disableAutoScrollRef = useRef(false)
  122. const [showScrollToBottom, setShowScrollToBottom] = useState(false)
  123. const [isAtBottom, setIsAtBottom] = useState(false)
  124. const lastTtsRef = useRef<string>("")
  125. const [wasStreaming, setWasStreaming] = useState<boolean>(false)
  126. const [showCheckpointWarning, setShowCheckpointWarning] = useState<boolean>(false)
  127. const [isCondensing, setIsCondensing] = useState<boolean>(false)
  128. const everVisibleMessagesTsRef = useRef<LRUCache<number, boolean>>(
  129. new LRUCache({
  130. max: 250,
  131. ttl: 1000 * 60 * 15, // 15 minutes TTL for long-running tasks
  132. }),
  133. )
  134. const clineAskRef = useRef(clineAsk)
  135. useEffect(() => {
  136. clineAskRef.current = clineAsk
  137. }, [clineAsk])
  138. // UI layout depends on the last 2 messages
  139. // (since it relies on the content of these messages, we are deep comparing. i.e. the button state after hitting button sets enableButtons to false, and this effect otherwise would have to true again even if messages didn't change
  140. const lastMessage = useMemo(() => messages.at(-1), [messages])
  141. const secondLastMessage = useMemo(() => messages.at(-2), [messages])
  142. // Setup sound hooks with use-sound
  143. const volume = typeof soundVolume === "number" ? soundVolume : 0.5
  144. const soundConfig = {
  145. volume,
  146. // useSound expects 'disabled' property, not 'soundEnabled'
  147. soundEnabled,
  148. }
  149. const getAudioUrl = (path: string) => {
  150. return `${audioBaseUri}/${path}`
  151. }
  152. // Use the getAudioUrl helper function
  153. const [playNotification] = useSound(getAudioUrl("notification.wav"), soundConfig)
  154. const [playCelebration] = useSound(getAudioUrl("celebration.wav"), soundConfig)
  155. const [playProgressLoop] = useSound(getAudioUrl("progress_loop.wav"), soundConfig)
  156. function playSound(audioType: AudioType) {
  157. // Play the appropriate sound based on type
  158. // The disabled state is handled by the useSound hook configuration
  159. switch (audioType) {
  160. case "notification":
  161. playNotification()
  162. break
  163. case "celebration":
  164. playCelebration()
  165. break
  166. case "progress_loop":
  167. playProgressLoop()
  168. break
  169. default:
  170. console.warn(`Unknown audio type: ${audioType}`)
  171. }
  172. }
  173. function playTts(text: string) {
  174. vscode.postMessage({ type: "playTts", text })
  175. }
  176. useDeepCompareEffect(() => {
  177. // if last message is an ask, show user ask UI
  178. // if user finished a task, then start a new task with a new conversation history since in this moment that the extension is waiting for user response, the user could close the extension and the conversation history would be lost.
  179. // basically as long as a task is active, the conversation history will be persisted
  180. if (lastMessage) {
  181. switch (lastMessage.type) {
  182. case "ask":
  183. const isPartial = lastMessage.partial === true
  184. switch (lastMessage.ask) {
  185. case "api_req_failed":
  186. playSound("progress_loop")
  187. setSendingDisabled(true)
  188. setClineAsk("api_req_failed")
  189. setEnableButtons(true)
  190. setPrimaryButtonText(t("chat:retry.title"))
  191. setSecondaryButtonText(t("chat:startNewTask.title"))
  192. break
  193. case "mistake_limit_reached":
  194. playSound("progress_loop")
  195. setSendingDisabled(false)
  196. setClineAsk("mistake_limit_reached")
  197. setEnableButtons(true)
  198. setPrimaryButtonText(t("chat:proceedAnyways.title"))
  199. setSecondaryButtonText(t("chat:startNewTask.title"))
  200. break
  201. case "followup":
  202. if (!isPartial) {
  203. playSound("notification")
  204. }
  205. setSendingDisabled(isPartial)
  206. setClineAsk("followup")
  207. // setting enable buttons to `false` would trigger a focus grab when
  208. // the text area is enabled which is undesirable.
  209. // We have no buttons for this tool, so no problem having them "enabled"
  210. // to workaround this issue. See #1358.
  211. setEnableButtons(true)
  212. setPrimaryButtonText(undefined)
  213. setSecondaryButtonText(undefined)
  214. break
  215. case "tool":
  216. if (!isAutoApproved(lastMessage) && !isPartial) {
  217. playSound("notification")
  218. }
  219. setSendingDisabled(isPartial)
  220. setClineAsk("tool")
  221. setEnableButtons(!isPartial)
  222. const tool = JSON.parse(lastMessage.text || "{}") as ClineSayTool
  223. switch (tool.tool) {
  224. case "editedExistingFile":
  225. case "appliedDiff":
  226. case "newFileCreated":
  227. case "insertContent":
  228. setPrimaryButtonText(t("chat:save.title"))
  229. setSecondaryButtonText(t("chat:reject.title"))
  230. break
  231. case "finishTask":
  232. setPrimaryButtonText(t("chat:completeSubtaskAndReturn"))
  233. setSecondaryButtonText(undefined)
  234. break
  235. default:
  236. setPrimaryButtonText(t("chat:approve.title"))
  237. setSecondaryButtonText(t("chat:reject.title"))
  238. break
  239. }
  240. break
  241. case "browser_action_launch":
  242. if (!isAutoApproved(lastMessage) && !isPartial) {
  243. playSound("notification")
  244. }
  245. setSendingDisabled(isPartial)
  246. setClineAsk("browser_action_launch")
  247. setEnableButtons(!isPartial)
  248. setPrimaryButtonText(t("chat:approve.title"))
  249. setSecondaryButtonText(t("chat:reject.title"))
  250. break
  251. case "command":
  252. if (!isAutoApproved(lastMessage) && !isPartial) {
  253. playSound("notification")
  254. }
  255. setSendingDisabled(isPartial)
  256. setClineAsk("command")
  257. setEnableButtons(!isPartial)
  258. setPrimaryButtonText(t("chat:runCommand.title"))
  259. setSecondaryButtonText(t("chat:reject.title"))
  260. break
  261. case "command_output":
  262. setSendingDisabled(false)
  263. setClineAsk("command_output")
  264. setEnableButtons(true)
  265. setPrimaryButtonText(t("chat:proceedWhileRunning.title"))
  266. setSecondaryButtonText(t("chat:killCommand.title"))
  267. break
  268. case "use_mcp_server":
  269. if (!isAutoApproved(lastMessage) && !isPartial) {
  270. playSound("notification")
  271. }
  272. setSendingDisabled(isPartial)
  273. setClineAsk("use_mcp_server")
  274. setEnableButtons(!isPartial)
  275. setPrimaryButtonText(t("chat:approve.title"))
  276. setSecondaryButtonText(t("chat:reject.title"))
  277. break
  278. case "completion_result":
  279. // extension waiting for feedback. but we can just present a new task button
  280. if (!isPartial) {
  281. playSound("celebration")
  282. }
  283. setSendingDisabled(isPartial)
  284. setClineAsk("completion_result")
  285. setEnableButtons(!isPartial)
  286. setPrimaryButtonText(t("chat:startNewTask.title"))
  287. setSecondaryButtonText(undefined)
  288. break
  289. case "resume_task":
  290. if (!isAutoApproved(lastMessage) && !isPartial) {
  291. playSound("notification")
  292. }
  293. setSendingDisabled(false)
  294. setClineAsk("resume_task")
  295. setEnableButtons(true)
  296. setPrimaryButtonText(t("chat:resumeTask.title"))
  297. setSecondaryButtonText(t("chat:terminate.title"))
  298. setDidClickCancel(false) // special case where we reset the cancel button state
  299. break
  300. case "resume_completed_task":
  301. if (!isPartial) {
  302. playSound("celebration")
  303. }
  304. setSendingDisabled(false)
  305. setClineAsk("resume_completed_task")
  306. setEnableButtons(true)
  307. setPrimaryButtonText(t("chat:startNewTask.title"))
  308. setSecondaryButtonText(undefined)
  309. setDidClickCancel(false)
  310. break
  311. }
  312. break
  313. case "say":
  314. // Don't want to reset since there could be a "say" after
  315. // an "ask" while ask is waiting for response.
  316. switch (lastMessage.say) {
  317. case "api_req_retry_delayed":
  318. setSendingDisabled(true)
  319. break
  320. case "api_req_started":
  321. if (secondLastMessage?.ask === "command_output") {
  322. setSendingDisabled(true)
  323. setSelectedImages([])
  324. setClineAsk(undefined)
  325. setEnableButtons(false)
  326. }
  327. break
  328. case "api_req_finished":
  329. case "error":
  330. case "text":
  331. case "browser_action":
  332. case "browser_action_result":
  333. case "command_output":
  334. case "mcp_server_request_started":
  335. case "mcp_server_response":
  336. case "completion_result":
  337. break
  338. }
  339. break
  340. }
  341. }
  342. }, [lastMessage, secondLastMessage])
  343. useEffect(() => {
  344. if (messages.length === 0) {
  345. setSendingDisabled(false)
  346. setClineAsk(undefined)
  347. setEnableButtons(false)
  348. setPrimaryButtonText(undefined)
  349. setSecondaryButtonText(undefined)
  350. }
  351. }, [messages.length])
  352. useEffect(() => {
  353. setExpandedRows({})
  354. everVisibleMessagesTsRef.current.clear() // Clear for new task
  355. }, [task?.ts])
  356. useEffect(() => () => everVisibleMessagesTsRef.current.clear(), [])
  357. useEffect(() => {
  358. const prev = prevExpandedRowsRef.current
  359. let wasAnyRowExpandedByUser = false
  360. if (prev) {
  361. // Check if any row transitioned from false/undefined to true
  362. for (const [tsKey, isExpanded] of Object.entries(expandedRows)) {
  363. const ts = Number(tsKey)
  364. if (isExpanded && !(prev[ts] ?? false)) {
  365. wasAnyRowExpandedByUser = true
  366. break
  367. }
  368. }
  369. }
  370. if (wasAnyRowExpandedByUser) {
  371. disableAutoScrollRef.current = true
  372. }
  373. prevExpandedRowsRef.current = expandedRows // Store current state for next comparison
  374. }, [expandedRows])
  375. const isStreaming = useMemo(() => {
  376. // Checking clineAsk isn't enough since messages effect may be called
  377. // again for a tool for example, set clineAsk to its value, and if the
  378. // next message is not an ask then it doesn't reset. This is likely due
  379. // to how much more often we're updating messages as compared to before,
  380. // and should be resolved with optimizations as it's likely a rendering
  381. // bug. But as a final guard for now, the cancel button will show if the
  382. // last message is not an ask.
  383. const isLastAsk = !!modifiedMessages.at(-1)?.ask
  384. const isToolCurrentlyAsking =
  385. isLastAsk && clineAsk !== undefined && enableButtons && primaryButtonText !== undefined
  386. if (isToolCurrentlyAsking) {
  387. return false
  388. }
  389. const isLastMessagePartial = modifiedMessages.at(-1)?.partial === true
  390. if (isLastMessagePartial) {
  391. return true
  392. } else {
  393. const lastApiReqStarted = findLast(modifiedMessages, (message) => message.say === "api_req_started")
  394. if (
  395. lastApiReqStarted &&
  396. lastApiReqStarted.text !== null &&
  397. lastApiReqStarted.text !== undefined &&
  398. lastApiReqStarted.say === "api_req_started"
  399. ) {
  400. const cost = JSON.parse(lastApiReqStarted.text).cost
  401. if (cost === undefined) {
  402. return true // API request has not finished yet.
  403. }
  404. }
  405. }
  406. return false
  407. }, [modifiedMessages, clineAsk, enableButtons, primaryButtonText])
  408. const handleChatReset = useCallback(() => {
  409. // Only reset message-specific state, preserving mode.
  410. setInputValue("")
  411. setSendingDisabled(true)
  412. setSelectedImages([])
  413. setClineAsk(undefined)
  414. setEnableButtons(false)
  415. // Do not reset mode here as it should persist.
  416. // setPrimaryButtonText(undefined)
  417. // setSecondaryButtonText(undefined)
  418. disableAutoScrollRef.current = false
  419. }, [])
  420. const handleSendMessage = useCallback(
  421. (text: string, images: string[]) => {
  422. text = text.trim()
  423. if (text || images.length > 0) {
  424. if (messagesRef.current.length === 0) {
  425. vscode.postMessage({ type: "newTask", text, images })
  426. } else if (clineAskRef.current) {
  427. // Use clineAskRef.current
  428. switch (
  429. clineAskRef.current // Use clineAskRef.current
  430. ) {
  431. case "followup":
  432. case "tool":
  433. case "browser_action_launch":
  434. case "command": // User can provide feedback to a tool or command use.
  435. case "command_output": // User can send input to command stdin.
  436. case "use_mcp_server":
  437. case "completion_result": // If this happens then the user has feedback for the completion result.
  438. case "resume_task":
  439. case "resume_completed_task":
  440. case "mistake_limit_reached":
  441. vscode.postMessage({ type: "askResponse", askResponse: "messageResponse", text, images })
  442. break
  443. // There is no other case that a textfield should be enabled.
  444. }
  445. }
  446. handleChatReset()
  447. }
  448. },
  449. [handleChatReset], // messagesRef and clineAskRef are stable
  450. )
  451. const handleSetChatBoxMessage = useCallback(
  452. (text: string, images: string[]) => {
  453. // Avoid nested template literals by breaking down the logic
  454. let newValue = text
  455. if (inputValue !== "") {
  456. newValue = inputValue + " " + text
  457. }
  458. setInputValue(newValue)
  459. setSelectedImages([...selectedImages, ...images])
  460. },
  461. [inputValue, selectedImages],
  462. )
  463. const startNewTask = useCallback(() => vscode.postMessage({ type: "clearTask" }), [])
  464. // This logic depends on the useEffect[messages] above to set clineAsk,
  465. // after which buttons are shown and we then send an askResponse to the
  466. // extension.
  467. const handlePrimaryButtonClick = useCallback(
  468. (text?: string, images?: string[]) => {
  469. const trimmedInput = text?.trim()
  470. switch (clineAsk) {
  471. case "api_req_failed":
  472. case "command":
  473. case "tool":
  474. case "browser_action_launch":
  475. case "use_mcp_server":
  476. case "resume_task":
  477. case "mistake_limit_reached":
  478. // Only send text/images if they exist
  479. if (trimmedInput || (images && images.length > 0)) {
  480. vscode.postMessage({
  481. type: "askResponse",
  482. askResponse: "yesButtonClicked",
  483. text: trimmedInput,
  484. images: images,
  485. })
  486. } else {
  487. vscode.postMessage({ type: "askResponse", askResponse: "yesButtonClicked" })
  488. }
  489. // Clear input state after sending
  490. setInputValue("")
  491. setSelectedImages([])
  492. break
  493. case "completion_result":
  494. case "resume_completed_task":
  495. // Waiting for feedback, but we can just present a new task button
  496. startNewTask()
  497. break
  498. case "command_output":
  499. vscode.postMessage({ type: "terminalOperation", terminalOperation: "continue" })
  500. break
  501. }
  502. setSendingDisabled(true)
  503. setClineAsk(undefined)
  504. setEnableButtons(false)
  505. },
  506. [clineAsk, startNewTask],
  507. )
  508. const handleSecondaryButtonClick = useCallback(
  509. (text?: string, images?: string[]) => {
  510. const trimmedInput = text?.trim()
  511. if (isStreaming) {
  512. vscode.postMessage({ type: "cancelTask" })
  513. setDidClickCancel(true)
  514. return
  515. }
  516. switch (clineAsk) {
  517. case "api_req_failed":
  518. case "mistake_limit_reached":
  519. case "resume_task":
  520. startNewTask()
  521. break
  522. case "command":
  523. case "tool":
  524. case "browser_action_launch":
  525. case "use_mcp_server":
  526. // Only send text/images if they exist
  527. if (trimmedInput || (images && images.length > 0)) {
  528. vscode.postMessage({
  529. type: "askResponse",
  530. askResponse: "noButtonClicked",
  531. text: trimmedInput,
  532. images: images,
  533. })
  534. } else {
  535. // Responds to the API with a "This operation failed" and lets it try again
  536. vscode.postMessage({ type: "askResponse", askResponse: "noButtonClicked" })
  537. }
  538. // Clear input state after sending
  539. setInputValue("")
  540. setSelectedImages([])
  541. break
  542. case "command_output":
  543. vscode.postMessage({ type: "terminalOperation", terminalOperation: "abort" })
  544. break
  545. }
  546. setSendingDisabled(true)
  547. setClineAsk(undefined)
  548. setEnableButtons(false)
  549. },
  550. [clineAsk, startNewTask, isStreaming],
  551. )
  552. const handleTaskCloseButtonClick = useCallback(() => startNewTask(), [startNewTask])
  553. const { info: model } = useSelectedModel(apiConfiguration)
  554. const selectImages = useCallback(() => vscode.postMessage({ type: "selectImages" }), [])
  555. const shouldDisableImages =
  556. !model?.supportsImages || sendingDisabled || selectedImages.length >= MAX_IMAGES_PER_MESSAGE
  557. const handleMessage = useCallback(
  558. (e: MessageEvent) => {
  559. const message: ExtensionMessage = e.data
  560. switch (message.type) {
  561. case "action":
  562. switch (message.action!) {
  563. case "didBecomeVisible":
  564. if (!isHidden && !sendingDisabled && !enableButtons) {
  565. textAreaRef.current?.focus()
  566. }
  567. break
  568. case "focusInput":
  569. textAreaRef.current?.focus()
  570. break
  571. }
  572. break
  573. case "selectedImages":
  574. const newImages = message.images ?? []
  575. if (newImages.length > 0) {
  576. setSelectedImages((prevImages) =>
  577. [...prevImages, ...newImages].slice(0, MAX_IMAGES_PER_MESSAGE),
  578. )
  579. }
  580. break
  581. case "invoke":
  582. switch (message.invoke!) {
  583. case "newChat":
  584. handleChatReset()
  585. break
  586. case "sendMessage":
  587. handleSendMessage(message.text ?? "", message.images ?? [])
  588. break
  589. case "setChatBoxMessage":
  590. handleSetChatBoxMessage(message.text ?? "", message.images ?? [])
  591. break
  592. case "primaryButtonClick":
  593. handlePrimaryButtonClick(message.text ?? "", message.images ?? [])
  594. break
  595. case "secondaryButtonClick":
  596. handleSecondaryButtonClick(message.text ?? "", message.images ?? [])
  597. break
  598. }
  599. break
  600. case "condenseTaskContextResponse":
  601. if (message.text && message.text === currentTaskItem?.id) {
  602. if (isCondensing && sendingDisabled) {
  603. setSendingDisabled(false)
  604. }
  605. setIsCondensing(false)
  606. }
  607. break
  608. }
  609. // textAreaRef.current is not explicitly required here since React
  610. // guarantees that ref will be stable across re-renders, and we're
  611. // not using its value but its reference.
  612. },
  613. [
  614. isCondensing,
  615. isHidden,
  616. sendingDisabled,
  617. enableButtons,
  618. currentTaskItem,
  619. handleChatReset,
  620. handleSendMessage,
  621. handleSetChatBoxMessage,
  622. handlePrimaryButtonClick,
  623. handleSecondaryButtonClick,
  624. ],
  625. )
  626. useEvent("message", handleMessage)
  627. // NOTE: the VSCode window needs to be focused for this to work.
  628. useMount(() => textAreaRef.current?.focus())
  629. useEffect(() => {
  630. const timer = setTimeout(() => {
  631. if (!isHidden && !sendingDisabled && !enableButtons) {
  632. textAreaRef.current?.focus()
  633. }
  634. }, 50)
  635. return () => {
  636. clearTimeout(timer)
  637. }
  638. }, [isHidden, sendingDisabled, enableButtons])
  639. const visibleMessages = useMemo(() => {
  640. const newVisibleMessages = modifiedMessages.filter((message) => {
  641. if (everVisibleMessagesTsRef.current.has(message.ts)) {
  642. // If it was ever visible, and it's not one of the types that should always be hidden once processed, keep it.
  643. // This helps prevent flickering for messages like 'api_req_retry_delayed' if they are no longer the absolute last.
  644. const alwaysHiddenOnceProcessedAsk: ClineAsk[] = [
  645. "api_req_failed",
  646. "resume_task",
  647. "resume_completed_task",
  648. ]
  649. const alwaysHiddenOnceProcessedSay = [
  650. "api_req_finished",
  651. "api_req_retried",
  652. "api_req_deleted",
  653. "mcp_server_request_started",
  654. ]
  655. if (message.ask && alwaysHiddenOnceProcessedAsk.includes(message.ask)) return false
  656. if (message.say && alwaysHiddenOnceProcessedSay.includes(message.say)) return false
  657. // Also, re-evaluate empty text messages if they were previously visible but now empty (e.g. partial stream ended)
  658. if (message.say === "text" && (message.text ?? "") === "" && (message.images?.length ?? 0) === 0) {
  659. return false
  660. }
  661. return true
  662. }
  663. // Original filter logic
  664. switch (message.ask) {
  665. case "completion_result":
  666. if (message.text === "") return false
  667. break
  668. case "api_req_failed":
  669. case "resume_task":
  670. case "resume_completed_task":
  671. return false
  672. }
  673. switch (message.say) {
  674. case "api_req_finished":
  675. case "api_req_retried":
  676. case "api_req_deleted":
  677. return false
  678. case "api_req_retry_delayed":
  679. const last1 = modifiedMessages.at(-1)
  680. const last2 = modifiedMessages.at(-2)
  681. if (last1?.ask === "resume_task" && last2 === message) {
  682. // This specific sequence should be visible
  683. } else if (message !== last1) {
  684. // If not the specific sequence above, and not the last message, hide it.
  685. return false
  686. }
  687. break
  688. case "text":
  689. if ((message.text ?? "") === "" && (message.images?.length ?? 0) === 0) return false
  690. break
  691. case "mcp_server_request_started":
  692. return false
  693. }
  694. return true
  695. })
  696. // Update the set of ever-visible messages (LRUCache automatically handles cleanup)
  697. newVisibleMessages.forEach((msg) => everVisibleMessagesTsRef.current.set(msg.ts, true))
  698. return newVisibleMessages
  699. }, [modifiedMessages])
  700. const isReadOnlyToolAction = useCallback((message: ClineMessage | undefined) => {
  701. if (message?.type === "ask") {
  702. if (!message.text) {
  703. return true
  704. }
  705. const tool = JSON.parse(message.text)
  706. return [
  707. "readFile",
  708. "listFiles",
  709. "listFilesTopLevel",
  710. "listFilesRecursive",
  711. "listCodeDefinitionNames",
  712. "searchFiles",
  713. "codebaseSearch",
  714. ].includes(tool.tool)
  715. }
  716. return false
  717. }, [])
  718. const isWriteToolAction = useCallback((message: ClineMessage | undefined) => {
  719. if (message?.type === "ask") {
  720. if (!message.text) {
  721. return true
  722. }
  723. const tool = JSON.parse(message.text)
  724. return [
  725. "editedExistingFile",
  726. "appliedDiff",
  727. "newFileCreated",
  728. "searchAndReplace",
  729. "insertContent",
  730. ].includes(tool.tool)
  731. }
  732. return false
  733. }, [])
  734. const isMcpToolAlwaysAllowed = useCallback(
  735. (message: ClineMessage | undefined) => {
  736. if (message?.type === "ask" && message.ask === "use_mcp_server") {
  737. if (!message.text) {
  738. return true
  739. }
  740. const mcpServerUse = JSON.parse(message.text) as { type: string; serverName: string; toolName: string }
  741. if (mcpServerUse.type === "use_mcp_tool") {
  742. const server = mcpServers?.find((s: McpServer) => s.name === mcpServerUse.serverName)
  743. const tool = server?.tools?.find((t: McpTool) => t.name === mcpServerUse.toolName)
  744. return tool?.alwaysAllow || false
  745. }
  746. }
  747. return false
  748. },
  749. [mcpServers],
  750. )
  751. // Check if a command message is allowed.
  752. const isAllowedCommand = useCallback(
  753. (message: ClineMessage | undefined): boolean => {
  754. if (message?.type !== "ask") return false
  755. return validateCommand(message.text || "", allowedCommands || [])
  756. },
  757. [allowedCommands],
  758. )
  759. const isAutoApproved = useCallback(
  760. (message: ClineMessage | undefined) => {
  761. if (!autoApprovalEnabled || !message || message.type !== "ask") {
  762. return false
  763. }
  764. if (message.ask === "browser_action_launch") {
  765. return alwaysAllowBrowser
  766. }
  767. if (message.ask === "use_mcp_server") {
  768. return alwaysAllowMcp && isMcpToolAlwaysAllowed(message)
  769. }
  770. if (message.ask === "command") {
  771. return alwaysAllowExecute && isAllowedCommand(message)
  772. }
  773. // For read/write operations, check if it's outside workspace and if
  774. // we have permission for that.
  775. if (message.ask === "tool") {
  776. let tool: any = {}
  777. try {
  778. tool = JSON.parse(message.text || "{}")
  779. } catch (error) {
  780. console.error("Failed to parse tool:", error)
  781. }
  782. if (!tool) {
  783. return false
  784. }
  785. if (tool?.tool === "fetchInstructions") {
  786. if (tool.content === "create_mode") {
  787. return alwaysAllowModeSwitch
  788. }
  789. if (tool.content === "create_mcp_server") {
  790. return alwaysAllowMcp
  791. }
  792. }
  793. if (tool?.tool === "switchMode") {
  794. return alwaysAllowModeSwitch
  795. }
  796. if (["newTask", "finishTask"].includes(tool?.tool)) {
  797. return alwaysAllowSubtasks
  798. }
  799. const isOutsideWorkspace = !!tool.isOutsideWorkspace
  800. if (isReadOnlyToolAction(message)) {
  801. return alwaysAllowReadOnly && (!isOutsideWorkspace || alwaysAllowReadOnlyOutsideWorkspace)
  802. }
  803. if (isWriteToolAction(message)) {
  804. return alwaysAllowWrite && (!isOutsideWorkspace || alwaysAllowWriteOutsideWorkspace)
  805. }
  806. }
  807. return false
  808. },
  809. [
  810. autoApprovalEnabled,
  811. alwaysAllowBrowser,
  812. alwaysAllowReadOnly,
  813. alwaysAllowReadOnlyOutsideWorkspace,
  814. isReadOnlyToolAction,
  815. alwaysAllowWrite,
  816. alwaysAllowWriteOutsideWorkspace,
  817. isWriteToolAction,
  818. alwaysAllowExecute,
  819. isAllowedCommand,
  820. alwaysAllowMcp,
  821. isMcpToolAlwaysAllowed,
  822. alwaysAllowModeSwitch,
  823. alwaysAllowSubtasks,
  824. ],
  825. )
  826. useEffect(() => {
  827. // This ensures the first message is not read, future user messages are
  828. // labeled as `user_feedback`.
  829. if (lastMessage && messages.length > 1) {
  830. if (
  831. lastMessage.text && // has text
  832. (lastMessage.say === "text" || lastMessage.say === "completion_result") && // is a text message
  833. !lastMessage.partial && // not a partial message
  834. !lastMessage.text.startsWith("{") // not a json object
  835. ) {
  836. let text = lastMessage?.text || ""
  837. const mermaidRegex = /```mermaid[\s\S]*?```/g
  838. // remove mermaid diagrams from text
  839. text = text.replace(mermaidRegex, "")
  840. // remove markdown from text
  841. text = removeMd(text)
  842. // ensure message is not a duplicate of last read message
  843. if (text !== lastTtsRef.current) {
  844. try {
  845. playTts(text)
  846. lastTtsRef.current = text
  847. } catch (error) {
  848. console.error("Failed to execute text-to-speech:", error)
  849. }
  850. }
  851. }
  852. }
  853. // Update previous value.
  854. setWasStreaming(isStreaming)
  855. }, [isStreaming, lastMessage, wasStreaming, isAutoApproved, messages.length])
  856. const isBrowserSessionMessage = (message: ClineMessage): boolean => {
  857. // Which of visible messages are browser session messages, see above.
  858. if (message.type === "ask") {
  859. return ["browser_action_launch"].includes(message.ask!)
  860. }
  861. if (message.type === "say") {
  862. return ["api_req_started", "text", "browser_action", "browser_action_result"].includes(message.say!)
  863. }
  864. return false
  865. }
  866. const groupedMessages = useMemo(() => {
  867. const result: (ClineMessage | ClineMessage[])[] = []
  868. let currentGroup: ClineMessage[] = []
  869. let isInBrowserSession = false
  870. const endBrowserSession = () => {
  871. if (currentGroup.length > 0) {
  872. result.push([...currentGroup])
  873. currentGroup = []
  874. isInBrowserSession = false
  875. }
  876. }
  877. visibleMessages.forEach((message) => {
  878. if (message.ask === "browser_action_launch") {
  879. // Complete existing browser session if any.
  880. endBrowserSession()
  881. // Start new.
  882. isInBrowserSession = true
  883. currentGroup.push(message)
  884. } else if (isInBrowserSession) {
  885. // End session if `api_req_started` is cancelled.
  886. if (message.say === "api_req_started") {
  887. // Get last `api_req_started` in currentGroup to check if
  888. // it's cancelled. If it is then this api req is not part
  889. // of the current browser session.
  890. const lastApiReqStarted = [...currentGroup].reverse().find((m) => m.say === "api_req_started")
  891. if (lastApiReqStarted?.text !== null && lastApiReqStarted?.text !== undefined) {
  892. const info = JSON.parse(lastApiReqStarted.text)
  893. const isCancelled = info.cancelReason !== null && info.cancelReason !== undefined
  894. if (isCancelled) {
  895. endBrowserSession()
  896. result.push(message)
  897. return
  898. }
  899. }
  900. }
  901. if (isBrowserSessionMessage(message)) {
  902. currentGroup.push(message)
  903. // Check if this is a close action
  904. if (message.say === "browser_action") {
  905. const browserAction = JSON.parse(message.text || "{}") as ClineSayBrowserAction
  906. if (browserAction.action === "close") {
  907. endBrowserSession()
  908. }
  909. }
  910. } else {
  911. // complete existing browser session if any
  912. endBrowserSession()
  913. result.push(message)
  914. }
  915. } else {
  916. result.push(message)
  917. }
  918. })
  919. // Handle case where browser session is the last group
  920. if (currentGroup.length > 0) {
  921. result.push([...currentGroup])
  922. }
  923. if (isCondensing) {
  924. // Show indicator after clicking condense button
  925. result.push({
  926. type: "say",
  927. say: "condense_context",
  928. ts: Date.now(),
  929. partial: true,
  930. })
  931. }
  932. return result
  933. }, [isCondensing, visibleMessages])
  934. // scrolling
  935. const scrollToBottomSmooth = useMemo(
  936. () =>
  937. debounce(() => virtuosoRef.current?.scrollTo({ top: Number.MAX_SAFE_INTEGER, behavior: "smooth" }), 10, {
  938. immediate: true,
  939. }),
  940. [],
  941. )
  942. const scrollToBottomAuto = useCallback(() => {
  943. virtuosoRef.current?.scrollTo({
  944. top: Number.MAX_SAFE_INTEGER,
  945. behavior: "auto", // Instant causes crash.
  946. })
  947. }, [])
  948. const handleSetExpandedRow = useCallback(
  949. (ts: number, expand?: boolean) => {
  950. setExpandedRows((prev) => ({ ...prev, [ts]: expand === undefined ? !prev[ts] : expand }))
  951. },
  952. [setExpandedRows], // setExpandedRows is stable
  953. )
  954. // Scroll when user toggles certain rows.
  955. const toggleRowExpansion = useCallback(
  956. (ts: number) => {
  957. handleSetExpandedRow(ts)
  958. // The logic to set disableAutoScrollRef.current = true on expansion
  959. // is now handled by the useEffect hook that observes expandedRows.
  960. },
  961. [handleSetExpandedRow],
  962. )
  963. const handleRowHeightChange = useCallback(
  964. (isTaller: boolean) => {
  965. if (!disableAutoScrollRef.current) {
  966. if (isTaller) {
  967. scrollToBottomSmooth()
  968. } else {
  969. setTimeout(() => scrollToBottomAuto(), 0)
  970. }
  971. }
  972. },
  973. [scrollToBottomSmooth, scrollToBottomAuto],
  974. )
  975. useEffect(() => {
  976. if (!disableAutoScrollRef.current) {
  977. setTimeout(() => scrollToBottomSmooth(), 50)
  978. // Don't cleanup since if visibleMessages.length changes it cancels.
  979. // return () => clearTimeout(timer)
  980. }
  981. }, [groupedMessages.length, scrollToBottomSmooth])
  982. const handleWheel = useCallback((event: Event) => {
  983. const wheelEvent = event as WheelEvent
  984. if (wheelEvent.deltaY && wheelEvent.deltaY < 0) {
  985. if (scrollContainerRef.current?.contains(wheelEvent.target as Node)) {
  986. // User scrolled up
  987. disableAutoScrollRef.current = true
  988. }
  989. }
  990. }, [])
  991. useEvent("wheel", handleWheel, window, { passive: true }) // passive improves scrolling performance
  992. // Effect to handle showing the checkpoint warning after a delay
  993. useEffect(() => {
  994. // Only show the warning when there's a task but no visible messages yet
  995. if (task && modifiedMessages.length === 0 && !isStreaming) {
  996. const timer = setTimeout(() => {
  997. setShowCheckpointWarning(true)
  998. }, 5000) // 5 seconds
  999. return () => clearTimeout(timer)
  1000. }
  1001. }, [task, modifiedMessages.length, isStreaming])
  1002. // Effect to hide the checkpoint warning when messages appear
  1003. useEffect(() => {
  1004. if (modifiedMessages.length > 0 || isStreaming) {
  1005. setShowCheckpointWarning(false)
  1006. }
  1007. }, [modifiedMessages.length, isStreaming])
  1008. const placeholderText = task ? t("chat:typeMessage") : t("chat:typeTask")
  1009. const handleSuggestionClickInRow = useCallback(
  1010. (answer: string, event?: React.MouseEvent) => {
  1011. if (event?.shiftKey) {
  1012. // Always append to existing text, don't overwrite
  1013. setInputValue((currentValue) => {
  1014. return currentValue !== "" ? `${currentValue} \n${answer}` : answer
  1015. })
  1016. } else {
  1017. handleSendMessage(answer, [])
  1018. }
  1019. },
  1020. [handleSendMessage, setInputValue], // setInputValue is stable, handleSendMessage depends on clineAsk
  1021. )
  1022. const itemContent = useCallback(
  1023. (index: number, messageOrGroup: ClineMessage | ClineMessage[]) => {
  1024. // browser session group
  1025. if (Array.isArray(messageOrGroup)) {
  1026. return (
  1027. <BrowserSessionRow
  1028. messages={messageOrGroup}
  1029. isLast={index === groupedMessages.length - 1}
  1030. lastModifiedMessage={modifiedMessages.at(-1)}
  1031. onHeightChange={handleRowHeightChange}
  1032. isStreaming={isStreaming}
  1033. isExpanded={(messageTs: number) => expandedRows[messageTs] ?? false}
  1034. onToggleExpand={(messageTs: number) => {
  1035. setExpandedRows((prev) => ({
  1036. ...prev,
  1037. [messageTs]: !prev[messageTs],
  1038. }))
  1039. }}
  1040. />
  1041. )
  1042. }
  1043. // regular message
  1044. return (
  1045. <ChatRow
  1046. key={messageOrGroup.ts}
  1047. message={messageOrGroup}
  1048. isExpanded={expandedRows[messageOrGroup.ts] || false}
  1049. onToggleExpand={toggleRowExpansion} // This was already stabilized
  1050. lastModifiedMessage={modifiedMessages.at(-1)} // Original direct access
  1051. isLast={index === groupedMessages.length - 1} // Original direct access
  1052. onHeightChange={handleRowHeightChange}
  1053. isStreaming={isStreaming}
  1054. onSuggestionClick={handleSuggestionClickInRow} // This was already stabilized
  1055. />
  1056. )
  1057. },
  1058. [
  1059. // Original broader dependencies
  1060. expandedRows,
  1061. groupedMessages,
  1062. modifiedMessages,
  1063. handleRowHeightChange,
  1064. isStreaming,
  1065. toggleRowExpansion,
  1066. handleSuggestionClickInRow,
  1067. setExpandedRows, // For the inline onToggleExpand in BrowserSessionRow
  1068. ],
  1069. )
  1070. useEffect(() => {
  1071. // Only proceed if we have an ask and buttons are enabled.
  1072. if (!clineAsk || !enableButtons) {
  1073. return
  1074. }
  1075. const autoApprove = async () => {
  1076. if (lastMessage?.ask && isAutoApproved(lastMessage)) {
  1077. // Note that `isAutoApproved` can only return true if
  1078. // lastMessage is an ask of type "browser_action_launch",
  1079. // "use_mcp_server", "command", or "tool".
  1080. // Add delay for write operations.
  1081. if (lastMessage.ask === "tool" && isWriteToolAction(lastMessage)) {
  1082. await new Promise((resolve) => setTimeout(resolve, writeDelayMs))
  1083. }
  1084. vscode.postMessage({ type: "askResponse", askResponse: "yesButtonClicked" })
  1085. // This is copied from `handlePrimaryButtonClick`, which we used
  1086. // to call from `autoApprove`. I'm not sure how many of these
  1087. // things are actually needed.
  1088. setSendingDisabled(true)
  1089. setClineAsk(undefined)
  1090. setEnableButtons(false)
  1091. }
  1092. }
  1093. autoApprove()
  1094. }, [
  1095. clineAsk,
  1096. enableButtons,
  1097. handlePrimaryButtonClick,
  1098. alwaysAllowBrowser,
  1099. alwaysAllowReadOnly,
  1100. alwaysAllowReadOnlyOutsideWorkspace,
  1101. alwaysAllowWrite,
  1102. alwaysAllowWriteOutsideWorkspace,
  1103. alwaysAllowExecute,
  1104. alwaysAllowMcp,
  1105. messages,
  1106. allowedCommands,
  1107. mcpServers,
  1108. isAutoApproved,
  1109. lastMessage,
  1110. writeDelayMs,
  1111. isWriteToolAction,
  1112. ])
  1113. // Function to handle mode switching
  1114. const switchToNextMode = useCallback(() => {
  1115. const allModes = getAllModes(customModes)
  1116. const currentModeIndex = allModes.findIndex((m) => m.slug === mode)
  1117. const nextModeIndex = (currentModeIndex + 1) % allModes.length
  1118. // Update local state and notify extension to sync mode change
  1119. setMode(allModes[nextModeIndex].slug)
  1120. vscode.postMessage({
  1121. type: "mode",
  1122. text: allModes[nextModeIndex].slug,
  1123. })
  1124. }, [mode, setMode, customModes])
  1125. // Add keyboard event handler
  1126. const handleKeyDown = useCallback(
  1127. (event: KeyboardEvent) => {
  1128. // Check for Command + . (period)
  1129. if ((event.metaKey || event.ctrlKey) && event.key === ".") {
  1130. event.preventDefault() // Prevent default browser behavior
  1131. switchToNextMode()
  1132. }
  1133. },
  1134. [switchToNextMode],
  1135. )
  1136. // Add event listener
  1137. useEffect(() => {
  1138. window.addEventListener("keydown", handleKeyDown)
  1139. return () => {
  1140. window.removeEventListener("keydown", handleKeyDown)
  1141. }
  1142. }, [handleKeyDown])
  1143. useImperativeHandle(ref, () => ({
  1144. acceptInput: () => {
  1145. if (enableButtons && primaryButtonText) {
  1146. handlePrimaryButtonClick(inputValue, selectedImages)
  1147. } else if (!sendingDisabled && (inputValue.trim() || selectedImages.length > 0)) {
  1148. handleSendMessage(inputValue, selectedImages)
  1149. }
  1150. },
  1151. }))
  1152. const handleCondenseContext = (taskId: string) => {
  1153. if (isCondensing || sendingDisabled) {
  1154. return
  1155. }
  1156. setIsCondensing(true)
  1157. setSendingDisabled(true)
  1158. vscode.postMessage({ type: "condenseTaskContextRequest", text: taskId })
  1159. }
  1160. return (
  1161. <div className={isHidden ? "hidden" : "fixed top-0 left-0 right-0 bottom-0 flex flex-col overflow-hidden"}>
  1162. {showAnnouncement && <Announcement hideAnnouncement={hideAnnouncement} />}
  1163. {task ? (
  1164. <>
  1165. <TaskHeader
  1166. task={task}
  1167. tokensIn={apiMetrics.totalTokensIn}
  1168. tokensOut={apiMetrics.totalTokensOut}
  1169. doesModelSupportPromptCache={model?.supportsPromptCache ?? false}
  1170. cacheWrites={apiMetrics.totalCacheWrites}
  1171. cacheReads={apiMetrics.totalCacheReads}
  1172. totalCost={apiMetrics.totalCost}
  1173. contextTokens={apiMetrics.contextTokens}
  1174. buttonsDisabled={sendingDisabled}
  1175. handleCondenseContext={handleCondenseContext}
  1176. onClose={handleTaskCloseButtonClick}
  1177. />
  1178. {hasSystemPromptOverride && (
  1179. <div className="px-3">
  1180. <SystemPromptWarning />
  1181. </div>
  1182. )}
  1183. {showCheckpointWarning && (
  1184. <div className="px-3">
  1185. <CheckpointWarning />
  1186. </div>
  1187. )}
  1188. </>
  1189. ) : (
  1190. <div className="flex-1 min-h-0 overflow-y-auto flex flex-col gap-4">
  1191. {/* Moved Task Bar Header Here */}
  1192. {tasks.length !== 0 && (
  1193. <div className="flex text-vscode-descriptionForeground w-full mx-auto px-5 pt-3">
  1194. <div className="flex items-center gap-1 cursor-pointer" onClick={toggleExpanded}>
  1195. {tasks.length < 10 && (
  1196. <span className={`font-medium text-xs `}>{t("history:recentTasks")}</span>
  1197. )}
  1198. <span
  1199. className={`codicon ${isExpanded ? "codicon-eye" : "codicon-eye-closed"} scale-90`}
  1200. />
  1201. </div>
  1202. </div>
  1203. )}
  1204. <div
  1205. className={` w-full flex flex-col gap-4 m-auto ${isExpanded && tasks.length > 0 ? "mt-0" : ""} px-3.5 min-[370px]:px-10 pt-5 transition-all duration-300`}>
  1206. <RooHero />
  1207. {telemetrySetting === "unset" && <TelemetryBanner />}
  1208. {/* Show the task history preview if expanded and tasks exist */}
  1209. {taskHistory.length > 0 && isExpanded && <HistoryPreview />}
  1210. <p className="text-vscode-editor-foreground leading-tight font-vscode-font-family text-center text-balance max-w-[380px] mx-auto">
  1211. <Trans
  1212. i18nKey="chat:about"
  1213. components={{
  1214. DocsLink: (
  1215. <a href={buildDocLink("", "welcome")} target="_blank" rel="noopener noreferrer">
  1216. the docs
  1217. </a>
  1218. ),
  1219. }}
  1220. />
  1221. </p>
  1222. <RooTips cycle={false} />
  1223. </div>
  1224. </div>
  1225. )}
  1226. {/*
  1227. // Flex layout explanation:
  1228. // 1. Content div above uses flex: "1 1 0" to:
  1229. // - Grow to fill available space (flex-grow: 1)
  1230. // - Shrink when AutoApproveMenu needs space (flex-shrink: 1)
  1231. // - Start from zero size (flex-basis: 0) to ensure proper distribution
  1232. // minHeight: 0 allows it to shrink below its content height
  1233. //
  1234. // 2. AutoApproveMenu uses flex: "0 1 auto" to:
  1235. // - Not grow beyond its content (flex-grow: 0)
  1236. // - Shrink when viewport is small (flex-shrink: 1)
  1237. // - Use its content size as basis (flex-basis: auto)
  1238. // This ensures it takes its natural height when there's space
  1239. // but becomes scrollable when the viewport is too small
  1240. */}
  1241. {!task && (
  1242. <div className="mb-[-2px] flex-initial min-h-0">
  1243. <AutoApproveMenu />
  1244. </div>
  1245. )}
  1246. {task && (
  1247. <>
  1248. <div className="grow flex" ref={scrollContainerRef}>
  1249. <Virtuoso
  1250. ref={virtuosoRef}
  1251. key={task.ts} // trick to make sure virtuoso re-renders when task changes, and we use initialTopMostItemIndex to start at the bottom
  1252. className="scrollable grow overflow-y-scroll"
  1253. components={{
  1254. Footer: () => <div className="h-[5px]" />, // Add empty padding at the bottom
  1255. }}
  1256. // increasing top by 3_000 to prevent jumping around when user collapses a row
  1257. increaseViewportBy={{ top: 3_000, bottom: Number.MAX_SAFE_INTEGER }} // hack to make sure the last message is always rendered to get truly perfect scroll to bottom animation when new messages are added (Number.MAX_SAFE_INTEGER is safe for arithmetic operations, which is all virtuoso uses this value for in src/sizeRangeSystem.ts)
  1258. data={groupedMessages} // messages is the raw format returned by extension, modifiedMessages is the manipulated structure that combines certain messages of related type, and visibleMessages is the filtered structure that removes messages that should not be rendered
  1259. itemContent={itemContent}
  1260. atBottomStateChange={(isAtBottom) => {
  1261. setIsAtBottom(isAtBottom)
  1262. if (isAtBottom) {
  1263. disableAutoScrollRef.current = false
  1264. }
  1265. setShowScrollToBottom(disableAutoScrollRef.current && !isAtBottom)
  1266. }}
  1267. atBottomThreshold={10} // anything lower causes issues with followOutput
  1268. initialTopMostItemIndex={groupedMessages.length - 1}
  1269. />
  1270. </div>
  1271. <AutoApproveMenu />
  1272. {showScrollToBottom ? (
  1273. <div className="flex px-[15px] pt-[10px]">
  1274. <div
  1275. className="bg-[color-mix(in_srgb,_var(--vscode-toolbar-hoverBackground)_55%,_transparent)] rounded-[3px] overflow-hidden cursor-pointer flex justify-center items-center flex-1 h-[25px] hover:bg-[color-mix(in_srgb,_var(--vscode-toolbar-hoverBackground)_90%,_transparent)] active:bg-[color-mix(in_srgb,_var(--vscode-toolbar-hoverBackground)_70%,_transparent)]"
  1276. onClick={() => {
  1277. scrollToBottomSmooth()
  1278. disableAutoScrollRef.current = false
  1279. }}
  1280. title={t("chat:scrollToBottom")}>
  1281. <span className="codicon codicon-chevron-down text-[18px]"></span>
  1282. </div>
  1283. </div>
  1284. ) : (
  1285. <div
  1286. className={`flex ${
  1287. primaryButtonText || secondaryButtonText || isStreaming ? "px-[15px] pt-[10px]" : "p-0"
  1288. } ${
  1289. primaryButtonText || secondaryButtonText || isStreaming
  1290. ? enableButtons || (isStreaming && !didClickCancel)
  1291. ? "opacity-100"
  1292. : "opacity-50"
  1293. : "opacity-0"
  1294. }`}>
  1295. {primaryButtonText && !isStreaming && (
  1296. <VSCodeButton
  1297. appearance="primary"
  1298. disabled={!enableButtons}
  1299. className={secondaryButtonText ? "flex-1 mr-[6px]" : "flex-[2] mr-0"}
  1300. title={
  1301. primaryButtonText === t("chat:retry.title")
  1302. ? t("chat:retry.tooltip")
  1303. : primaryButtonText === t("chat:save.title")
  1304. ? t("chat:save.tooltip")
  1305. : primaryButtonText === t("chat:approve.title")
  1306. ? t("chat:approve.tooltip")
  1307. : primaryButtonText === t("chat:runCommand.title")
  1308. ? t("chat:runCommand.tooltip")
  1309. : primaryButtonText === t("chat:startNewTask.title")
  1310. ? t("chat:startNewTask.tooltip")
  1311. : primaryButtonText === t("chat:resumeTask.title")
  1312. ? t("chat:resumeTask.tooltip")
  1313. : primaryButtonText === t("chat:proceedAnyways.title")
  1314. ? t("chat:proceedAnyways.tooltip")
  1315. : primaryButtonText ===
  1316. t("chat:proceedWhileRunning.title")
  1317. ? t("chat:proceedWhileRunning.tooltip")
  1318. : undefined
  1319. }
  1320. onClick={() => handlePrimaryButtonClick(inputValue, selectedImages)}>
  1321. {primaryButtonText}
  1322. </VSCodeButton>
  1323. )}
  1324. {(secondaryButtonText || isStreaming) && (
  1325. <VSCodeButton
  1326. appearance="secondary"
  1327. disabled={!enableButtons && !(isStreaming && !didClickCancel)}
  1328. className={isStreaming ? "flex-[2] ml-0" : "flex-1 ml-[6px]"}
  1329. title={
  1330. isStreaming
  1331. ? t("chat:cancel.tooltip")
  1332. : secondaryButtonText === t("chat:startNewTask.title")
  1333. ? t("chat:startNewTask.tooltip")
  1334. : secondaryButtonText === t("chat:reject.title")
  1335. ? t("chat:reject.tooltip")
  1336. : secondaryButtonText === t("chat:terminate.title")
  1337. ? t("chat:terminate.tooltip")
  1338. : undefined
  1339. }
  1340. onClick={() => handleSecondaryButtonClick(inputValue, selectedImages)}>
  1341. {isStreaming ? t("chat:cancel.title") : secondaryButtonText}
  1342. </VSCodeButton>
  1343. )}
  1344. </div>
  1345. )}
  1346. </>
  1347. )}
  1348. <ChatTextArea
  1349. ref={textAreaRef}
  1350. inputValue={inputValue}
  1351. setInputValue={setInputValue}
  1352. sendingDisabled={sendingDisabled}
  1353. selectApiConfigDisabled={sendingDisabled && clineAsk !== "api_req_failed"}
  1354. placeholderText={placeholderText}
  1355. selectedImages={selectedImages}
  1356. setSelectedImages={setSelectedImages}
  1357. onSend={() => handleSendMessage(inputValue, selectedImages)}
  1358. onSelectImages={selectImages}
  1359. shouldDisableImages={shouldDisableImages}
  1360. onHeightChange={() => {
  1361. if (isAtBottom) {
  1362. scrollToBottomAuto()
  1363. }
  1364. }}
  1365. mode={mode}
  1366. setMode={setMode}
  1367. modeShortcutText={modeShortcutText}
  1368. />
  1369. <div id="roo-portal" />
  1370. </div>
  1371. )
  1372. }
  1373. const ChatView = forwardRef(ChatViewComponent)
  1374. export default ChatView