clone/gemini-cli
1
0
Fork 0
mirror of https://github.com/google-gemini/gemini-cli.git synced 2026-02-01 22:48:03 +00:00
Code Issues Packages Projects Releases Wiki Activity

disable/save/restore listeners on process.stdin

Browse Source
This commit is contained in:
olcan
2025-05-22 12:25:37 -07:00
parent 99289463ea
commit 535f11e9bf
1 changed files with 24 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
packages/cli/src/ui/hooks/shellCommandProcessor.ts
View File

@@ -129,13 +129,37 @@ export const useShellCommandProcessor = (
setPendingHistoryItem({ type: 'info', text: output });
});
// disable & save all stdin listeners to be restored onExit
const origStdinListeners = new Map<
string | symbol,
Array<(...args: unknown[]) => void>
>();
process.stdin.eventNames().forEach((eventName) => {
const listeners = process.stdin
.listeners(eventName)
.slice() as Array<(...args: unknown[]) => void>;
if (listeners.length > 0) {
origStdinListeners.set(eventName, listeners);
listeners.forEach((listener) => {
process.stdin.off(eventName, listener);
});
}
});
const stdinListener = (data: Buffer) => {
child.write(data.toString());
};
process.stdin.on('data', stdinListener);
child.onExit(({ exitCode, signal }) => {
// restore original stdin listeners
process.stdin.removeListener('data', stdinListener);
origStdinListeners.forEach((listeners, eventName) => {
listeners.forEach((listener) =>
process.stdin.on(eventName as string, listener),
);
});
setPendingHistoryItem(null);
output = output.trim() || '(Command produced no output)';