@machina.ai/cell-cli 1.16.0-rc1 → 1.18.4-rc1

package/dist/src/ui/components/InputPrompt.test.js

@@ -16,6 +16,7 @@ import { useShellHistory } from '../hooks/useShellHistory.js';
 import { useCommandCompletion } from '../hooks/useCommandCompletion.js';
 import { useInputHistory } from '../hooks/useInputHistory.js';
 import { useReverseSearchCompletion } from '../hooks/useReverseSearchCompletion.js';
+import clipboardy from 'clipboardy';
 import * as clipboardUtils from '../utils/clipboardUtils.js';
 import { useKittyKeyboardProtocol } from '../hooks/useKittyKeyboardProtocol.js';
 import { createMockCommandContext } from '../../test-utils/mockCommandContext.js';
@@ -26,6 +27,7 @@ vi.mock('../hooks/useShellHistory.js');
 vi.mock('../hooks/useCommandCompletion.js');
 vi.mock('../hooks/useInputHistory.js');
 vi.mock('../hooks/useReverseSearchCompletion.js');
+vi.mock('clipboardy');
 vi.mock('../utils/clipboardUtils.js');
 vi.mock('../hooks/useKittyKeyboardProtocol.js');
 const mockSlashCommands = [
@@ -88,6 +90,10 @@ describe('InputPrompt', () => {
     const mockedUseInputHistory = vi.mocked(useInputHistory);
     const mockedUseReverseSearchCompletion = vi.mocked(useReverseSearchCompletion);
     const mockedUseKittyKeyboardProtocol = vi.mocked(useKittyKeyboardProtocol);
+    const mockSetEmbeddedShellFocused = vi.fn();
+    const uiActions = {
+        setEmbeddedShellFocused: mockSetEmbeddedShellFocused,
+    };
     beforeEach(() => {
         vi.resetAllMocks();
         mockCommandContext = createMockCommandContext();
@@ -129,6 +135,7 @@ describe('InputPrompt', () => {
             deleteWordLeft: vi.fn(),
             deleteWordRight: vi.fn(),
             visualToLogicalMap: [[0, 0]],
+            getOffset: vi.fn().mockReturnValue(0),
         };
         mockShellHistory = {
             history: [],
@@ -180,7 +187,6 @@ describe('InputPrompt', () => {
         };
         mockedUseReverseSearchCompletion.mockReturnValue(mockReverseSearchCompletion);
         mockedUseKittyKeyboardProtocol.mockReturnValue({
-            supported: false,
             enabled: false,
             checking: false,
         });
@@ -212,7 +218,9 @@ describe('InputPrompt', () => {
     });
     it('should call shellHistory.getPreviousCommand on up arrow in shell mode', async () => {
         props.shellModeActive = true;
-        const { stdin, unmount } = renderWithProviders(_jsx(InputPrompt, { ...props }));
+        const { stdin, unmount } = renderWithProviders(_jsx(InputPrompt, { ...props }), {
+            uiActions,
+        });
         await act(async () => {
             stdin.write('\u001B[A');
         });
@@ -221,7 +229,9 @@ describe('InputPrompt', () => {
     });
     it('should call shellHistory.getNextCommand on down arrow in shell mode', async () => {
         props.shellModeActive = true;
-        const { stdin, unmount } = renderWithProviders(_jsx(InputPrompt, { ...props }));
+        const { stdin, unmount } = renderWithProviders(_jsx(InputPrompt, { ...props }), {
+            uiActions,
+        });
         await act(async () => {
             stdin.write('\u001B[B');
             await waitFor(() => expect(mockShellHistory.getNextCommand).toHaveBeenCalled());
@@ -231,7 +241,9 @@ describe('InputPrompt', () => {
     it('should set the buffer text when a shell history command is retrieved', async () => {
         props.shellModeActive = true;
         vi.mocked(mockShellHistory.getPreviousCommand).mockReturnValue('previous command');
-        const { stdin, unmount } = renderWithProviders(_jsx(InputPrompt, { ...props }));
+        const { stdin, unmount } = renderWithProviders(_jsx(InputPrompt, { ...props }), {
+            uiActions,
+        });
         await act(async () => {
             stdin.write('\u001B[A');
         });
@@ -244,7 +256,9 @@ describe('InputPrompt', () => {
     it('should call shellHistory.addCommandToHistory on submit in shell mode', async () => {
         props.shellModeActive = true;
         props.buffer.setText('ls -l');
-        const { stdin, unmount } = renderWithProviders(_jsx(InputPrompt, { ...props }));
+        const { stdin, unmount } = renderWithProviders(_jsx(InputPrompt, { ...props }), {
+            uiActions,
+        });
         await act(async () => {
             stdin.write('\r');
         });
@@ -256,7 +270,9 @@ describe('InputPrompt', () => {
     });
     it('should NOT call shell history methods when not in shell mode', async () => {
         props.buffer.setText('some text');
-        const { stdin, unmount } = renderWithProviders(_jsx(InputPrompt, { ...props }));
+        const { stdin, unmount } = renderWithProviders(_jsx(InputPrompt, { ...props }), {
+            uiActions,
+        });
         await act(async () => {
             stdin.write('\u001B[A'); // Up arrow
         });
@@ -284,7 +300,9 @@ describe('InputPrompt', () => {
             ],
         });
         props.buffer.setText('/mem');
-        const { stdin, unmount } = renderWithProviders(_jsx(InputPrompt, { ...props }));
+        const { stdin, unmount } = renderWithProviders(_jsx(InputPrompt, { ...props }), {
+            uiActions,
+        });
         // Test up arrow
         await act(async () => {
             stdin.write('\u001B[A'); // Up arrow
@@ -307,7 +325,9 @@ describe('InputPrompt', () => {
             ],
         });
         props.buffer.setText('/mem');
-        const { stdin, unmount } = renderWithProviders(_jsx(InputPrompt, { ...props }));
+        const { stdin, unmount } = renderWithProviders(_jsx(InputPrompt, { ...props }), {
+            uiActions,
+        });
         // Test down arrow
         await act(async () => {
             stdin.write('\u001B[B'); // Down arrow
@@ -326,7 +346,9 @@ describe('InputPrompt', () => {
             showSuggestions: false,
         });
         props.buffer.setText('some text');
-        const { stdin, unmount } = renderWithProviders(_jsx(InputPrompt, { ...props }));
+        const { stdin, unmount } = renderWithProviders(_jsx(InputPrompt, { ...props }), {
+            uiActions,
+        });
         await act(async () => {
             stdin.write('\u001B[A'); // Up arrow
         });
@@ -402,6 +424,7 @@ describe('InputPrompt', () => {
             // Set initial text and cursor position
             mockBuffer.text = 'Hello world';
             mockBuffer.cursor = [0, 5]; // Cursor after "Hello"
+            vi.mocked(mockBuffer.getOffset).mockReturnValue(5);
             mockBuffer.lines = ['Hello world'];
             mockBuffer.replaceRangeByOffset = vi.fn();
             const { stdin, unmount } = renderWithProviders(_jsx(InputPrompt, { ...props }));
@@ -437,6 +460,22 @@ describe('InputPrompt', () => {
             unmount();
         });
     });
+    describe('clipboard text paste', () => {
+        it('should insert text from clipboard on Ctrl+V', async () => {
+            vi.mocked(clipboardUtils.clipboardHasImage).mockResolvedValue(false);
+            vi.mocked(clipboardy.read).mockResolvedValue('pasted text');
+            vi.mocked(mockBuffer.replaceRangeByOffset).mockClear();
+            const { stdin, unmount } = renderWithProviders(_jsx(InputPrompt, { ...props }));
+            await act(async () => {
+                stdin.write('\x16'); // Ctrl+V
+            });
+            await waitFor(() => {
+                expect(clipboardy.read).toHaveBeenCalled();
+                expect(mockBuffer.replaceRangeByOffset).toHaveBeenCalledWith(expect.any(Number), expect.any(Number), 'pasted text');
+            });
+            unmount();
+        });
+    });
     it.each([
         {
             name: 'should complete a partial parent command',
@@ -476,7 +515,9 @@ describe('InputPrompt', () => {
             activeSuggestionIndex: activeIndex,
         });
         props.buffer.setText(bufferText);
-        const { stdin, unmount } = renderWithProviders(_jsx(InputPrompt, { ...props }));
+        const { stdin, unmount } = renderWithProviders(_jsx(InputPrompt, { ...props }), {
+            uiActions,
+        });
         await act(async () => stdin.write('\t'));
         await waitFor(() => expect(mockCommandCompletion.handleAutocomplete).toHaveBeenCalledWith(activeIndex));
         unmount();
@@ -489,7 +530,9 @@ describe('InputPrompt', () => {
             activeSuggestionIndex: 0,
         });
         props.buffer.setText('/mem');
-        const { stdin, unmount } = renderWithProviders(_jsx(InputPrompt, { ...props }));
+        const { stdin, unmount } = renderWithProviders(_jsx(InputPrompt, { ...props }), {
+            uiActions,
+        });
         await act(async () => {
             stdin.write('\r');
         });
@@ -516,7 +559,9 @@ describe('InputPrompt', () => {
             activeSuggestionIndex: 0,
         });
         props.buffer.setText('/?');
-        const { stdin, unmount } = renderWithProviders(_jsx(InputPrompt, { ...props }));
+        const { stdin, unmount } = renderWithProviders(_jsx(InputPrompt, { ...props }), {
+            uiActions,
+        });
         await act(async () => {
             stdin.write('\t'); // Press Tab for autocomplete
         });
@@ -525,7 +570,9 @@ describe('InputPrompt', () => {
     });
     it('should not submit on Enter when the buffer is empty or only contains whitespace', async () => {
         props.buffer.setText('   '); // Set buffer to whitespace
-        const { stdin, unmount } = renderWithProviders(_jsx(InputPrompt, { ...props }));
+        const { stdin, unmount } = renderWithProviders(_jsx(InputPrompt, { ...props }), {
+            uiActions,
+        });
         await act(async () => {
             stdin.write('\r'); // Press Enter
         });
@@ -541,7 +588,9 @@ describe('InputPrompt', () => {
             isPerfectMatch: true,
         });
         props.buffer.setText('/clear');
-        const { stdin, unmount } = renderWithProviders(_jsx(InputPrompt, { ...props }));
+        const { stdin, unmount } = renderWithProviders(_jsx(InputPrompt, { ...props }), {
+            uiActions,
+        });
         await act(async () => {
             stdin.write('\r');
         });
@@ -555,7 +604,9 @@ describe('InputPrompt', () => {
             isPerfectMatch: false, // Added explicit isPerfectMatch false
         });
         props.buffer.setText('/clear');
-        const { stdin, unmount } = renderWithProviders(_jsx(InputPrompt, { ...props }));
+        const { stdin, unmount } = renderWithProviders(_jsx(InputPrompt, { ...props }), {
+            uiActions,
+        });
         await act(async () => {
             stdin.write('\r');
         });
@@ -570,7 +621,9 @@ describe('InputPrompt', () => {
             activeSuggestionIndex: 0,
         });
         props.buffer.setText('@src/components/');
-        const { stdin, unmount } = renderWithProviders(_jsx(InputPrompt, { ...props }));
+        const { stdin, unmount } = renderWithProviders(_jsx(InputPrompt, { ...props }), {
+            uiActions,
+        });
         await act(async () => {
             stdin.write('\r');
         });
@@ -583,7 +636,9 @@ describe('InputPrompt', () => {
         mockBuffer.text = 'first line\\';
         mockBuffer.cursor = [0, 11];
         mockBuffer.lines = ['first line\\'];
-        const { stdin, unmount } = renderWithProviders(_jsx(InputPrompt, { ...props }));
+        const { stdin, unmount } = renderWithProviders(_jsx(InputPrompt, { ...props }), {
+            uiActions,
+        });
         await act(async () => {
             stdin.write('\r');
         });
@@ -598,7 +653,9 @@ describe('InputPrompt', () => {
         await act(async () => {
             props.buffer.setText('some text to clear');
         });
-        const { stdin, unmount } = renderWithProviders(_jsx(InputPrompt, { ...props }));
+        const { stdin, unmount } = renderWithProviders(_jsx(InputPrompt, { ...props }), {
+            uiActions,
+        });
         await act(async () => {
             stdin.write('\x03'); // Ctrl+C character
         });
@@ -611,7 +668,9 @@ describe('InputPrompt', () => {
     });
     it('should NOT clear the buffer on Ctrl+C if it is empty', async () => {
         props.buffer.text = '';
-        const { stdin, unmount } = renderWithProviders(_jsx(InputPrompt, { ...props }));
+        const { stdin, unmount } = renderWithProviders(_jsx(InputPrompt, { ...props }), {
+            uiActions,
+        });
         await act(async () => {
             stdin.write('\x03'); // Ctrl+C character
         });
@@ -621,7 +680,9 @@ describe('InputPrompt', () => {
         unmount();
     });
     it('should call setBannerVisible(false) when clear screen key is pressed', async () => {
-        const { stdin, unmount } = renderWithProviders(_jsx(InputPrompt, { ...props }));
+        const { stdin, unmount } = renderWithProviders(_jsx(InputPrompt, { ...props }), {
+            uiActions,
+        });
         await act(async () => {
             stdin.write('\x0C'); // Ctrl+L
         });
@@ -721,7 +782,9 @@ describe('InputPrompt', () => {
                     ? [{ label: 'suggestion', value: 'suggestion' }]
                     : [],
             });
-            const { unmount } = renderWithProviders(_jsx(InputPrompt, { ...props }));
+            const { unmount } = renderWithProviders(_jsx(InputPrompt, { ...props }), {
+                uiActions,
+            });
             await waitFor(() => {
                 expect(mockedUseCommandCompletion).toHaveBeenCalledWith(mockBuffer, ['/test/project/src'], path.join('test', 'project', 'src'), mockSlashCommands, mockCommandContext, false, false, expect.any(Object));
             });
@@ -982,7 +1045,6 @@ describe('InputPrompt', () => {
         beforeEach(() => {
             vi.useFakeTimers();
             mockedUseKittyKeyboardProtocol.mockReturnValue({
-                supported: false,
                 enabled: false,
                 checking: false,
             });
@@ -1047,7 +1109,6 @@ describe('InputPrompt', () => {
             {
                 name: 'kitty',
                 setup: () => mockedUseKittyKeyboardProtocol.mockReturnValue({
-                    supported: true,
                     enabled: true,
                     checking: false,
                 }),
@@ -1540,7 +1601,7 @@ describe('InputPrompt', () => {
             ];
             props.buffer.visualCursor = [0, 11];
             props.buffer.visualScrollRow = 0;
-            const { stdin, stdout, unmount } = renderWithProviders(_jsx(InputPrompt, { ...props }), { mouseEventsEnabled: true });
+            const { stdin, stdout, unmount } = renderWithProviders(_jsx(InputPrompt, { ...props }), { mouseEventsEnabled: true, uiActions });
             // Wait for initial render
             await waitFor(() => {
                 expect(stdout.lastFrame()).toContain('hello world');
@@ -1555,6 +1616,25 @@ describe('InputPrompt', () => {
             });
             unmount();
         });
+        it('should unfocus embedded shell on click', async () => {
+            props.buffer.text = 'hello';
+            props.buffer.lines = ['hello'];
+            props.buffer.viewportVisualLines = ['hello'];
+            props.buffer.visualToLogicalMap = [[0, 0]];
+            props.isEmbeddedShellFocused = true;
+            const { stdin, stdout, unmount } = renderWithProviders(_jsx(InputPrompt, { ...props }), { mouseEventsEnabled: true, uiActions });
+            await waitFor(() => {
+                expect(stdout.lastFrame()).toContain('hello');
+            });
+            await act(async () => {
+                // Click somewhere in the prompt
+                stdin.write(`\x1b[<0;5;2M`);
+            });
+            await waitFor(() => {
+                expect(mockSetEmbeddedShellFocused).toHaveBeenCalledWith(false);
+            });
+            unmount();
+        });
     });
     describe('queued message editing', () => {
         it('should load all queued messages when up arrow is pressed with empty input', async () => {