mirror of
https://github.com/google-gemini/gemini-cli.git
synced 2026-03-10 22:21:22 -07:00
1083 lines
29 KiB
TypeScript
1083 lines
29 KiB
TypeScript
/**
|
|
* @license
|
|
* Copyright 2025 Google LLC
|
|
* SPDX-License-Identifier: Apache-2.0
|
|
*/
|
|
|
|
import type React from 'react';
|
|
import { renderHook, act, waitFor } from '@testing-library/react';
|
|
import type { Mock } from 'vitest';
|
|
import { vi } from 'vitest';
|
|
import type { Key } from './KeypressContext.js';
|
|
import {
|
|
KeypressProvider,
|
|
useKeypressContext,
|
|
DRAG_COMPLETION_TIMEOUT_MS,
|
|
// CSI_END_O,
|
|
// SS3_END,
|
|
SINGLE_QUOTE,
|
|
DOUBLE_QUOTE,
|
|
} from './KeypressContext.js';
|
|
import { useStdin } from 'ink';
|
|
import { EventEmitter } from 'node:events';
|
|
|
|
// Mock the 'ink' module to control stdin
|
|
vi.mock('ink', async (importOriginal) => {
|
|
const original = await importOriginal<typeof import('ink')>();
|
|
return {
|
|
...original,
|
|
useStdin: vi.fn(),
|
|
};
|
|
});
|
|
|
|
class MockStdin extends EventEmitter {
|
|
isTTY = true;
|
|
setRawMode = vi.fn();
|
|
override on = this.addListener;
|
|
override removeListener = super.removeListener;
|
|
write = vi.fn();
|
|
resume = vi.fn();
|
|
pause = vi.fn();
|
|
|
|
// Helper to simulate a keypress event
|
|
pressKey(key: Partial<Key>) {
|
|
this.emit('keypress', null, key);
|
|
}
|
|
|
|
// Helper to simulate a kitty protocol sequence
|
|
sendKittySequence(sequence: string) {
|
|
this.emit('data', Buffer.from(sequence));
|
|
}
|
|
|
|
// Helper to simulate a paste event
|
|
sendPaste(text: string) {
|
|
const PASTE_MODE_PREFIX = `\x1b[200~`;
|
|
const PASTE_MODE_SUFFIX = `\x1b[201~`;
|
|
this.emit('data', Buffer.from(PASTE_MODE_PREFIX));
|
|
this.emit('data', Buffer.from(text));
|
|
this.emit('data', Buffer.from(PASTE_MODE_SUFFIX));
|
|
}
|
|
}
|
|
|
|
describe('KeypressContext - Kitty Protocol', () => {
|
|
let stdin: MockStdin;
|
|
const mockSetRawMode = vi.fn();
|
|
|
|
const wrapper = ({
|
|
children,
|
|
kittyProtocolEnabled = true,
|
|
}: {
|
|
children: React.ReactNode;
|
|
kittyProtocolEnabled?: boolean;
|
|
}) => (
|
|
<KeypressProvider kittyProtocolEnabled={kittyProtocolEnabled}>
|
|
{children}
|
|
</KeypressProvider>
|
|
);
|
|
|
|
beforeEach(() => {
|
|
vi.clearAllMocks();
|
|
stdin = new MockStdin();
|
|
(useStdin as Mock).mockReturnValue({
|
|
stdin,
|
|
setRawMode: mockSetRawMode,
|
|
});
|
|
});
|
|
|
|
describe('Enter key handling', () => {
|
|
it('should recognize regular enter key (keycode 13) in kitty protocol', async () => {
|
|
const keyHandler = vi.fn();
|
|
|
|
const { result } = renderHook(() => useKeypressContext(), {
|
|
wrapper: ({ children }) =>
|
|
wrapper({ children, kittyProtocolEnabled: true }),
|
|
});
|
|
|
|
act(() => {
|
|
result.current.subscribe(keyHandler);
|
|
});
|
|
|
|
// Send kitty protocol sequence for regular enter: ESC[13u
|
|
act(() => {
|
|
stdin.sendKittySequence(`\x1b[13u`);
|
|
});
|
|
|
|
expect(keyHandler).toHaveBeenCalledWith(
|
|
expect.objectContaining({
|
|
name: 'return',
|
|
kittyProtocol: true,
|
|
ctrl: false,
|
|
meta: false,
|
|
shift: false,
|
|
}),
|
|
);
|
|
});
|
|
|
|
it('should recognize numpad enter key (keycode 57414) in kitty protocol', async () => {
|
|
const keyHandler = vi.fn();
|
|
|
|
const { result } = renderHook(() => useKeypressContext(), {
|
|
wrapper: ({ children }) =>
|
|
wrapper({ children, kittyProtocolEnabled: true }),
|
|
});
|
|
|
|
act(() => {
|
|
result.current.subscribe(keyHandler);
|
|
});
|
|
|
|
// Send kitty protocol sequence for numpad enter: ESC[57414u
|
|
act(() => {
|
|
stdin.sendKittySequence(`\x1b[57414u`);
|
|
});
|
|
|
|
expect(keyHandler).toHaveBeenCalledWith(
|
|
expect.objectContaining({
|
|
name: 'return',
|
|
kittyProtocol: true,
|
|
ctrl: false,
|
|
meta: false,
|
|
shift: false,
|
|
}),
|
|
);
|
|
});
|
|
|
|
it('should handle numpad enter with modifiers', async () => {
|
|
const keyHandler = vi.fn();
|
|
|
|
const { result } = renderHook(() => useKeypressContext(), {
|
|
wrapper: ({ children }) =>
|
|
wrapper({ children, kittyProtocolEnabled: true }),
|
|
});
|
|
|
|
act(() => {
|
|
result.current.subscribe(keyHandler);
|
|
});
|
|
|
|
// Send kitty protocol sequence for numpad enter with Shift (modifier 2): ESC[57414;2u
|
|
act(() => {
|
|
stdin.sendKittySequence(`\x1b[57414;2u`);
|
|
});
|
|
|
|
expect(keyHandler).toHaveBeenCalledWith(
|
|
expect.objectContaining({
|
|
name: 'return',
|
|
kittyProtocol: true,
|
|
ctrl: false,
|
|
meta: false,
|
|
shift: true,
|
|
}),
|
|
);
|
|
});
|
|
|
|
it('should handle numpad enter with Ctrl modifier', async () => {
|
|
const keyHandler = vi.fn();
|
|
|
|
const { result } = renderHook(() => useKeypressContext(), {
|
|
wrapper: ({ children }) =>
|
|
wrapper({ children, kittyProtocolEnabled: true }),
|
|
});
|
|
|
|
act(() => {
|
|
result.current.subscribe(keyHandler);
|
|
});
|
|
|
|
// Send kitty protocol sequence for numpad enter with Ctrl (modifier 5): ESC[57414;5u
|
|
act(() => {
|
|
stdin.sendKittySequence(`\x1b[57414;5u`);
|
|
});
|
|
|
|
expect(keyHandler).toHaveBeenCalledWith(
|
|
expect.objectContaining({
|
|
name: 'return',
|
|
kittyProtocol: true,
|
|
ctrl: true,
|
|
meta: false,
|
|
shift: false,
|
|
}),
|
|
);
|
|
});
|
|
|
|
it('should handle numpad enter with Alt modifier', async () => {
|
|
const keyHandler = vi.fn();
|
|
|
|
const { result } = renderHook(() => useKeypressContext(), {
|
|
wrapper: ({ children }) =>
|
|
wrapper({ children, kittyProtocolEnabled: true }),
|
|
});
|
|
|
|
act(() => {
|
|
result.current.subscribe(keyHandler);
|
|
});
|
|
|
|
// Send kitty protocol sequence for numpad enter with Alt (modifier 3): ESC[57414;3u
|
|
act(() => {
|
|
stdin.sendKittySequence(`\x1b[57414;3u`);
|
|
});
|
|
|
|
expect(keyHandler).toHaveBeenCalledWith(
|
|
expect.objectContaining({
|
|
name: 'return',
|
|
kittyProtocol: true,
|
|
ctrl: false,
|
|
meta: true,
|
|
shift: false,
|
|
}),
|
|
);
|
|
});
|
|
|
|
it('should not process kitty sequences when kitty protocol is disabled', async () => {
|
|
const keyHandler = vi.fn();
|
|
|
|
const { result } = renderHook(() => useKeypressContext(), {
|
|
wrapper: ({ children }) =>
|
|
wrapper({ children, kittyProtocolEnabled: false }),
|
|
});
|
|
|
|
act(() => {
|
|
result.current.subscribe(keyHandler);
|
|
});
|
|
|
|
// Send kitty protocol sequence for numpad enter
|
|
act(() => {
|
|
stdin.sendKittySequence(`\x1b[57414u`);
|
|
});
|
|
|
|
// When kitty protocol is disabled, the sequence should be passed through
|
|
// as individual keypresses, not recognized as a single enter key
|
|
expect(keyHandler).not.toHaveBeenCalledWith(
|
|
expect.objectContaining({
|
|
name: 'return',
|
|
kittyProtocol: true,
|
|
}),
|
|
);
|
|
});
|
|
});
|
|
|
|
describe('Escape key handling', () => {
|
|
it('should recognize escape key (keycode 27) in kitty protocol', async () => {
|
|
const keyHandler = vi.fn();
|
|
|
|
const { result } = renderHook(() => useKeypressContext(), {
|
|
wrapper: ({ children }) =>
|
|
wrapper({ children, kittyProtocolEnabled: true }),
|
|
});
|
|
|
|
act(() => {
|
|
result.current.subscribe(keyHandler);
|
|
});
|
|
|
|
// Send kitty protocol sequence for escape: ESC[27u
|
|
act(() => {
|
|
stdin.sendKittySequence('\x1b[27u');
|
|
});
|
|
|
|
expect(keyHandler).toHaveBeenCalledWith(
|
|
expect.objectContaining({
|
|
name: 'escape',
|
|
kittyProtocol: true,
|
|
}),
|
|
);
|
|
});
|
|
});
|
|
|
|
describe('Tab and Backspace handling', () => {
|
|
it('should recognize Tab key in kitty protocol', async () => {
|
|
const keyHandler = vi.fn();
|
|
const { result } = renderHook(() => useKeypressContext(), { wrapper });
|
|
act(() => result.current.subscribe(keyHandler));
|
|
|
|
act(() => {
|
|
stdin.sendKittySequence(`\x1b[9u`);
|
|
});
|
|
|
|
expect(keyHandler).toHaveBeenCalledWith(
|
|
expect.objectContaining({
|
|
name: 'tab',
|
|
kittyProtocol: true,
|
|
shift: false,
|
|
}),
|
|
);
|
|
});
|
|
|
|
it('should recognize Shift+Tab in kitty protocol', async () => {
|
|
const keyHandler = vi.fn();
|
|
const { result } = renderHook(() => useKeypressContext(), { wrapper });
|
|
act(() => result.current.subscribe(keyHandler));
|
|
|
|
// Modifier 2 is Shift
|
|
act(() => {
|
|
stdin.sendKittySequence(`\x1b[9;2u`);
|
|
});
|
|
|
|
expect(keyHandler).toHaveBeenCalledWith(
|
|
expect.objectContaining({
|
|
name: 'tab',
|
|
kittyProtocol: true,
|
|
shift: true,
|
|
}),
|
|
);
|
|
});
|
|
|
|
it('should recognize Backspace key in kitty protocol', async () => {
|
|
const keyHandler = vi.fn();
|
|
const { result } = renderHook(() => useKeypressContext(), { wrapper });
|
|
act(() => result.current.subscribe(keyHandler));
|
|
|
|
act(() => {
|
|
stdin.sendKittySequence(`\x1b[127u`);
|
|
});
|
|
|
|
expect(keyHandler).toHaveBeenCalledWith(
|
|
expect.objectContaining({
|
|
name: 'backspace',
|
|
kittyProtocol: true,
|
|
meta: false,
|
|
}),
|
|
);
|
|
});
|
|
|
|
it('should recognize Option+Backspace in kitty protocol', async () => {
|
|
const keyHandler = vi.fn();
|
|
const { result } = renderHook(() => useKeypressContext(), { wrapper });
|
|
act(() => result.current.subscribe(keyHandler));
|
|
|
|
// Modifier 3 is Alt/Option
|
|
act(() => {
|
|
stdin.sendKittySequence(`\x1b[127;3u`);
|
|
});
|
|
|
|
expect(keyHandler).toHaveBeenCalledWith(
|
|
expect.objectContaining({
|
|
name: 'backspace',
|
|
kittyProtocol: true,
|
|
meta: true,
|
|
}),
|
|
);
|
|
});
|
|
|
|
it('should recognize Ctrl+Backspace in kitty protocol', async () => {
|
|
const keyHandler = vi.fn();
|
|
const { result } = renderHook(() => useKeypressContext(), { wrapper });
|
|
act(() => result.current.subscribe(keyHandler));
|
|
|
|
// Modifier 5 is Ctrl
|
|
act(() => {
|
|
stdin.sendKittySequence(`\x1b[127;5u`);
|
|
});
|
|
|
|
expect(keyHandler).toHaveBeenCalledWith(
|
|
expect.objectContaining({
|
|
name: 'backspace',
|
|
kittyProtocol: true,
|
|
ctrl: true,
|
|
}),
|
|
);
|
|
});
|
|
});
|
|
|
|
describe('paste mode', () => {
|
|
it('should handle multiline paste as a single event', async () => {
|
|
const keyHandler = vi.fn();
|
|
const pastedText = 'This \n is \n a \n multiline \n paste.';
|
|
|
|
const { result } = renderHook(() => useKeypressContext(), {
|
|
wrapper,
|
|
});
|
|
|
|
act(() => {
|
|
result.current.subscribe(keyHandler);
|
|
});
|
|
|
|
// Simulate a bracketed paste event
|
|
act(() => {
|
|
stdin.sendPaste(pastedText);
|
|
});
|
|
|
|
await waitFor(() => {
|
|
// Expect the handler to be called exactly once for the entire paste
|
|
expect(keyHandler).toHaveBeenCalledTimes(1);
|
|
});
|
|
|
|
// Verify the single event contains the full pasted text
|
|
expect(keyHandler).toHaveBeenCalledWith(
|
|
expect.objectContaining({
|
|
paste: true,
|
|
sequence: pastedText,
|
|
}),
|
|
);
|
|
});
|
|
});
|
|
|
|
describe('debug keystroke logging', () => {
|
|
let consoleLogSpy: ReturnType<typeof vi.spyOn>;
|
|
let consoleWarnSpy: ReturnType<typeof vi.spyOn>;
|
|
|
|
beforeEach(() => {
|
|
consoleLogSpy = vi.spyOn(console, 'log').mockImplementation(() => {});
|
|
consoleWarnSpy = vi.spyOn(console, 'warn').mockImplementation(() => {});
|
|
});
|
|
|
|
afterEach(() => {
|
|
consoleLogSpy.mockRestore();
|
|
consoleWarnSpy.mockRestore();
|
|
});
|
|
|
|
it('should not log keystrokes when debugKeystrokeLogging is false', async () => {
|
|
const keyHandler = vi.fn();
|
|
|
|
const wrapper = ({ children }: { children: React.ReactNode }) => (
|
|
<KeypressProvider
|
|
kittyProtocolEnabled={true}
|
|
debugKeystrokeLogging={false}
|
|
>
|
|
{children}
|
|
</KeypressProvider>
|
|
);
|
|
|
|
const { result } = renderHook(() => useKeypressContext(), { wrapper });
|
|
|
|
act(() => {
|
|
result.current.subscribe(keyHandler);
|
|
});
|
|
|
|
// Send a kitty sequence
|
|
act(() => {
|
|
stdin.sendKittySequence('\x1b[27u');
|
|
});
|
|
|
|
expect(keyHandler).toHaveBeenCalled();
|
|
expect(consoleLogSpy).not.toHaveBeenCalledWith(
|
|
expect.stringContaining('[DEBUG] Kitty'),
|
|
);
|
|
});
|
|
|
|
it('should log kitty buffer accumulation when debugKeystrokeLogging is true', async () => {
|
|
const keyHandler = vi.fn();
|
|
|
|
const wrapper = ({ children }: { children: React.ReactNode }) => (
|
|
<KeypressProvider
|
|
kittyProtocolEnabled={true}
|
|
debugKeystrokeLogging={true}
|
|
>
|
|
{children}
|
|
</KeypressProvider>
|
|
);
|
|
|
|
const { result } = renderHook(() => useKeypressContext(), { wrapper });
|
|
|
|
act(() => {
|
|
result.current.subscribe(keyHandler);
|
|
});
|
|
|
|
// Send a complete kitty sequence for escape
|
|
act(() => {
|
|
stdin.sendKittySequence('\x1b[27u');
|
|
});
|
|
|
|
expect(consoleLogSpy).toHaveBeenCalledWith(
|
|
'[DEBUG] Kitty buffer accumulating:',
|
|
expect.stringContaining('\x1b[27u'),
|
|
);
|
|
const parsedCall = consoleLogSpy.mock.calls.find(
|
|
(args) =>
|
|
typeof args[0] === 'string' &&
|
|
args[0].includes('[DEBUG] Kitty sequence parsed successfully'),
|
|
);
|
|
expect(parsedCall).toBeTruthy();
|
|
expect(parsedCall?.[1]).toEqual(expect.stringContaining('\x1b[27u'));
|
|
});
|
|
|
|
it('should log kitty buffer overflow when debugKeystrokeLogging is true', async () => {
|
|
const keyHandler = vi.fn();
|
|
|
|
const wrapper = ({ children }: { children: React.ReactNode }) => (
|
|
<KeypressProvider
|
|
kittyProtocolEnabled={true}
|
|
debugKeystrokeLogging={true}
|
|
>
|
|
{children}
|
|
</KeypressProvider>
|
|
);
|
|
|
|
const { result } = renderHook(() => useKeypressContext(), { wrapper });
|
|
|
|
act(() => {
|
|
result.current.subscribe(keyHandler);
|
|
});
|
|
|
|
// Send an invalid long sequence to trigger overflow
|
|
const longInvalidSequence = '\x1b[' + 'x'.repeat(100);
|
|
act(() => {
|
|
stdin.sendKittySequence(longInvalidSequence);
|
|
});
|
|
|
|
expect(consoleLogSpy).toHaveBeenCalledWith(
|
|
'[DEBUG] Kitty buffer overflow, clearing:',
|
|
expect.any(String),
|
|
);
|
|
});
|
|
|
|
it('should log kitty buffer clear on Ctrl+C when debugKeystrokeLogging is true', async () => {
|
|
const keyHandler = vi.fn();
|
|
|
|
const wrapper = ({ children }: { children: React.ReactNode }) => (
|
|
<KeypressProvider
|
|
kittyProtocolEnabled={true}
|
|
debugKeystrokeLogging={true}
|
|
>
|
|
{children}
|
|
</KeypressProvider>
|
|
);
|
|
|
|
const { result } = renderHook(() => useKeypressContext(), { wrapper });
|
|
|
|
act(() => {
|
|
result.current.subscribe(keyHandler);
|
|
});
|
|
|
|
// Send incomplete kitty sequence
|
|
act(() => {
|
|
stdin.pressKey({
|
|
name: undefined,
|
|
ctrl: false,
|
|
meta: false,
|
|
shift: false,
|
|
sequence: '\x1b[1',
|
|
});
|
|
});
|
|
|
|
// Send Ctrl+C
|
|
act(() => {
|
|
stdin.pressKey({
|
|
name: 'c',
|
|
ctrl: true,
|
|
meta: false,
|
|
shift: false,
|
|
sequence: '\x03',
|
|
});
|
|
});
|
|
|
|
expect(consoleLogSpy).toHaveBeenCalledWith(
|
|
'[DEBUG] Kitty buffer cleared on Ctrl+C:',
|
|
'\x1b[1',
|
|
);
|
|
|
|
// Verify Ctrl+C was handled
|
|
expect(keyHandler).toHaveBeenCalledWith(
|
|
expect.objectContaining({
|
|
name: 'c',
|
|
ctrl: true,
|
|
}),
|
|
);
|
|
});
|
|
|
|
it('should show char codes when debugKeystrokeLogging is true even without debug mode', async () => {
|
|
const keyHandler = vi.fn();
|
|
|
|
const wrapper = ({ children }: { children: React.ReactNode }) => (
|
|
<KeypressProvider
|
|
kittyProtocolEnabled={true}
|
|
debugKeystrokeLogging={true}
|
|
>
|
|
{children}
|
|
</KeypressProvider>
|
|
);
|
|
|
|
const { result } = renderHook(() => useKeypressContext(), { wrapper });
|
|
|
|
act(() => {
|
|
result.current.subscribe(keyHandler);
|
|
});
|
|
|
|
// Send incomplete kitty sequence
|
|
const sequence = '\x1b[12';
|
|
act(() => {
|
|
stdin.pressKey({
|
|
name: undefined,
|
|
ctrl: false,
|
|
meta: false,
|
|
shift: false,
|
|
sequence,
|
|
});
|
|
});
|
|
|
|
// Verify debug logging for accumulation
|
|
expect(consoleLogSpy).toHaveBeenCalledWith(
|
|
'[DEBUG] Kitty buffer accumulating:',
|
|
sequence,
|
|
);
|
|
|
|
// Verify warning for char codes
|
|
expect(consoleWarnSpy).toHaveBeenCalledWith(
|
|
'Kitty sequence buffer has char codes:',
|
|
[27, 91, 49, 50],
|
|
);
|
|
});
|
|
});
|
|
|
|
describe('Parameterized functional keys', () => {
|
|
it.each([
|
|
// Parameterized
|
|
{ sequence: `\x1b[1;2H`, expected: { name: 'home', shift: true } },
|
|
{ sequence: `\x1b[1;5F`, expected: { name: 'end', ctrl: true } },
|
|
{ sequence: `\x1b[1;1P`, expected: { name: 'f1' } },
|
|
{ sequence: `\x1b[1;3Q`, expected: { name: 'f2', meta: true } },
|
|
{ sequence: `\x1b[3~`, expected: { name: 'delete' } },
|
|
{ sequence: `\x1b[5~`, expected: { name: 'pageup' } },
|
|
{ sequence: `\x1b[6~`, expected: { name: 'pagedown' } },
|
|
{ sequence: `\x1b[1~`, expected: { name: 'home' } },
|
|
{ sequence: `\x1b[4~`, expected: { name: 'end' } },
|
|
{ sequence: `\x1b[2~`, expected: { name: 'insert' } },
|
|
// Legacy Arrows
|
|
{
|
|
sequence: `\x1b[A`,
|
|
expected: { name: 'up', ctrl: false, meta: false, shift: false },
|
|
},
|
|
{
|
|
sequence: `\x1b[B`,
|
|
expected: { name: 'down', ctrl: false, meta: false, shift: false },
|
|
},
|
|
{
|
|
sequence: `\x1b[C`,
|
|
expected: { name: 'right', ctrl: false, meta: false, shift: false },
|
|
},
|
|
{
|
|
sequence: `\x1b[D`,
|
|
expected: { name: 'left', ctrl: false, meta: false, shift: false },
|
|
},
|
|
// Legacy Home/End
|
|
{
|
|
sequence: `\x1b[H`,
|
|
expected: { name: 'home', ctrl: false, meta: false, shift: false },
|
|
},
|
|
{
|
|
sequence: `\x1b[F`,
|
|
expected: { name: 'end', ctrl: false, meta: false, shift: false },
|
|
},
|
|
])(
|
|
'should recognize sequence "$sequence" as $expected.name',
|
|
({ sequence, expected }) => {
|
|
const keyHandler = vi.fn();
|
|
const { result } = renderHook(() => useKeypressContext(), { wrapper });
|
|
act(() => result.current.subscribe(keyHandler));
|
|
|
|
act(() => stdin.sendKittySequence(sequence));
|
|
|
|
expect(keyHandler).toHaveBeenCalledWith(
|
|
expect.objectContaining(expected),
|
|
);
|
|
},
|
|
);
|
|
});
|
|
|
|
describe('Shift+Tab forms', () => {
|
|
it.each([
|
|
{ sequence: `\x1b[Z`, description: 'legacy reverse Tab' },
|
|
{ sequence: `\x1b[1;2Z`, description: 'parameterized reverse Tab' },
|
|
])(
|
|
'should recognize $description "$sequence" as Shift+Tab',
|
|
({ sequence }) => {
|
|
const keyHandler = vi.fn();
|
|
const { result } = renderHook(() => useKeypressContext(), { wrapper });
|
|
act(() => result.current.subscribe(keyHandler));
|
|
|
|
act(() => stdin.sendKittySequence(sequence));
|
|
expect(keyHandler).toHaveBeenCalledWith(
|
|
expect.objectContaining({ name: 'tab', shift: true }),
|
|
);
|
|
},
|
|
);
|
|
});
|
|
|
|
describe('Double-tap and batching', () => {
|
|
it('should emit two delete events for double-tap CSI[3~', async () => {
|
|
const keyHandler = vi.fn();
|
|
const { result } = renderHook(() => useKeypressContext(), { wrapper });
|
|
act(() => result.current.subscribe(keyHandler));
|
|
|
|
act(() => stdin.sendKittySequence(`\x1b[3~`));
|
|
act(() => stdin.sendKittySequence(`\x1b[3~`));
|
|
|
|
expect(keyHandler).toHaveBeenNthCalledWith(
|
|
1,
|
|
expect.objectContaining({ name: 'delete' }),
|
|
);
|
|
expect(keyHandler).toHaveBeenNthCalledWith(
|
|
2,
|
|
expect.objectContaining({ name: 'delete' }),
|
|
);
|
|
});
|
|
|
|
it('should parse two concatenated tilde-coded sequences in one chunk', async () => {
|
|
const keyHandler = vi.fn();
|
|
const { result } = renderHook(() => useKeypressContext(), { wrapper });
|
|
act(() => result.current.subscribe(keyHandler));
|
|
|
|
act(() => stdin.sendKittySequence(`\x1b[3~\x1b[5~`));
|
|
|
|
expect(keyHandler).toHaveBeenCalledWith(
|
|
expect.objectContaining({ name: 'delete' }),
|
|
);
|
|
expect(keyHandler).toHaveBeenCalledWith(
|
|
expect.objectContaining({ name: 'pageup' }),
|
|
);
|
|
});
|
|
|
|
it('should ignore incomplete CSI then parse the next complete sequence', async () => {
|
|
const keyHandler = vi.fn();
|
|
const { result } = renderHook(() => useKeypressContext(), { wrapper });
|
|
act(() => result.current.subscribe(keyHandler));
|
|
|
|
// Incomplete ESC sequence then a complete Delete
|
|
act(() => {
|
|
// Provide an incomplete ESC sequence chunk with a real ESC character
|
|
stdin.pressKey({
|
|
name: undefined,
|
|
ctrl: false,
|
|
meta: false,
|
|
shift: false,
|
|
sequence: '\x1b[1;',
|
|
});
|
|
});
|
|
act(() => stdin.sendKittySequence(`\x1b[3~`));
|
|
|
|
expect(keyHandler).toHaveBeenCalledTimes(1);
|
|
expect(keyHandler).toHaveBeenCalledWith(
|
|
expect.objectContaining({ name: 'delete' }),
|
|
);
|
|
});
|
|
});
|
|
});
|
|
|
|
describe('Drag and Drop Handling', () => {
|
|
let stdin: MockStdin;
|
|
const mockSetRawMode = vi.fn();
|
|
|
|
const wrapper = ({ children }: { children: React.ReactNode }) => (
|
|
<KeypressProvider kittyProtocolEnabled={true}>{children}</KeypressProvider>
|
|
);
|
|
|
|
beforeEach(() => {
|
|
vi.clearAllMocks();
|
|
vi.useFakeTimers();
|
|
stdin = new MockStdin();
|
|
(useStdin as Mock).mockReturnValue({
|
|
stdin,
|
|
setRawMode: mockSetRawMode,
|
|
});
|
|
});
|
|
|
|
afterEach(() => {
|
|
vi.useRealTimers();
|
|
});
|
|
|
|
describe('drag start by quotes', () => {
|
|
it('should start collecting when single quote arrives and not broadcast immediately', async () => {
|
|
const keyHandler = vi.fn();
|
|
|
|
const { result } = renderHook(() => useKeypressContext(), { wrapper });
|
|
|
|
act(() => {
|
|
result.current.subscribe(keyHandler);
|
|
});
|
|
|
|
act(() => {
|
|
stdin.pressKey({
|
|
name: undefined,
|
|
ctrl: false,
|
|
meta: false,
|
|
shift: false,
|
|
paste: false,
|
|
sequence: SINGLE_QUOTE,
|
|
});
|
|
});
|
|
|
|
expect(keyHandler).not.toHaveBeenCalled();
|
|
});
|
|
|
|
it('should start collecting when double quote arrives and not broadcast immediately', async () => {
|
|
const keyHandler = vi.fn();
|
|
|
|
const { result } = renderHook(() => useKeypressContext(), { wrapper });
|
|
|
|
act(() => {
|
|
result.current.subscribe(keyHandler);
|
|
});
|
|
|
|
act(() => {
|
|
stdin.pressKey({
|
|
name: undefined,
|
|
ctrl: false,
|
|
meta: false,
|
|
shift: false,
|
|
paste: false,
|
|
sequence: DOUBLE_QUOTE,
|
|
});
|
|
});
|
|
|
|
expect(keyHandler).not.toHaveBeenCalled();
|
|
});
|
|
});
|
|
|
|
describe('drag collection and completion', () => {
|
|
it('should collect single character inputs during drag mode', async () => {
|
|
const keyHandler = vi.fn();
|
|
|
|
const { result } = renderHook(() => useKeypressContext(), { wrapper });
|
|
|
|
act(() => {
|
|
result.current.subscribe(keyHandler);
|
|
});
|
|
|
|
// Start by single quote
|
|
act(() => {
|
|
stdin.pressKey({
|
|
name: undefined,
|
|
ctrl: false,
|
|
meta: false,
|
|
shift: false,
|
|
paste: false,
|
|
sequence: SINGLE_QUOTE,
|
|
});
|
|
});
|
|
|
|
// Send single character
|
|
act(() => {
|
|
stdin.pressKey({
|
|
name: undefined,
|
|
ctrl: false,
|
|
meta: false,
|
|
shift: false,
|
|
paste: false,
|
|
sequence: 'a',
|
|
});
|
|
});
|
|
|
|
// Character should not be immediately broadcast
|
|
expect(keyHandler).not.toHaveBeenCalled();
|
|
|
|
// Fast-forward to completion timeout
|
|
act(() => {
|
|
vi.advanceTimersByTime(DRAG_COMPLETION_TIMEOUT_MS + 10);
|
|
});
|
|
|
|
// Should broadcast the collected path as paste (includes starting quote)
|
|
expect(keyHandler).toHaveBeenCalledWith(
|
|
expect.objectContaining({
|
|
name: '',
|
|
paste: true,
|
|
sequence: `${SINGLE_QUOTE}a`,
|
|
}),
|
|
);
|
|
});
|
|
|
|
it('should collect multiple characters and complete on timeout', async () => {
|
|
const keyHandler = vi.fn();
|
|
|
|
const { result } = renderHook(() => useKeypressContext(), { wrapper });
|
|
|
|
act(() => {
|
|
result.current.subscribe(keyHandler);
|
|
});
|
|
|
|
// Start by single quote
|
|
act(() => {
|
|
stdin.pressKey({
|
|
name: undefined,
|
|
ctrl: false,
|
|
meta: false,
|
|
shift: false,
|
|
paste: false,
|
|
sequence: SINGLE_QUOTE,
|
|
});
|
|
});
|
|
|
|
// Send multiple characters
|
|
act(() => {
|
|
stdin.pressKey({
|
|
name: undefined,
|
|
ctrl: false,
|
|
meta: false,
|
|
shift: false,
|
|
paste: false,
|
|
sequence: 'p',
|
|
});
|
|
});
|
|
|
|
act(() => {
|
|
stdin.pressKey({
|
|
name: undefined,
|
|
ctrl: false,
|
|
meta: false,
|
|
shift: false,
|
|
paste: false,
|
|
sequence: 'a',
|
|
});
|
|
});
|
|
|
|
act(() => {
|
|
stdin.pressKey({
|
|
name: undefined,
|
|
ctrl: false,
|
|
meta: false,
|
|
shift: false,
|
|
paste: false,
|
|
sequence: 't',
|
|
});
|
|
});
|
|
|
|
act(() => {
|
|
stdin.pressKey({
|
|
name: undefined,
|
|
ctrl: false,
|
|
meta: false,
|
|
shift: false,
|
|
paste: false,
|
|
sequence: 'h',
|
|
});
|
|
});
|
|
|
|
// Characters should not be immediately broadcast
|
|
expect(keyHandler).not.toHaveBeenCalled();
|
|
|
|
// Fast-forward to completion timeout
|
|
act(() => {
|
|
vi.advanceTimersByTime(DRAG_COMPLETION_TIMEOUT_MS + 10);
|
|
});
|
|
|
|
// Should broadcast the collected path as paste (includes starting quote)
|
|
expect(keyHandler).toHaveBeenCalledWith(
|
|
expect.objectContaining({
|
|
name: '',
|
|
paste: true,
|
|
sequence: `${SINGLE_QUOTE}path`,
|
|
}),
|
|
);
|
|
});
|
|
});
|
|
});
|
|
|
|
describe('Terminal-specific Alt+key combinations', () => {
|
|
let stdin: MockStdin;
|
|
const mockSetRawMode = vi.fn();
|
|
|
|
const wrapper = ({ children }: { children: React.ReactNode }) => (
|
|
<KeypressProvider kittyProtocolEnabled={true}>{children}</KeypressProvider>
|
|
);
|
|
|
|
beforeEach(() => {
|
|
vi.clearAllMocks();
|
|
stdin = new MockStdin();
|
|
(useStdin as Mock).mockReturnValue({
|
|
stdin,
|
|
setRawMode: mockSetRawMode,
|
|
});
|
|
});
|
|
|
|
// Terminals to test
|
|
const terminals = ['iTerm2', 'Ghostty', 'MacTerminal', 'VSCodeTerminal'];
|
|
|
|
// Key mappings: letter -> [keycode, accented character, shouldHaveMeta]
|
|
// Note: µ (mu) is sent with meta:false on iTerm2/VSCode
|
|
const keys: Record<string, [number, string, boolean]> = {
|
|
a: [97, 'å', true],
|
|
o: [111, 'ø', true],
|
|
m: [109, 'µ', false],
|
|
};
|
|
|
|
it.each(
|
|
terminals.flatMap((terminal) =>
|
|
Object.entries(keys).map(
|
|
([key, [keycode, accentedChar, shouldHaveMeta]]) => {
|
|
if (terminal === 'Ghostty') {
|
|
// Ghostty uses kitty protocol sequences
|
|
return {
|
|
terminal,
|
|
key,
|
|
kittySequence: `\x1b[${keycode};3u`,
|
|
expected: {
|
|
name: key,
|
|
ctrl: false,
|
|
meta: true,
|
|
shift: false,
|
|
paste: false,
|
|
kittyProtocol: true,
|
|
},
|
|
};
|
|
} else if (terminal === 'MacTerminal') {
|
|
// Mac Terminal sends ESC + letter
|
|
return {
|
|
terminal,
|
|
key,
|
|
input: {
|
|
sequence: `\x1b${key}`,
|
|
name: key,
|
|
ctrl: false,
|
|
meta: true,
|
|
shift: false,
|
|
paste: false,
|
|
},
|
|
expected: {
|
|
sequence: `\x1b${key}`,
|
|
name: key,
|
|
ctrl: false,
|
|
meta: true,
|
|
shift: false,
|
|
paste: false,
|
|
},
|
|
};
|
|
} else {
|
|
// iTerm2 and VSCode send accented characters (å, ø, µ)
|
|
// Note: µ comes with meta:false but gets converted to m with meta:true
|
|
return {
|
|
terminal,
|
|
key,
|
|
input: {
|
|
name: key,
|
|
ctrl: false,
|
|
meta: shouldHaveMeta,
|
|
shift: false,
|
|
paste: false,
|
|
sequence: accentedChar,
|
|
},
|
|
expected: {
|
|
name: key,
|
|
ctrl: false,
|
|
meta: true, // Always expect meta:true after conversion
|
|
shift: false,
|
|
paste: false,
|
|
sequence: accentedChar,
|
|
},
|
|
};
|
|
}
|
|
},
|
|
),
|
|
),
|
|
)(
|
|
'should handle Alt+$key in $terminal',
|
|
({
|
|
kittySequence,
|
|
input,
|
|
expected,
|
|
}: {
|
|
kittySequence?: string;
|
|
input?: Partial<Key>;
|
|
expected: Partial<Key>;
|
|
}) => {
|
|
const keyHandler = vi.fn();
|
|
const { result } = renderHook(() => useKeypressContext(), { wrapper });
|
|
act(() => result.current.subscribe(keyHandler));
|
|
|
|
if (kittySequence) {
|
|
act(() => stdin.sendKittySequence(kittySequence));
|
|
} else if (input) {
|
|
act(() => stdin.pressKey(input));
|
|
}
|
|
|
|
expect(keyHandler).toHaveBeenCalledWith(
|
|
expect.objectContaining(expected),
|
|
);
|
|
},
|
|
);
|
|
});
|