feat: Add exit UI w/ stats (#924)

This commit is contained in:
Abhi
2025-06-11 16:40:31 -04:00
committed by GitHub
parent 4160d904da
commit 7a72d255d8
13 changed files with 522 additions and 88 deletions

View File

@@ -396,6 +396,43 @@ Add any other context about the problem here.
});
});
describe('/quit and /exit commands', () => {
beforeEach(() => {
vi.useFakeTimers();
});
afterEach(() => {
vi.useRealTimers();
});
it.each([['/quit'], ['/exit']])(
'should handle %s, add a quit message, and exit the process',
async (command) => {
const { handleSlashCommand } = getProcessor();
const mockDate = new Date('2025-01-01T01:02:03.000Z');
vi.setSystemTime(mockDate);
await act(async () => {
handleSlashCommand(command);
});
expect(mockAddItem).toHaveBeenCalledTimes(2);
expect(mockAddItem).toHaveBeenNthCalledWith(
2,
expect.objectContaining({
type: MessageType.QUIT,
duration: '1h 2m 3s',
}),
expect.any(Number),
);
// Fast-forward timers to trigger process.exit
vi.advanceTimersByTime(100);
expect(mockProcessExit).toHaveBeenCalledWith(0);
},
);
});
describe('Unknown command', () => {
it('should show an error and return true for a general unknown command', async () => {
const { handleSlashCommand } = getProcessor();

View File

@@ -97,6 +97,12 @@ export const useSlashCommandProcessor = (
lastTurnStats: message.lastTurnStats,
duration: message.duration,
};
} else if (message.type === MessageType.QUIT) {
historyItemContent = {
type: 'quit',
stats: message.stats,
duration: message.duration,
};
} else {
historyItemContent = {
type: message.type as
@@ -594,8 +600,20 @@ Add any other context about the problem here.
altName: 'exit',
description: 'exit the cli',
action: async (_mainCommand, _subCommand, _args) => {
onDebugMessage('Quitting. Good-bye.');
process.exit(0);
const now = new Date();
const { sessionStartTime, cumulative } = session.stats;
const wallDuration = now.getTime() - sessionStartTime.getTime();
addMessage({
type: MessageType.QUIT,
stats: cumulative,
duration: formatDuration(wallDuration),
timestamp: new Date(),
});
setTimeout(() => {
process.exit(0);
}, 100);
},
},
];
@@ -721,6 +739,7 @@ Add any other context about the problem here.
session,
gitService,
loadHistory,
addItem,
]);
const handleSlashCommand = useCallback(

View File

@@ -132,7 +132,7 @@ export function useReactToolScheduler(
});
onComplete(completedToolCalls);
},
[onComplete],
[onComplete, config],
);
const toolCallsUpdateHandler: ToolCallsUpdateHandler = useCallback(