Merge pull request #37 from ai-action/feat/exit

Author: remarkablemark

src/components/App.test.tsx

@@ -3,6 +3,17 @@ import { render } from 'ink-testing-library';
 
 import { time } from '../utils';
 
+const { mockExit } = vi.hoisted(() => ({
+  mockExit: vi.fn(),
+}));
+
+vi.mock('ink', async () => ({
+  ...(await vi.importActual('ink')),
+  useApp: vi.fn(() => ({
+    exit: mockExit,
+  })),
+}));
+
 const resetSystemMessage = vi.hoisted(() => vi.fn());
 
 vi.mock('../utils', async () => ({
@@ -90,6 +101,7 @@ describe('App', () => {
     capturedCallbacks.onClose = null;
     capturedCallbacks.onToggleMode = null;
     resetSystemMessage.mockClear();
+    mockExit.mockReset();
   });
 
   it('renders title', () => {
@@ -137,6 +149,12 @@ describe('App', () => {
     expect(lastFrame()).not.toContain('ModelPicker');
   });
 
+  it('calls exit when /exit command is issued', () => {
+    render(<App />);
+    capturedCallbacks.onCommand?.('/exit');
+    expect(mockExit).toHaveBeenCalledOnce();
+  });
+
   it('resets the chat session when /clear is issued', async () => {
     const { lastFrame, rerender } = render(<App />);
 

src/components/App.tsx

@@ -1,4 +1,4 @@
-import { Box } from 'ink';
+import { Box, useApp } from 'ink';
 import { useCallback, useState } from 'react';
 
 import { MODE } from '../constants';
@@ -9,24 +9,32 @@ import { Header } from './Header';
 import { ModelPicker } from './ModelPicker';
 
 export function App() {
+  const { exit } = useApp();
   const [model, setModel] = useState(() => config.loadConfig().model);
   const [picking, setPicking] = useState(false);
   const [mode, setMode] = useState<MODE.Name>(MODE.NAME.SAFE);
   const [sessionId, setSessionId] = useState(0);
 
-  const handleCommand = useCallback((command: string) => {
-    switch (command) {
-      case '/model':
-        setPicking(true);
-        break;
+  const handleCommand = useCallback(
+    (command: string) => {
+      switch (command) {
+        case '/model':
+          setPicking(true);
+          break;
 
-      case '/clear':
-        agents.resetSystemMessage();
-        setPicking(false);
-        setSessionId((sessionId) => sessionId + 1);
-        break;
-    }
-  }, []);
+        case '/clear':
+          agents.resetSystemMessage();
+          setPicking(false);
+          setSessionId((sessionId) => sessionId + 1);
+          break;
+
+        case '/exit':
+          exit();
+          break;
+      }
+    },
+    [exit],
+  );
 
   const handleSelect = useCallback((selected: string) => {
     setModel(selected);

src/constants/command.ts

@@ -6,4 +6,5 @@ export interface CommandList {
 export const LIST: CommandList[] = [
   { name: '/clear', description: 'clear the current session' },
   { name: '/model', description: 'switch the model' },
+  { name: '/exit', description: 'exit the application' },
 ] as const;