From 8454f94dbec57fbcbe658aec029a9736abf437df Mon Sep 17 00:00:00 2001
From: Aiden Cline <rekram1-node@users.noreply.github.com>
Date: Sun, 19 Jul 2026 04:54:32 +0000
Subject: [PATCH] fix(tui): collapse direct shell output

---
 packages/tui/src/routes/session/index.tsx | 21 +++++++++------------
 1 file changed, 9 insertions(+), 12 deletions(-)

diff --git a/packages/tui/src/routes/session/index.tsx b/packages/tui/src/routes/session/index.tsx
index a6cf7532fc..2cd92782ed 100644
--- a/packages/tui/src/routes/session/index.tsx
+++ b/packages/tui/src/routes/session/index.tsx
@@ -1607,24 +1607,21 @@ function RevertMessage(props: {
 
 function ShellMessage(props: { message: Extract<SessionMessageInfo, { type: "shell" }> }) {
   const { themeV2 } = useTheme().contextual("elevated")
+  const ctx = use()
   const output = createMemo(() => stripAnsi(props.message.output?.output.trim() ?? ""))
+  const [expanded, setExpanded] = createSignal(false)
+  const maxLines = 10
+  const maxChars = createMemo(() => maxLines * Math.max(20, ctx.width - 6))
+  const collapsed = createMemo(() => collapseToolOutput(output(), maxLines, maxChars()))
+  const limited = createMemo(() => (expanded() ? output() : collapsed().output))
 
   return (
-    <box
-      border={["left"]}
-      paddingTop={1}
-      paddingBottom={1}
-      paddingLeft={2}
-      gap={1}
-      backgroundColor={themeV2.background()}
-      customBorderChars={SplitBorder.customBorderChars}
-      borderColor={themeV2.background()}
-    >
+    <BlockTool onClick={collapsed().overflow ? () => setExpanded((value) => !value) : undefined}>
       <text fg={themeV2.text()}>$ {props.message.command}</text>
       <Show when={output()}>
-        <text fg={themeV2.text.subdued()}>{output()}</text>
+        <text fg={themeV2.text.subdued()}>{limited()}</text>
       </Show>
-    </box>
+    </BlockTool>
   )
 }