From 8a70b98d1dd492c645b828a2e180a66e1c20d5cd Mon Sep 17 00:00:00 2001
From: Olcan <olcans@gmail.com>
Date: Wed, 21 May 2025 09:00:54 -0700
Subject: [PATCH] fix exit code for shell mode also (#466)

---
 packages/cli/src/ui/hooks/shellCommandProcessor.test.ts | 4 ++--
 packages/cli/src/ui/hooks/shellCommandProcessor.ts      | 2 +-
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/packages/cli/src/ui/hooks/shellCommandProcessor.test.ts b/packages/cli/src/ui/hooks/shellCommandProcessor.test.ts
index 6d7ea5d6..3645cc80 100644
--- a/packages/cli/src/ui/hooks/shellCommandProcessor.test.ts
+++ b/packages/cli/src/ui/hooks/shellCommandProcessor.test.ts
@@ -155,7 +155,7 @@ describe('useShellCommandProcessor', () => {
       expect.stringContaining('Executing shell command in /current/dir:'),
     );
     expect(mockExecuteCommand).toHaveBeenCalledWith(
-      '{ !ls -l; }; pwd >/tmp/shell_pwd_randomBytes.tmp',
+      '{ !ls -l; }; __code=$?; pwd >/tmp/shell_pwd_randomBytes.tmp; exit $__code',
       { cwd: '/current/dir' },
       expect.any(Function),
     );
@@ -272,7 +272,7 @@ describe('useShellCommandProcessor', () => {
       expect.any(Number),
     );
     expect(mockExecuteCommand).toHaveBeenCalledWith(
-      '{ !sleep 5 & }; pwd >/tmp/shell_pwd_randomBytes.tmp',
+      '{ !sleep 5 & }; __code=$?; pwd >/tmp/shell_pwd_randomBytes.tmp; exit $__code',
       { cwd: '/current/dir' },
       expect.any(Function),
     );
diff --git a/packages/cli/src/ui/hooks/shellCommandProcessor.ts b/packages/cli/src/ui/hooks/shellCommandProcessor.ts
index 35935e7d..ae8367d6 100644
--- a/packages/cli/src/ui/hooks/shellCommandProcessor.ts
+++ b/packages/cli/src/ui/hooks/shellCommandProcessor.ts
@@ -43,7 +43,7 @@ export const useShellCommandProcessor = (
       const pwdFilePath = path.join(os.tmpdir(), pwdFileName);
       if (!commandToExecute.endsWith('&')) commandToExecute += ';';
       // note here we could also restore a previous pwd with `cd {cwd}; { ... }`
-      commandToExecute = `{ ${commandToExecute} }; pwd >${pwdFilePath}`;
+      commandToExecute = `{ ${commandToExecute} }; __code=$?; pwd >${pwdFilePath}; exit $__code`;
 
       const userMessageTimestamp = Date.now();
       addItemToHistory(