mirror of
https://github.com/google-gemini/gemini-cli.git
synced 2026-03-11 14:40:52 -07:00
57c7f6faeb
Co-authored-by: Jacob Richman <jacob314@gmail.com> Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>
3604 lines
108 KiB
TypeScript
3604 lines
108 KiB
TypeScript
/**
|
||
* @license
|
||
* Copyright 2025 Google LLC
|
||
* SPDX-License-Identifier: Apache-2.0
|
||
*/
|
||
|
||
import { spawnSync } from 'node:child_process';
|
||
import fs from 'node:fs';
|
||
import os from 'node:os';
|
||
import pathMod from 'node:path';
|
||
import * as path from 'node:path';
|
||
import { useState, useCallback, useEffect, useMemo, useReducer } from 'react';
|
||
import { LRUCache } from 'mnemonist';
|
||
import {
|
||
coreEvents,
|
||
CoreEvent,
|
||
debugLogger,
|
||
unescapePath,
|
||
type EditorType,
|
||
getEditorCommand,
|
||
isGuiEditor,
|
||
} from '@google/gemini-cli-core';
|
||
import {
|
||
toCodePoints,
|
||
cpLen,
|
||
cpSlice,
|
||
stripUnsafeCharacters,
|
||
getCachedStringWidth,
|
||
} from '../../utils/textUtils.js';
|
||
import { parsePastedPaths } from '../../utils/clipboardUtils.js';
|
||
import type { Key } from '../../contexts/KeypressContext.js';
|
||
import { keyMatchers, Command } from '../../keyMatchers.js';
|
||
import type { VimAction } from './vim-buffer-actions.js';
|
||
import { handleVimAction } from './vim-buffer-actions.js';
|
||
import { LRU_BUFFER_PERF_CACHE_LIMIT } from '../../constants.js';
|
||
|
||
const LARGE_PASTE_LINE_THRESHOLD = 5;
|
||
const LARGE_PASTE_CHAR_THRESHOLD = 500;
|
||
|
||
// Regex to match paste placeholders like [Pasted Text: 6 lines] or [Pasted Text: 501 chars #2]
|
||
export const PASTED_TEXT_PLACEHOLDER_REGEX =
|
||
/\[Pasted Text: \d+ (?:lines|chars)(?: #\d+)?\]/g;
|
||
|
||
export type Direction =
|
||
| 'left'
|
||
| 'right'
|
||
| 'up'
|
||
| 'down'
|
||
| 'wordLeft'
|
||
| 'wordRight'
|
||
| 'home'
|
||
| 'end';
|
||
|
||
// Helper functions for line-based word navigation
|
||
export const isWordCharStrict = (char: string): boolean =>
|
||
/[\w\p{L}\p{N}]/u.test(char); // Matches a single character that is any Unicode letter, any Unicode number, or an underscore
|
||
|
||
export const isWhitespace = (char: string): boolean => /\s/.test(char);
|
||
|
||
// Check if a character is a combining mark (only diacritics for now)
|
||
export const isCombiningMark = (char: string): boolean => /\p{M}/u.test(char);
|
||
|
||
// Check if a character should be considered part of a word (including combining marks)
|
||
export const isWordCharWithCombining = (char: string): boolean =>
|
||
isWordCharStrict(char) || isCombiningMark(char);
|
||
|
||
// Get the script of a character (simplified for common scripts)
|
||
export const getCharScript = (char: string): string => {
|
||
if (/[\p{Script=Latin}]/u.test(char)) return 'latin'; // All Latin script chars including diacritics
|
||
if (/[\p{Script=Han}]/u.test(char)) return 'han'; // Chinese
|
||
if (/[\p{Script=Arabic}]/u.test(char)) return 'arabic';
|
||
if (/[\p{Script=Hiragana}]/u.test(char)) return 'hiragana';
|
||
if (/[\p{Script=Katakana}]/u.test(char)) return 'katakana';
|
||
if (/[\p{Script=Cyrillic}]/u.test(char)) return 'cyrillic';
|
||
return 'other';
|
||
};
|
||
|
||
// Check if two characters are from different scripts (indicating word boundary)
|
||
export const isDifferentScript = (char1: string, char2: string): boolean => {
|
||
if (!isWordCharStrict(char1) || !isWordCharStrict(char2)) return false;
|
||
return getCharScript(char1) !== getCharScript(char2);
|
||
};
|
||
|
||
// Find next word start within a line, starting from col
|
||
export const findNextWordStartInLine = (
|
||
line: string,
|
||
col: number,
|
||
): number | null => {
|
||
const chars = toCodePoints(line);
|
||
let i = col;
|
||
|
||
if (i >= chars.length) return null;
|
||
|
||
const currentChar = chars[i];
|
||
|
||
// Skip current word/sequence based on character type
|
||
if (isWordCharStrict(currentChar)) {
|
||
while (i < chars.length && isWordCharWithCombining(chars[i])) {
|
||
// Check for script boundary - if next character is from different script, stop here
|
||
if (
|
||
i + 1 < chars.length &&
|
||
isWordCharStrict(chars[i + 1]) &&
|
||
isDifferentScript(chars[i], chars[i + 1])
|
||
) {
|
||
i++; // Include current character
|
||
break; // Stop at script boundary
|
||
}
|
||
i++;
|
||
}
|
||
} else if (!isWhitespace(currentChar)) {
|
||
while (
|
||
i < chars.length &&
|
||
!isWordCharStrict(chars[i]) &&
|
||
!isWhitespace(chars[i])
|
||
) {
|
||
i++;
|
||
}
|
||
}
|
||
|
||
// Skip whitespace
|
||
while (i < chars.length && isWhitespace(chars[i])) {
|
||
i++;
|
||
}
|
||
|
||
return i < chars.length ? i : null;
|
||
};
|
||
|
||
// Find previous word start within a line
|
||
export const findPrevWordStartInLine = (
|
||
line: string,
|
||
col: number,
|
||
): number | null => {
|
||
const chars = toCodePoints(line);
|
||
let i = col;
|
||
|
||
if (i <= 0) return null;
|
||
|
||
i--;
|
||
|
||
// Skip whitespace moving backwards
|
||
while (i >= 0 && isWhitespace(chars[i])) {
|
||
i--;
|
||
}
|
||
|
||
if (i < 0) return null;
|
||
|
||
if (isWordCharStrict(chars[i])) {
|
||
// We're in a word, move to its beginning
|
||
while (i >= 0 && isWordCharStrict(chars[i])) {
|
||
// Check for script boundary - if previous character is from different script, stop here
|
||
if (
|
||
i - 1 >= 0 &&
|
||
isWordCharStrict(chars[i - 1]) &&
|
||
isDifferentScript(chars[i], chars[i - 1])
|
||
) {
|
||
return i; // Return current position at script boundary
|
||
}
|
||
i--;
|
||
}
|
||
return i + 1;
|
||
} else {
|
||
// We're in punctuation, move to its beginning
|
||
while (i >= 0 && !isWordCharStrict(chars[i]) && !isWhitespace(chars[i])) {
|
||
i--;
|
||
}
|
||
return i + 1;
|
||
}
|
||
};
|
||
|
||
// Find word end within a line
|
||
export const findWordEndInLine = (line: string, col: number): number | null => {
|
||
const chars = toCodePoints(line);
|
||
let i = col;
|
||
|
||
// If we're already at the end of a word (including punctuation sequences), advance to next word
|
||
// This includes both regular word endings and script boundaries
|
||
const atEndOfWordChar =
|
||
i < chars.length &&
|
||
isWordCharWithCombining(chars[i]) &&
|
||
(i + 1 >= chars.length ||
|
||
!isWordCharWithCombining(chars[i + 1]) ||
|
||
(isWordCharStrict(chars[i]) &&
|
||
i + 1 < chars.length &&
|
||
isWordCharStrict(chars[i + 1]) &&
|
||
isDifferentScript(chars[i], chars[i + 1])));
|
||
|
||
const atEndOfPunctuation =
|
||
i < chars.length &&
|
||
!isWordCharWithCombining(chars[i]) &&
|
||
!isWhitespace(chars[i]) &&
|
||
(i + 1 >= chars.length ||
|
||
isWhitespace(chars[i + 1]) ||
|
||
isWordCharWithCombining(chars[i + 1]));
|
||
|
||
if (atEndOfWordChar || atEndOfPunctuation) {
|
||
// We're at the end of a word or punctuation sequence, move forward to find next word
|
||
i++;
|
||
// Skip whitespace to find next word or punctuation
|
||
while (i < chars.length && isWhitespace(chars[i])) {
|
||
i++;
|
||
}
|
||
}
|
||
|
||
// If we're not on a word character, find the next word or punctuation sequence
|
||
if (i < chars.length && !isWordCharWithCombining(chars[i])) {
|
||
// Skip whitespace to find next word or punctuation
|
||
while (i < chars.length && isWhitespace(chars[i])) {
|
||
i++;
|
||
}
|
||
}
|
||
|
||
// Move to end of current word (including combining marks, but stop at script boundaries)
|
||
let foundWord = false;
|
||
let lastBaseCharPos = -1;
|
||
|
||
if (i < chars.length && isWordCharWithCombining(chars[i])) {
|
||
// Handle word characters
|
||
while (i < chars.length && isWordCharWithCombining(chars[i])) {
|
||
foundWord = true;
|
||
|
||
// Track the position of the last base character (not combining mark)
|
||
if (isWordCharStrict(chars[i])) {
|
||
lastBaseCharPos = i;
|
||
}
|
||
|
||
// Check if next character is from a different script (word boundary)
|
||
if (
|
||
i + 1 < chars.length &&
|
||
isWordCharStrict(chars[i + 1]) &&
|
||
isDifferentScript(chars[i], chars[i + 1])
|
||
) {
|
||
i++; // Include current character
|
||
if (isWordCharStrict(chars[i - 1])) {
|
||
lastBaseCharPos = i - 1;
|
||
}
|
||
break; // Stop at script boundary
|
||
}
|
||
|
||
i++;
|
||
}
|
||
} else if (i < chars.length && !isWhitespace(chars[i])) {
|
||
// Handle punctuation sequences (like ████)
|
||
while (
|
||
i < chars.length &&
|
||
!isWordCharStrict(chars[i]) &&
|
||
!isWhitespace(chars[i])
|
||
) {
|
||
foundWord = true;
|
||
lastBaseCharPos = i;
|
||
i++;
|
||
}
|
||
}
|
||
|
||
// Only return a position if we actually found a word
|
||
// Return the position of the last base character, not combining marks
|
||
if (foundWord && lastBaseCharPos >= col) {
|
||
return lastBaseCharPos;
|
||
}
|
||
|
||
return null;
|
||
};
|
||
|
||
// Initialize segmenter for word boundary detection
|
||
const segmenter = new Intl.Segmenter(undefined, { granularity: 'word' });
|
||
|
||
function findPrevWordBoundary(line: string, cursorCol: number): number {
|
||
const codePoints = toCodePoints(line);
|
||
// Convert cursorCol (CP index) to string index
|
||
const prefix = codePoints.slice(0, cursorCol).join('');
|
||
const cursorIdx = prefix.length;
|
||
|
||
let targetIdx = 0;
|
||
|
||
for (const seg of segmenter.segment(line)) {
|
||
// We want the last word start strictly before the cursor.
|
||
// If we've reached or passed the cursor, we stop.
|
||
if (seg.index >= cursorIdx) break;
|
||
|
||
if (seg.isWordLike) {
|
||
targetIdx = seg.index;
|
||
}
|
||
}
|
||
|
||
return toCodePoints(line.slice(0, targetIdx)).length;
|
||
}
|
||
|
||
function findNextWordBoundary(line: string, cursorCol: number): number {
|
||
const codePoints = toCodePoints(line);
|
||
const prefix = codePoints.slice(0, cursorCol).join('');
|
||
const cursorIdx = prefix.length;
|
||
|
||
let targetIdx = line.length;
|
||
|
||
for (const seg of segmenter.segment(line)) {
|
||
const segEnd = seg.index + seg.segment.length;
|
||
|
||
if (segEnd > cursorIdx) {
|
||
if (seg.isWordLike) {
|
||
targetIdx = segEnd;
|
||
break;
|
||
}
|
||
}
|
||
}
|
||
|
||
return toCodePoints(line.slice(0, targetIdx)).length;
|
||
}
|
||
|
||
// Find next word across lines
|
||
export const findNextWordAcrossLines = (
|
||
lines: string[],
|
||
cursorRow: number,
|
||
cursorCol: number,
|
||
searchForWordStart: boolean,
|
||
): { row: number; col: number } | null => {
|
||
// First try current line
|
||
const currentLine = lines[cursorRow] || '';
|
||
const colInCurrentLine = searchForWordStart
|
||
? findNextWordStartInLine(currentLine, cursorCol)
|
||
: findWordEndInLine(currentLine, cursorCol);
|
||
|
||
if (colInCurrentLine !== null) {
|
||
return { row: cursorRow, col: colInCurrentLine };
|
||
}
|
||
|
||
// Search subsequent lines
|
||
for (let row = cursorRow + 1; row < lines.length; row++) {
|
||
const line = lines[row] || '';
|
||
const chars = toCodePoints(line);
|
||
|
||
// For empty lines, if we haven't found any words yet, return the empty line
|
||
if (chars.length === 0) {
|
||
// Check if there are any words in remaining lines
|
||
let hasWordsInLaterLines = false;
|
||
for (let laterRow = row + 1; laterRow < lines.length; laterRow++) {
|
||
const laterLine = lines[laterRow] || '';
|
||
const laterChars = toCodePoints(laterLine);
|
||
let firstNonWhitespace = 0;
|
||
while (
|
||
firstNonWhitespace < laterChars.length &&
|
||
isWhitespace(laterChars[firstNonWhitespace])
|
||
) {
|
||
firstNonWhitespace++;
|
||
}
|
||
if (firstNonWhitespace < laterChars.length) {
|
||
hasWordsInLaterLines = true;
|
||
break;
|
||
}
|
||
}
|
||
|
||
// If no words in later lines, return the empty line
|
||
if (!hasWordsInLaterLines) {
|
||
return { row, col: 0 };
|
||
}
|
||
continue;
|
||
}
|
||
|
||
// Find first non-whitespace
|
||
let firstNonWhitespace = 0;
|
||
while (
|
||
firstNonWhitespace < chars.length &&
|
||
isWhitespace(chars[firstNonWhitespace])
|
||
) {
|
||
firstNonWhitespace++;
|
||
}
|
||
|
||
if (firstNonWhitespace < chars.length) {
|
||
if (searchForWordStart) {
|
||
return { row, col: firstNonWhitespace };
|
||
} else {
|
||
// For word end, find the end of the first word
|
||
const endCol = findWordEndInLine(line, firstNonWhitespace);
|
||
if (endCol !== null) {
|
||
return { row, col: endCol };
|
||
}
|
||
}
|
||
}
|
||
}
|
||
|
||
return null;
|
||
};
|
||
|
||
// Find previous word across lines
|
||
export const findPrevWordAcrossLines = (
|
||
lines: string[],
|
||
cursorRow: number,
|
||
cursorCol: number,
|
||
): { row: number; col: number } | null => {
|
||
// First try current line
|
||
const currentLine = lines[cursorRow] || '';
|
||
const colInCurrentLine = findPrevWordStartInLine(currentLine, cursorCol);
|
||
|
||
if (colInCurrentLine !== null) {
|
||
return { row: cursorRow, col: colInCurrentLine };
|
||
}
|
||
|
||
// Search previous lines
|
||
for (let row = cursorRow - 1; row >= 0; row--) {
|
||
const line = lines[row] || '';
|
||
const chars = toCodePoints(line);
|
||
|
||
if (chars.length === 0) continue;
|
||
|
||
// Find last word start
|
||
let lastWordStart = chars.length;
|
||
while (lastWordStart > 0 && isWhitespace(chars[lastWordStart - 1])) {
|
||
lastWordStart--;
|
||
}
|
||
|
||
if (lastWordStart > 0) {
|
||
// Find start of this word
|
||
const wordStart = findPrevWordStartInLine(line, lastWordStart);
|
||
if (wordStart !== null) {
|
||
return { row, col: wordStart };
|
||
}
|
||
}
|
||
}
|
||
|
||
return null;
|
||
};
|
||
|
||
// Helper functions for vim line operations
|
||
export const getPositionFromOffsets = (
|
||
startOffset: number,
|
||
endOffset: number,
|
||
lines: string[],
|
||
) => {
|
||
let offset = 0;
|
||
let startRow = 0;
|
||
let startCol = 0;
|
||
let endRow = 0;
|
||
let endCol = 0;
|
||
|
||
// Find start position
|
||
for (let i = 0; i < lines.length; i++) {
|
||
const lineLength = lines[i].length + 1; // +1 for newline
|
||
if (offset + lineLength > startOffset) {
|
||
startRow = i;
|
||
startCol = startOffset - offset;
|
||
break;
|
||
}
|
||
offset += lineLength;
|
||
}
|
||
|
||
// Find end position
|
||
offset = 0;
|
||
for (let i = 0; i < lines.length; i++) {
|
||
const lineLength = lines[i].length + (i < lines.length - 1 ? 1 : 0); // +1 for newline except last line
|
||
if (offset + lineLength >= endOffset) {
|
||
endRow = i;
|
||
endCol = endOffset - offset;
|
||
break;
|
||
}
|
||
offset += lineLength;
|
||
}
|
||
|
||
return { startRow, startCol, endRow, endCol };
|
||
};
|
||
|
||
export const getLineRangeOffsets = (
|
||
startRow: number,
|
||
lineCount: number,
|
||
lines: string[],
|
||
) => {
|
||
let startOffset = 0;
|
||
|
||
// Calculate start offset
|
||
for (let i = 0; i < startRow; i++) {
|
||
startOffset += lines[i].length + 1; // +1 for newline
|
||
}
|
||
|
||
// Calculate end offset
|
||
let endOffset = startOffset;
|
||
for (let i = 0; i < lineCount; i++) {
|
||
const lineIndex = startRow + i;
|
||
if (lineIndex < lines.length) {
|
||
endOffset += lines[lineIndex].length;
|
||
if (lineIndex < lines.length - 1) {
|
||
endOffset += 1; // +1 for newline
|
||
}
|
||
}
|
||
}
|
||
|
||
return { startOffset, endOffset };
|
||
};
|
||
|
||
export const replaceRangeInternal = (
|
||
state: TextBufferState,
|
||
startRow: number,
|
||
startCol: number,
|
||
endRow: number,
|
||
endCol: number,
|
||
text: string,
|
||
): TextBufferState => {
|
||
const currentLine = (row: number) => state.lines[row] || '';
|
||
const currentLineLen = (row: number) => cpLen(currentLine(row));
|
||
const clamp = (value: number, min: number, max: number) =>
|
||
Math.min(Math.max(value, min), max);
|
||
|
||
if (
|
||
startRow > endRow ||
|
||
(startRow === endRow && startCol > endCol) ||
|
||
startRow < 0 ||
|
||
startCol < 0 ||
|
||
endRow >= state.lines.length ||
|
||
(endRow < state.lines.length && endCol > currentLineLen(endRow))
|
||
) {
|
||
return state; // Invalid range
|
||
}
|
||
|
||
const newLines = [...state.lines];
|
||
|
||
const sCol = clamp(startCol, 0, currentLineLen(startRow));
|
||
const eCol = clamp(endCol, 0, currentLineLen(endRow));
|
||
|
||
const prefix = cpSlice(currentLine(startRow), 0, sCol);
|
||
const suffix = cpSlice(currentLine(endRow), eCol);
|
||
|
||
const normalisedReplacement = text
|
||
.replace(/\r\n/g, '\n')
|
||
.replace(/\r/g, '\n');
|
||
const replacementParts = normalisedReplacement.split('\n');
|
||
|
||
// The combined first line of the new text
|
||
const firstLine = prefix + replacementParts[0];
|
||
|
||
if (replacementParts.length === 1) {
|
||
// No newlines in replacement: combine prefix, replacement, and suffix on one line.
|
||
newLines.splice(startRow, endRow - startRow + 1, firstLine + suffix);
|
||
} else {
|
||
// Newlines in replacement: create new lines.
|
||
const lastLine = replacementParts[replacementParts.length - 1] + suffix;
|
||
const middleLines = replacementParts.slice(1, -1);
|
||
newLines.splice(
|
||
startRow,
|
||
endRow - startRow + 1,
|
||
firstLine,
|
||
...middleLines,
|
||
lastLine,
|
||
);
|
||
}
|
||
|
||
const finalCursorRow = startRow + replacementParts.length - 1;
|
||
const finalCursorCol =
|
||
(replacementParts.length > 1 ? 0 : sCol) +
|
||
cpLen(replacementParts[replacementParts.length - 1]);
|
||
|
||
return {
|
||
...state,
|
||
lines: newLines,
|
||
cursorRow: Math.min(Math.max(finalCursorRow, 0), newLines.length - 1),
|
||
cursorCol: Math.max(
|
||
0,
|
||
Math.min(finalCursorCol, cpLen(newLines[finalCursorRow] || '')),
|
||
),
|
||
preferredCol: null,
|
||
};
|
||
};
|
||
|
||
export interface Viewport {
|
||
height: number;
|
||
width: number;
|
||
}
|
||
|
||
function clamp(v: number, min: number, max: number): number {
|
||
return v < min ? min : v > max ? max : v;
|
||
}
|
||
|
||
/* ────────────────────────────────────────────────────────────────────────── */
|
||
|
||
interface UseTextBufferProps {
|
||
initialText?: string;
|
||
initialCursorOffset?: number;
|
||
viewport: Viewport; // Viewport dimensions needed for scrolling
|
||
stdin?: NodeJS.ReadStream | null; // For external editor
|
||
setRawMode?: (mode: boolean) => void; // For external editor
|
||
onChange?: (text: string) => void; // Callback for when text changes
|
||
isValidPath: (path: string) => boolean;
|
||
shellModeActive?: boolean; // Whether the text buffer is in shell mode
|
||
inputFilter?: (text: string) => string; // Optional filter for input text
|
||
singleLine?: boolean;
|
||
getPreferredEditor?: () => EditorType | undefined;
|
||
}
|
||
|
||
interface UndoHistoryEntry {
|
||
lines: string[];
|
||
cursorRow: number;
|
||
cursorCol: number;
|
||
pastedContent: Record<string, string>;
|
||
expandedPaste: ExpandedPasteInfo | null;
|
||
}
|
||
|
||
function calculateInitialCursorPosition(
|
||
initialLines: string[],
|
||
offset: number,
|
||
): [number, number] {
|
||
let remainingChars = offset;
|
||
let row = 0;
|
||
while (row < initialLines.length) {
|
||
const lineLength = cpLen(initialLines[row]);
|
||
// Add 1 for the newline character (except for the last line)
|
||
const totalCharsInLineAndNewline =
|
||
lineLength + (row < initialLines.length - 1 ? 1 : 0);
|
||
|
||
if (remainingChars <= lineLength) {
|
||
// Cursor is on this line
|
||
return [row, remainingChars];
|
||
}
|
||
remainingChars -= totalCharsInLineAndNewline;
|
||
row++;
|
||
}
|
||
// Offset is beyond the text, place cursor at the end of the last line
|
||
if (initialLines.length > 0) {
|
||
const lastRow = initialLines.length - 1;
|
||
return [lastRow, cpLen(initialLines[lastRow])];
|
||
}
|
||
return [0, 0]; // Default for empty text
|
||
}
|
||
|
||
export function offsetToLogicalPos(
|
||
text: string,
|
||
offset: number,
|
||
): [number, number] {
|
||
let row = 0;
|
||
let col = 0;
|
||
let currentOffset = 0;
|
||
|
||
if (offset === 0) return [0, 0];
|
||
|
||
const lines = text.split('\n');
|
||
for (let i = 0; i < lines.length; i++) {
|
||
const line = lines[i];
|
||
const lineLength = cpLen(line);
|
||
const lineLengthWithNewline = lineLength + (i < lines.length - 1 ? 1 : 0);
|
||
|
||
if (offset <= currentOffset + lineLength) {
|
||
// Check against lineLength first
|
||
row = i;
|
||
col = offset - currentOffset;
|
||
return [row, col];
|
||
} else if (offset <= currentOffset + lineLengthWithNewline) {
|
||
// Check if offset is the newline itself
|
||
row = i;
|
||
col = lineLength; // Position cursor at the end of the current line content
|
||
// If the offset IS the newline, and it's not the last line, advance to next line, col 0
|
||
if (
|
||
offset === currentOffset + lineLengthWithNewline &&
|
||
i < lines.length - 1
|
||
) {
|
||
return [i + 1, 0];
|
||
}
|
||
return [row, col]; // Otherwise, it's at the end of the current line content
|
||
}
|
||
currentOffset += lineLengthWithNewline;
|
||
}
|
||
|
||
// If offset is beyond the text length, place cursor at the end of the last line
|
||
// or [0,0] if text is empty
|
||
if (lines.length > 0) {
|
||
row = lines.length - 1;
|
||
col = cpLen(lines[row]);
|
||
} else {
|
||
row = 0;
|
||
col = 0;
|
||
}
|
||
return [row, col];
|
||
}
|
||
|
||
/**
|
||
* Converts logical row/col position to absolute text offset
|
||
* Inverse operation of offsetToLogicalPos
|
||
*/
|
||
export function logicalPosToOffset(
|
||
lines: string[],
|
||
row: number,
|
||
col: number,
|
||
): number {
|
||
let offset = 0;
|
||
|
||
// Clamp row to valid range
|
||
const actualRow = Math.min(row, lines.length - 1);
|
||
|
||
// Add lengths of all lines before the target row
|
||
for (let i = 0; i < actualRow; i++) {
|
||
offset += cpLen(lines[i]) + 1; // +1 for newline
|
||
}
|
||
|
||
// Add column offset within the target row
|
||
if (actualRow >= 0 && actualRow < lines.length) {
|
||
offset += Math.min(col, cpLen(lines[actualRow]));
|
||
}
|
||
|
||
return offset;
|
||
}
|
||
/**
|
||
* Transformations allow for the CLI to render terse representations of things like file paths
|
||
* (e.g., "@some/path/to/an/image.png" to "[Image image.png]")
|
||
* When the cursor enters a transformed representation, it expands to reveal the logical representation.
|
||
* (e.g., "[Image image.png]" to "@some/path/to/an/image.png")
|
||
*/
|
||
export interface Transformation {
|
||
logStart: number;
|
||
logEnd: number;
|
||
logicalText: string;
|
||
collapsedText: string;
|
||
type: 'image' | 'paste';
|
||
id?: string; // For paste placeholders
|
||
}
|
||
export const imagePathRegex =
|
||
/@((?:\\.|[^\s\r\n\\])+?\.(?:png|jpg|jpeg|gif|webp|svg|bmp))\b/gi;
|
||
|
||
export function getTransformedImagePath(filePath: string): string {
|
||
const raw = filePath;
|
||
|
||
// Ignore leading @ when stripping directories, but keep it for simple '@file.png'
|
||
const withoutAt = raw.startsWith('@') ? raw.slice(1) : raw;
|
||
|
||
// Unescape the path to handle escaped spaces and other characters
|
||
const unescaped = unescapePath(withoutAt);
|
||
|
||
// Find last directory separator, supporting both POSIX and Windows styles
|
||
const lastSepIndex = Math.max(
|
||
unescaped.lastIndexOf('/'),
|
||
unescaped.lastIndexOf('\\'),
|
||
);
|
||
|
||
// If we saw a separator, take the segment after it; otherwise fall back to the unescaped string
|
||
const fileName =
|
||
lastSepIndex >= 0 ? unescaped.slice(lastSepIndex + 1) : unescaped;
|
||
|
||
const extension = path.extname(fileName);
|
||
const baseName = path.basename(fileName, extension);
|
||
const maxBaseLength = 10;
|
||
|
||
const truncatedBase =
|
||
baseName.length > maxBaseLength
|
||
? `...${baseName.slice(-maxBaseLength)}`
|
||
: baseName;
|
||
|
||
return `[Image ${truncatedBase}${extension}]`;
|
||
}
|
||
|
||
const transformationsCache = new LRUCache<string, Transformation[]>(
|
||
LRU_BUFFER_PERF_CACHE_LIMIT,
|
||
);
|
||
|
||
export function calculateTransformationsForLine(
|
||
line: string,
|
||
): Transformation[] {
|
||
const cached = transformationsCache.get(line);
|
||
if (cached) {
|
||
return cached;
|
||
}
|
||
|
||
const transformations: Transformation[] = [];
|
||
|
||
// 1. Detect image paths
|
||
imagePathRegex.lastIndex = 0;
|
||
let match: RegExpExecArray | null;
|
||
while ((match = imagePathRegex.exec(line)) !== null) {
|
||
const logicalText = match[0];
|
||
const logStart = cpLen(line.substring(0, match.index));
|
||
const logEnd = logStart + cpLen(logicalText);
|
||
|
||
transformations.push({
|
||
logStart,
|
||
logEnd,
|
||
logicalText,
|
||
collapsedText: getTransformedImagePath(logicalText),
|
||
type: 'image',
|
||
});
|
||
}
|
||
|
||
// 2. Detect paste placeholders
|
||
const pasteRegex = new RegExp(PASTED_TEXT_PLACEHOLDER_REGEX.source, 'g');
|
||
while ((match = pasteRegex.exec(line)) !== null) {
|
||
const logicalText = match[0];
|
||
const logStart = cpLen(line.substring(0, match.index));
|
||
const logEnd = logStart + cpLen(logicalText);
|
||
|
||
transformations.push({
|
||
logStart,
|
||
logEnd,
|
||
logicalText,
|
||
collapsedText: logicalText,
|
||
type: 'paste',
|
||
id: logicalText,
|
||
});
|
||
}
|
||
|
||
// Sort transformations by logStart to maintain consistency
|
||
transformations.sort((a, b) => a.logStart - b.logStart);
|
||
|
||
transformationsCache.set(line, transformations);
|
||
|
||
return transformations;
|
||
}
|
||
|
||
export function calculateTransformations(lines: string[]): Transformation[][] {
|
||
return lines.map((ln) => calculateTransformationsForLine(ln));
|
||
}
|
||
|
||
export function getTransformUnderCursor(
|
||
row: number,
|
||
col: number,
|
||
spansByLine: Transformation[][],
|
||
): Transformation | null {
|
||
const spans = spansByLine[row];
|
||
if (!spans || spans.length === 0) return null;
|
||
for (const span of spans) {
|
||
if (col >= span.logStart && col < span.logEnd) {
|
||
return span;
|
||
}
|
||
if (col < span.logStart) break;
|
||
}
|
||
return null;
|
||
}
|
||
|
||
export interface ExpandedPasteInfo {
|
||
id: string;
|
||
startLine: number;
|
||
lineCount: number;
|
||
prefix: string;
|
||
suffix: string;
|
||
}
|
||
|
||
/**
|
||
* Check if a line index falls within an expanded paste region.
|
||
* Returns the paste placeholder ID if found, null otherwise.
|
||
*/
|
||
export function getExpandedPasteAtLine(
|
||
lineIndex: number,
|
||
expandedPaste: ExpandedPasteInfo | null,
|
||
): string | null {
|
||
if (
|
||
expandedPaste &&
|
||
lineIndex >= expandedPaste.startLine &&
|
||
lineIndex < expandedPaste.startLine + expandedPaste.lineCount
|
||
) {
|
||
return expandedPaste.id;
|
||
}
|
||
return null;
|
||
}
|
||
|
||
/**
|
||
* Surgery for expanded paste regions when lines are added or removed.
|
||
* Adjusts startLine indices and detaches any region that is partially or fully deleted.
|
||
*/
|
||
export function shiftExpandedRegions(
|
||
expandedPaste: ExpandedPasteInfo | null,
|
||
changeStartLine: number,
|
||
lineDelta: number,
|
||
changeEndLine?: number, // Inclusive
|
||
): {
|
||
newInfo: ExpandedPasteInfo | null;
|
||
isDetached: boolean;
|
||
} {
|
||
if (!expandedPaste) return { newInfo: null, isDetached: false };
|
||
|
||
const effectiveEndLine = changeEndLine ?? changeStartLine;
|
||
const infoEndLine = expandedPaste.startLine + expandedPaste.lineCount - 1;
|
||
|
||
// 1. Check for overlap/intersection with the changed range
|
||
const isOverlapping =
|
||
changeStartLine <= infoEndLine &&
|
||
effectiveEndLine >= expandedPaste.startLine;
|
||
|
||
if (isOverlapping) {
|
||
// If the change is a deletion (lineDelta < 0) that touches this region, we detach.
|
||
// If it's an insertion, we only detach if it's a multi-line insertion (lineDelta > 0)
|
||
// that isn't at the very start of the region (which would shift it).
|
||
// Regular character typing (lineDelta === 0) does NOT detach.
|
||
if (
|
||
lineDelta < 0 ||
|
||
(lineDelta > 0 &&
|
||
changeStartLine > expandedPaste.startLine &&
|
||
changeStartLine <= infoEndLine)
|
||
) {
|
||
return { newInfo: null, isDetached: true };
|
||
}
|
||
}
|
||
|
||
// 2. Shift regions that start at or after the change point
|
||
if (expandedPaste.startLine >= changeStartLine) {
|
||
return {
|
||
newInfo: {
|
||
...expandedPaste,
|
||
startLine: expandedPaste.startLine + lineDelta,
|
||
},
|
||
isDetached: false,
|
||
};
|
||
}
|
||
|
||
return { newInfo: expandedPaste, isDetached: false };
|
||
}
|
||
|
||
/**
|
||
* Detach any expanded paste region if the cursor is within it.
|
||
* This converts the expanded content to regular text that can no longer be collapsed.
|
||
* Returns the state unchanged if cursor is not in an expanded region.
|
||
*/
|
||
export function detachExpandedPaste(state: TextBufferState): TextBufferState {
|
||
const expandedId = getExpandedPasteAtLine(
|
||
state.cursorRow,
|
||
state.expandedPaste,
|
||
);
|
||
if (!expandedId) return state;
|
||
|
||
const { [expandedId]: _, ...newPastedContent } = state.pastedContent;
|
||
return {
|
||
...state,
|
||
expandedPaste: null,
|
||
pastedContent: newPastedContent,
|
||
};
|
||
}
|
||
|
||
/**
|
||
* Represents an atomic placeholder that should be deleted as a unit.
|
||
* Extensible to support future placeholder types.
|
||
*/
|
||
interface AtomicPlaceholder {
|
||
start: number; // Start position in logical text
|
||
end: number; // End position in logical text
|
||
type: 'paste' | 'image'; // Type for cleanup logic
|
||
id?: string; // For paste placeholders: the pastedContent key
|
||
}
|
||
|
||
/**
|
||
* Find atomic placeholder at cursor for backspace (cursor at end).
|
||
* Checks all placeholder types in priority order.
|
||
*/
|
||
function findAtomicPlaceholderForBackspace(
|
||
line: string,
|
||
cursorCol: number,
|
||
transformations: Transformation[],
|
||
): AtomicPlaceholder | null {
|
||
for (const transform of transformations) {
|
||
if (cursorCol === transform.logEnd) {
|
||
return {
|
||
start: transform.logStart,
|
||
end: transform.logEnd,
|
||
type: transform.type,
|
||
id: transform.id,
|
||
};
|
||
}
|
||
}
|
||
|
||
return null;
|
||
}
|
||
|
||
/**
|
||
* Find atomic placeholder at cursor for delete (cursor at start).
|
||
*/
|
||
function findAtomicPlaceholderForDelete(
|
||
line: string,
|
||
cursorCol: number,
|
||
transformations: Transformation[],
|
||
): AtomicPlaceholder | null {
|
||
for (const transform of transformations) {
|
||
if (cursorCol === transform.logStart) {
|
||
return {
|
||
start: transform.logStart,
|
||
end: transform.logEnd,
|
||
type: transform.type,
|
||
id: transform.id,
|
||
};
|
||
}
|
||
}
|
||
|
||
return null;
|
||
}
|
||
|
||
export function calculateTransformedLine(
|
||
logLine: string,
|
||
logIndex: number,
|
||
logicalCursor: [number, number],
|
||
transformations: Transformation[],
|
||
): { transformedLine: string; transformedToLogMap: number[] } {
|
||
let transformedLine = '';
|
||
const transformedToLogMap: number[] = [];
|
||
let lastLogPos = 0;
|
||
|
||
const cursorIsOnThisLine = logIndex === logicalCursor[0];
|
||
const cursorCol = logicalCursor[1];
|
||
|
||
for (const transform of transformations) {
|
||
const textBeforeTransformation = cpSlice(
|
||
logLine,
|
||
lastLogPos,
|
||
transform.logStart,
|
||
);
|
||
transformedLine += textBeforeTransformation;
|
||
for (let i = 0; i < cpLen(textBeforeTransformation); i++) {
|
||
transformedToLogMap.push(lastLogPos + i);
|
||
}
|
||
|
||
const isExpanded =
|
||
transform.type === 'image' &&
|
||
cursorIsOnThisLine &&
|
||
cursorCol >= transform.logStart &&
|
||
cursorCol <= transform.logEnd;
|
||
const transformedText = isExpanded
|
||
? transform.logicalText
|
||
: transform.collapsedText;
|
||
transformedLine += transformedText;
|
||
|
||
// Map transformed characters back to logical characters
|
||
const transformedLen = cpLen(transformedText);
|
||
if (isExpanded) {
|
||
for (let i = 0; i < transformedLen; i++) {
|
||
transformedToLogMap.push(transform.logStart + i);
|
||
}
|
||
} else {
|
||
// Collapsed: distribute transformed positions monotonically across the raw span.
|
||
// This preserves ordering across wrapped slices so logicalToVisualMap has
|
||
// increasing startColInLogical and visual cursor mapping remains consistent.
|
||
const logicalLength = Math.max(0, transform.logEnd - transform.logStart);
|
||
for (let i = 0; i < transformedLen; i++) {
|
||
// Map the i-th transformed code point into [logStart, logEnd)
|
||
const transformationToLogicalOffset =
|
||
logicalLength === 0
|
||
? 0
|
||
: Math.floor((i * logicalLength) / transformedLen);
|
||
const transformationToLogicalIndex =
|
||
transform.logStart +
|
||
Math.min(
|
||
transformationToLogicalOffset,
|
||
Math.max(logicalLength - 1, 0),
|
||
);
|
||
transformedToLogMap.push(transformationToLogicalIndex);
|
||
}
|
||
}
|
||
lastLogPos = transform.logEnd;
|
||
}
|
||
|
||
// Append text after last transform
|
||
const remainingUntransformedText = cpSlice(logLine, lastLogPos);
|
||
transformedLine += remainingUntransformedText;
|
||
for (let i = 0; i < cpLen(remainingUntransformedText); i++) {
|
||
transformedToLogMap.push(lastLogPos + i);
|
||
}
|
||
|
||
// For a cursor at the very end of the transformed line
|
||
transformedToLogMap.push(cpLen(logLine));
|
||
|
||
return { transformedLine, transformedToLogMap };
|
||
}
|
||
|
||
export interface VisualLayout {
|
||
visualLines: string[];
|
||
// For each logical line, an array of [visualLineIndex, startColInLogical]
|
||
logicalToVisualMap: Array<Array<[number, number]>>;
|
||
// For each visual line, its [logicalLineIndex, startColInLogical]
|
||
visualToLogicalMap: Array<[number, number]>;
|
||
// Image paths are transformed (e.g., "@some/path/to/an/image.png" to "[Image image.png]")
|
||
// For each logical line, an array that maps each transformedCol to a logicalCol
|
||
transformedToLogicalMaps: number[][];
|
||
// For each visual line, its [startColInTransformed]
|
||
visualToTransformedMap: number[];
|
||
}
|
||
|
||
// Caches for layout calculation
|
||
interface LineLayoutResult {
|
||
visualLines: string[];
|
||
logicalToVisualMap: Array<[number, number]>;
|
||
visualToLogicalMap: Array<[number, number]>;
|
||
transformedToLogMap: number[];
|
||
visualToTransformedMap: number[];
|
||
}
|
||
|
||
const lineLayoutCache = new LRUCache<string, LineLayoutResult>(
|
||
LRU_BUFFER_PERF_CACHE_LIMIT,
|
||
);
|
||
|
||
function getLineLayoutCacheKey(
|
||
line: string,
|
||
viewportWidth: number,
|
||
isCursorOnLine: boolean,
|
||
cursorCol: number,
|
||
): string {
|
||
// Most lines (99.9% in a large buffer) are not cursor lines.
|
||
// We use a simpler key for them to reduce string allocation overhead.
|
||
if (!isCursorOnLine) {
|
||
return `${viewportWidth}:N:${line}`;
|
||
}
|
||
return `${viewportWidth}:C:${cursorCol}:${line}`;
|
||
}
|
||
|
||
// Calculates the visual wrapping of lines and the mapping between logical and visual coordinates.
|
||
// This is an expensive operation and should be memoized.
|
||
function calculateLayout(
|
||
logicalLines: string[],
|
||
viewportWidth: number,
|
||
logicalCursor: [number, number],
|
||
): VisualLayout {
|
||
const visualLines: string[] = [];
|
||
const logicalToVisualMap: Array<Array<[number, number]>> = [];
|
||
const visualToLogicalMap: Array<[number, number]> = [];
|
||
const transformedToLogicalMaps: number[][] = [];
|
||
const visualToTransformedMap: number[] = [];
|
||
|
||
logicalLines.forEach((logLine, logIndex) => {
|
||
logicalToVisualMap[logIndex] = [];
|
||
|
||
const isCursorOnLine = logIndex === logicalCursor[0];
|
||
const cacheKey = getLineLayoutCacheKey(
|
||
logLine,
|
||
viewportWidth,
|
||
isCursorOnLine,
|
||
logicalCursor[1],
|
||
);
|
||
const cached = lineLayoutCache.get(cacheKey);
|
||
|
||
if (cached) {
|
||
const visualLineOffset = visualLines.length;
|
||
visualLines.push(...cached.visualLines);
|
||
cached.logicalToVisualMap.forEach(([relVisualIdx, logCol]) => {
|
||
logicalToVisualMap[logIndex].push([
|
||
visualLineOffset + relVisualIdx,
|
||
logCol,
|
||
]);
|
||
});
|
||
cached.visualToLogicalMap.forEach(([, logCol]) => {
|
||
visualToLogicalMap.push([logIndex, logCol]);
|
||
});
|
||
transformedToLogicalMaps[logIndex] = cached.transformedToLogMap;
|
||
visualToTransformedMap.push(...cached.visualToTransformedMap);
|
||
return;
|
||
}
|
||
|
||
// Not in cache, calculate
|
||
const transformations = calculateTransformationsForLine(logLine);
|
||
const { transformedLine, transformedToLogMap } = calculateTransformedLine(
|
||
logLine,
|
||
logIndex,
|
||
logicalCursor,
|
||
transformations,
|
||
);
|
||
|
||
const lineVisualLines: string[] = [];
|
||
const lineLogicalToVisualMap: Array<[number, number]> = [];
|
||
const lineVisualToLogicalMap: Array<[number, number]> = [];
|
||
const lineVisualToTransformedMap: number[] = [];
|
||
|
||
if (transformedLine.length === 0) {
|
||
// Handle empty logical line
|
||
lineLogicalToVisualMap.push([0, 0]);
|
||
lineVisualToLogicalMap.push([logIndex, 0]);
|
||
lineVisualToTransformedMap.push(0);
|
||
lineVisualLines.push('');
|
||
} else {
|
||
// Non-empty logical line
|
||
let currentPosInLogLine = 0; // Tracks position within the current logical line (code point index)
|
||
const codePointsInLogLine = toCodePoints(transformedLine);
|
||
|
||
while (currentPosInLogLine < codePointsInLogLine.length) {
|
||
let currentChunk = '';
|
||
let currentChunkVisualWidth = 0;
|
||
let numCodePointsInChunk = 0;
|
||
let lastWordBreakPoint = -1; // Index in codePointsInLogLine for word break
|
||
let numCodePointsAtLastWordBreak = 0;
|
||
|
||
// Iterate through code points to build the current visual line (chunk)
|
||
for (let i = currentPosInLogLine; i < codePointsInLogLine.length; i++) {
|
||
const char = codePointsInLogLine[i];
|
||
const charVisualWidth = getCachedStringWidth(char);
|
||
|
||
if (currentChunkVisualWidth + charVisualWidth > viewportWidth) {
|
||
// Character would exceed viewport width
|
||
if (
|
||
lastWordBreakPoint !== -1 &&
|
||
numCodePointsAtLastWordBreak > 0 &&
|
||
currentPosInLogLine + numCodePointsAtLastWordBreak < i
|
||
) {
|
||
// We have a valid word break point to use, and it's not the start of the current segment
|
||
currentChunk = codePointsInLogLine
|
||
.slice(
|
||
currentPosInLogLine,
|
||
currentPosInLogLine + numCodePointsAtLastWordBreak,
|
||
)
|
||
.join('');
|
||
numCodePointsInChunk = numCodePointsAtLastWordBreak;
|
||
} else {
|
||
// No word break, or word break is at the start of this potential chunk, or word break leads to empty chunk.
|
||
// Hard break: take characters up to viewportWidth, or just the current char if it alone is too wide.
|
||
if (
|
||
numCodePointsInChunk === 0 &&
|
||
charVisualWidth > viewportWidth
|
||
) {
|
||
// Single character is wider than viewport, take it anyway
|
||
currentChunk = char;
|
||
numCodePointsInChunk = 1;
|
||
}
|
||
}
|
||
break; // Break from inner loop to finalize this chunk
|
||
}
|
||
|
||
currentChunk += char;
|
||
currentChunkVisualWidth += charVisualWidth;
|
||
numCodePointsInChunk++;
|
||
|
||
// Check for word break opportunity (space)
|
||
if (char === ' ') {
|
||
lastWordBreakPoint = i; // Store code point index of the space
|
||
// Store the state *before* adding the space, if we decide to break here.
|
||
numCodePointsAtLastWordBreak = numCodePointsInChunk - 1; // Chars *before* the space
|
||
}
|
||
}
|
||
|
||
if (
|
||
numCodePointsInChunk === 0 &&
|
||
currentPosInLogLine < codePointsInLogLine.length
|
||
) {
|
||
const firstChar = codePointsInLogLine[currentPosInLogLine];
|
||
currentChunk = firstChar;
|
||
numCodePointsInChunk = 1;
|
||
}
|
||
|
||
const logicalStartCol = transformedToLogMap[currentPosInLogLine] ?? 0;
|
||
lineLogicalToVisualMap.push([lineVisualLines.length, logicalStartCol]);
|
||
lineVisualToLogicalMap.push([logIndex, logicalStartCol]);
|
||
lineVisualToTransformedMap.push(currentPosInLogLine);
|
||
lineVisualLines.push(currentChunk);
|
||
|
||
const logicalStartOfThisChunk = currentPosInLogLine;
|
||
currentPosInLogLine += numCodePointsInChunk;
|
||
|
||
if (
|
||
logicalStartOfThisChunk + numCodePointsInChunk <
|
||
codePointsInLogLine.length &&
|
||
currentPosInLogLine < codePointsInLogLine.length &&
|
||
codePointsInLogLine[currentPosInLogLine] === ' '
|
||
) {
|
||
currentPosInLogLine++;
|
||
}
|
||
}
|
||
}
|
||
|
||
// Cache the result for this line
|
||
lineLayoutCache.set(cacheKey, {
|
||
visualLines: lineVisualLines,
|
||
logicalToVisualMap: lineLogicalToVisualMap,
|
||
visualToLogicalMap: lineVisualToLogicalMap,
|
||
transformedToLogMap,
|
||
visualToTransformedMap: lineVisualToTransformedMap,
|
||
});
|
||
|
||
const visualLineOffset = visualLines.length;
|
||
visualLines.push(...lineVisualLines);
|
||
lineLogicalToVisualMap.forEach(([relVisualIdx, logCol]) => {
|
||
logicalToVisualMap[logIndex].push([
|
||
visualLineOffset + relVisualIdx,
|
||
logCol,
|
||
]);
|
||
});
|
||
lineVisualToLogicalMap.forEach(([, logCol]) => {
|
||
visualToLogicalMap.push([logIndex, logCol]);
|
||
});
|
||
transformedToLogicalMaps[logIndex] = transformedToLogMap;
|
||
visualToTransformedMap.push(...lineVisualToTransformedMap);
|
||
});
|
||
|
||
// If the entire logical text was empty, ensure there's one empty visual line.
|
||
if (
|
||
logicalLines.length === 0 ||
|
||
(logicalLines.length === 1 && logicalLines[0] === '')
|
||
) {
|
||
if (visualLines.length === 0) {
|
||
visualLines.push('');
|
||
if (!logicalToVisualMap[0]) logicalToVisualMap[0] = [];
|
||
logicalToVisualMap[0].push([0, 0]);
|
||
visualToLogicalMap.push([0, 0]);
|
||
visualToTransformedMap.push(0);
|
||
}
|
||
}
|
||
|
||
return {
|
||
visualLines,
|
||
logicalToVisualMap,
|
||
visualToLogicalMap,
|
||
transformedToLogicalMaps,
|
||
visualToTransformedMap,
|
||
};
|
||
}
|
||
|
||
// Calculates the visual cursor position based on a pre-calculated layout.
|
||
// This is a lightweight operation.
|
||
function calculateVisualCursorFromLayout(
|
||
layout: VisualLayout,
|
||
logicalCursor: [number, number],
|
||
): [number, number] {
|
||
const { logicalToVisualMap, visualLines, transformedToLogicalMaps } = layout;
|
||
const [logicalRow, logicalCol] = logicalCursor;
|
||
|
||
const segmentsForLogicalLine = logicalToVisualMap[logicalRow];
|
||
|
||
if (!segmentsForLogicalLine || segmentsForLogicalLine.length === 0) {
|
||
// This can happen for an empty document.
|
||
return [0, 0];
|
||
}
|
||
|
||
// Find the segment where the logical column fits.
|
||
// The segments are sorted by startColInLogical.
|
||
let targetSegmentIndex = segmentsForLogicalLine.findIndex(
|
||
([, startColInLogical], index) => {
|
||
const nextStartColInLogical =
|
||
index + 1 < segmentsForLogicalLine.length
|
||
? segmentsForLogicalLine[index + 1][1]
|
||
: Infinity;
|
||
return (
|
||
logicalCol >= startColInLogical && logicalCol < nextStartColInLogical
|
||
);
|
||
},
|
||
);
|
||
|
||
// If not found, it means the cursor is at the end of the logical line.
|
||
if (targetSegmentIndex === -1) {
|
||
if (logicalCol === 0) {
|
||
targetSegmentIndex = 0;
|
||
} else {
|
||
targetSegmentIndex = segmentsForLogicalLine.length - 1;
|
||
}
|
||
}
|
||
|
||
const [visualRow, startColInLogical] =
|
||
segmentsForLogicalLine[targetSegmentIndex];
|
||
|
||
// Find the coordinates in transformed space in order to conver to visual
|
||
const transformedToLogicalMap = transformedToLogicalMaps[logicalRow] ?? [];
|
||
let transformedCol = 0;
|
||
for (let i = 0; i < transformedToLogicalMap.length; i++) {
|
||
if (transformedToLogicalMap[i] > logicalCol) {
|
||
transformedCol = Math.max(0, i - 1);
|
||
break;
|
||
}
|
||
if (i === transformedToLogicalMap.length - 1) {
|
||
transformedCol = transformedToLogicalMap.length - 1;
|
||
}
|
||
}
|
||
let startColInTransformed = 0;
|
||
while (
|
||
startColInTransformed < transformedToLogicalMap.length &&
|
||
transformedToLogicalMap[startColInTransformed] < startColInLogical
|
||
) {
|
||
startColInTransformed++;
|
||
}
|
||
const clampedTransformedCol = Math.min(
|
||
transformedCol,
|
||
Math.max(0, transformedToLogicalMap.length - 1),
|
||
);
|
||
const visualCol = clampedTransformedCol - startColInTransformed;
|
||
const clampedVisualCol = Math.min(
|
||
Math.max(visualCol, 0),
|
||
cpLen(visualLines[visualRow] ?? ''),
|
||
);
|
||
return [visualRow, clampedVisualCol];
|
||
}
|
||
|
||
// --- Start of reducer logic ---
|
||
|
||
export interface TextBufferState {
|
||
lines: string[];
|
||
cursorRow: number;
|
||
cursorCol: number;
|
||
transformationsByLine: Transformation[][];
|
||
preferredCol: number | null; // This is the logical character offset in the visual line
|
||
undoStack: UndoHistoryEntry[];
|
||
redoStack: UndoHistoryEntry[];
|
||
clipboard: string | null;
|
||
selectionAnchor: [number, number] | null;
|
||
viewportWidth: number;
|
||
viewportHeight: number;
|
||
visualLayout: VisualLayout;
|
||
pastedContent: Record<string, string>;
|
||
expandedPaste: ExpandedPasteInfo | null;
|
||
}
|
||
|
||
const historyLimit = 100;
|
||
|
||
export const pushUndo = (currentState: TextBufferState): TextBufferState => {
|
||
const snapshot: UndoHistoryEntry = {
|
||
lines: [...currentState.lines],
|
||
cursorRow: currentState.cursorRow,
|
||
cursorCol: currentState.cursorCol,
|
||
pastedContent: { ...currentState.pastedContent },
|
||
expandedPaste: currentState.expandedPaste
|
||
? { ...currentState.expandedPaste }
|
||
: null,
|
||
};
|
||
const newStack = [...currentState.undoStack, snapshot];
|
||
if (newStack.length > historyLimit) {
|
||
newStack.shift();
|
||
}
|
||
return { ...currentState, undoStack: newStack, redoStack: [] };
|
||
};
|
||
|
||
function generatePastedTextId(
|
||
content: string,
|
||
lineCount: number,
|
||
pastedContent: Record<string, string>,
|
||
): string {
|
||
const base =
|
||
lineCount > LARGE_PASTE_LINE_THRESHOLD
|
||
? `[Pasted Text: ${lineCount} lines]`
|
||
: `[Pasted Text: ${content.length} chars]`;
|
||
|
||
let id = base;
|
||
let suffix = 2;
|
||
while (pastedContent[id]) {
|
||
id = base.replace(']', ` #${suffix}]`);
|
||
suffix++;
|
||
}
|
||
return id;
|
||
}
|
||
|
||
export type TextBufferAction =
|
||
| { type: 'set_text'; payload: string; pushToUndo?: boolean }
|
||
| { type: 'insert'; payload: string; isPaste?: boolean }
|
||
| { type: 'add_pasted_content'; payload: { id: string; text: string } }
|
||
| { type: 'backspace' }
|
||
| {
|
||
type: 'move';
|
||
payload: {
|
||
dir: Direction;
|
||
};
|
||
}
|
||
| {
|
||
type: 'set_cursor';
|
||
payload: {
|
||
cursorRow: number;
|
||
cursorCol: number;
|
||
preferredCol: number | null;
|
||
};
|
||
}
|
||
| { type: 'delete' }
|
||
| { type: 'delete_word_left' }
|
||
| { type: 'delete_word_right' }
|
||
| { type: 'kill_line_right' }
|
||
| { type: 'kill_line_left' }
|
||
| { type: 'undo' }
|
||
| { type: 'redo' }
|
||
| {
|
||
type: 'replace_range';
|
||
payload: {
|
||
startRow: number;
|
||
startCol: number;
|
||
endRow: number;
|
||
endCol: number;
|
||
text: string;
|
||
};
|
||
}
|
||
| { type: 'move_to_offset'; payload: { offset: number } }
|
||
| { type: 'create_undo_snapshot' }
|
||
| { type: 'set_viewport'; payload: { width: number; height: number } }
|
||
| { type: 'vim_delete_word_forward'; payload: { count: number } }
|
||
| { type: 'vim_delete_word_backward'; payload: { count: number } }
|
||
| { type: 'vim_delete_word_end'; payload: { count: number } }
|
||
| { type: 'vim_change_word_forward'; payload: { count: number } }
|
||
| { type: 'vim_change_word_backward'; payload: { count: number } }
|
||
| { type: 'vim_change_word_end'; payload: { count: number } }
|
||
| { type: 'vim_delete_line'; payload: { count: number } }
|
||
| { type: 'vim_change_line'; payload: { count: number } }
|
||
| { type: 'vim_delete_to_end_of_line' }
|
||
| { type: 'vim_change_to_end_of_line' }
|
||
| {
|
||
type: 'vim_change_movement';
|
||
payload: { movement: 'h' | 'j' | 'k' | 'l'; count: number };
|
||
}
|
||
// New vim actions for stateless command handling
|
||
| { type: 'vim_move_left'; payload: { count: number } }
|
||
| { type: 'vim_move_right'; payload: { count: number } }
|
||
| { type: 'vim_move_up'; payload: { count: number } }
|
||
| { type: 'vim_move_down'; payload: { count: number } }
|
||
| { type: 'vim_move_word_forward'; payload: { count: number } }
|
||
| { type: 'vim_move_word_backward'; payload: { count: number } }
|
||
| { type: 'vim_move_word_end'; payload: { count: number } }
|
||
| { type: 'vim_delete_char'; payload: { count: number } }
|
||
| { type: 'vim_insert_at_cursor' }
|
||
| { type: 'vim_append_at_cursor' }
|
||
| { type: 'vim_open_line_below' }
|
||
| { type: 'vim_open_line_above' }
|
||
| { type: 'vim_append_at_line_end' }
|
||
| { type: 'vim_insert_at_line_start' }
|
||
| { type: 'vim_move_to_line_start' }
|
||
| { type: 'vim_move_to_line_end' }
|
||
| { type: 'vim_move_to_first_nonwhitespace' }
|
||
| { type: 'vim_move_to_first_line' }
|
||
| { type: 'vim_move_to_last_line' }
|
||
| { type: 'vim_move_to_line'; payload: { lineNumber: number } }
|
||
| { type: 'vim_escape_insert_mode' }
|
||
| {
|
||
type: 'toggle_paste_expansion';
|
||
payload: { id: string; row: number; col: number };
|
||
};
|
||
|
||
export interface TextBufferOptions {
|
||
inputFilter?: (text: string) => string;
|
||
singleLine?: boolean;
|
||
}
|
||
|
||
function textBufferReducerLogic(
|
||
state: TextBufferState,
|
||
action: TextBufferAction,
|
||
options: TextBufferOptions = {},
|
||
): TextBufferState {
|
||
const pushUndoLocal = pushUndo;
|
||
|
||
const currentLine = (r: number): string => state.lines[r] ?? '';
|
||
const currentLineLen = (r: number): number => cpLen(currentLine(r));
|
||
|
||
switch (action.type) {
|
||
case 'set_text': {
|
||
let nextState = state;
|
||
if (action.pushToUndo !== false) {
|
||
nextState = pushUndoLocal(state);
|
||
}
|
||
const newContentLines = action.payload
|
||
.replace(/\r\n?/g, '\n')
|
||
.split('\n');
|
||
const lines = newContentLines.length === 0 ? [''] : newContentLines;
|
||
const lastNewLineIndex = lines.length - 1;
|
||
return {
|
||
...nextState,
|
||
lines,
|
||
cursorRow: lastNewLineIndex,
|
||
cursorCol: cpLen(lines[lastNewLineIndex] ?? ''),
|
||
preferredCol: null,
|
||
pastedContent: action.payload === '' ? {} : nextState.pastedContent,
|
||
};
|
||
}
|
||
|
||
case 'insert': {
|
||
const nextState = detachExpandedPaste(pushUndoLocal(state));
|
||
const newLines = [...nextState.lines];
|
||
let newCursorRow = nextState.cursorRow;
|
||
let newCursorCol = nextState.cursorCol;
|
||
|
||
const currentLine = (r: number) => newLines[r] ?? '';
|
||
|
||
let payload = action.payload;
|
||
let newPastedContent = nextState.pastedContent;
|
||
|
||
if (action.isPaste) {
|
||
// Normalize line endings for pastes
|
||
payload = payload.replace(/\r\n|\r/g, '\n');
|
||
const lineCount = payload.split('\n').length;
|
||
if (
|
||
lineCount > LARGE_PASTE_LINE_THRESHOLD ||
|
||
payload.length > LARGE_PASTE_CHAR_THRESHOLD
|
||
) {
|
||
const id = generatePastedTextId(payload, lineCount, newPastedContent);
|
||
newPastedContent = {
|
||
...newPastedContent,
|
||
[id]: payload,
|
||
};
|
||
payload = id;
|
||
}
|
||
}
|
||
|
||
if (options.singleLine) {
|
||
payload = payload.replace(/[\r\n]/g, '');
|
||
}
|
||
if (options.inputFilter) {
|
||
payload = options.inputFilter(payload);
|
||
}
|
||
|
||
if (payload.length === 0) {
|
||
return state;
|
||
}
|
||
|
||
const str = stripUnsafeCharacters(
|
||
payload.replace(/\r\n/g, '\n').replace(/\r/g, '\n'),
|
||
);
|
||
const parts = str.split('\n');
|
||
const lineContent = currentLine(newCursorRow);
|
||
const before = cpSlice(lineContent, 0, newCursorCol);
|
||
const after = cpSlice(lineContent, newCursorCol);
|
||
|
||
let lineDelta = 0;
|
||
if (parts.length > 1) {
|
||
newLines[newCursorRow] = before + parts[0];
|
||
const remainingParts = parts.slice(1);
|
||
const lastPartOriginal = remainingParts.pop() ?? '';
|
||
newLines.splice(newCursorRow + 1, 0, ...remainingParts);
|
||
newLines.splice(
|
||
newCursorRow + parts.length - 1,
|
||
0,
|
||
lastPartOriginal + after,
|
||
);
|
||
lineDelta = parts.length - 1;
|
||
newCursorRow = newCursorRow + parts.length - 1;
|
||
newCursorCol = cpLen(lastPartOriginal);
|
||
} else {
|
||
newLines[newCursorRow] = before + parts[0] + after;
|
||
newCursorCol = cpLen(before) + cpLen(parts[0]);
|
||
}
|
||
|
||
const { newInfo: newExpandedPaste, isDetached } = shiftExpandedRegions(
|
||
nextState.expandedPaste,
|
||
nextState.cursorRow,
|
||
lineDelta,
|
||
);
|
||
|
||
if (isDetached && newExpandedPaste === null && nextState.expandedPaste) {
|
||
delete newPastedContent[nextState.expandedPaste.id];
|
||
}
|
||
|
||
return {
|
||
...nextState,
|
||
lines: newLines,
|
||
cursorRow: newCursorRow,
|
||
cursorCol: newCursorCol,
|
||
preferredCol: null,
|
||
pastedContent: newPastedContent,
|
||
expandedPaste: newExpandedPaste,
|
||
};
|
||
}
|
||
|
||
case 'add_pasted_content': {
|
||
const { id, text } = action.payload;
|
||
return {
|
||
...state,
|
||
pastedContent: {
|
||
...state.pastedContent,
|
||
[id]: text,
|
||
},
|
||
};
|
||
}
|
||
|
||
case 'backspace': {
|
||
const stateWithUndo = pushUndoLocal(state);
|
||
const currentState = detachExpandedPaste(stateWithUndo);
|
||
const { cursorRow, cursorCol, lines, transformationsByLine } =
|
||
currentState;
|
||
|
||
// Early return if at start of buffer
|
||
if (cursorCol === 0 && cursorRow === 0) return currentState;
|
||
|
||
// Check if cursor is at end of an atomic placeholder
|
||
const transformations = transformationsByLine[cursorRow] ?? [];
|
||
const placeholder = findAtomicPlaceholderForBackspace(
|
||
lines[cursorRow],
|
||
cursorCol,
|
||
transformations,
|
||
);
|
||
|
||
if (placeholder) {
|
||
const nextState = currentState;
|
||
const newLines = [...nextState.lines];
|
||
newLines[cursorRow] =
|
||
cpSlice(newLines[cursorRow], 0, placeholder.start) +
|
||
cpSlice(newLines[cursorRow], placeholder.end);
|
||
|
||
// Recalculate transformations for the modified line
|
||
const newTransformations = [...nextState.transformationsByLine];
|
||
newTransformations[cursorRow] = calculateTransformationsForLine(
|
||
newLines[cursorRow],
|
||
);
|
||
|
||
// Clean up pastedContent if this was a paste placeholder
|
||
let newPastedContent = nextState.pastedContent;
|
||
if (placeholder.type === 'paste' && placeholder.id) {
|
||
const { [placeholder.id]: _, ...remaining } = nextState.pastedContent;
|
||
newPastedContent = remaining;
|
||
}
|
||
|
||
return {
|
||
...nextState,
|
||
lines: newLines,
|
||
cursorCol: placeholder.start,
|
||
preferredCol: null,
|
||
transformationsByLine: newTransformations,
|
||
pastedContent: newPastedContent,
|
||
};
|
||
}
|
||
|
||
// Standard backspace logic
|
||
const nextState = currentState;
|
||
const newLines = [...nextState.lines];
|
||
let newCursorRow = nextState.cursorRow;
|
||
let newCursorCol = nextState.cursorCol;
|
||
|
||
const currentLine = (r: number) => newLines[r] ?? '';
|
||
|
||
let lineDelta = 0;
|
||
if (newCursorCol > 0) {
|
||
const lineContent = currentLine(newCursorRow);
|
||
newLines[newCursorRow] =
|
||
cpSlice(lineContent, 0, newCursorCol - 1) +
|
||
cpSlice(lineContent, newCursorCol);
|
||
newCursorCol--;
|
||
} else if (newCursorRow > 0) {
|
||
const prevLineContent = currentLine(newCursorRow - 1);
|
||
const currentLineContentVal = currentLine(newCursorRow);
|
||
const newCol = cpLen(prevLineContent);
|
||
newLines[newCursorRow - 1] = prevLineContent + currentLineContentVal;
|
||
newLines.splice(newCursorRow, 1);
|
||
lineDelta = -1;
|
||
newCursorRow--;
|
||
newCursorCol = newCol;
|
||
}
|
||
|
||
const { newInfo: newExpandedPaste, isDetached } = shiftExpandedRegions(
|
||
nextState.expandedPaste,
|
||
nextState.cursorRow + lineDelta, // shift based on the line that was removed
|
||
lineDelta,
|
||
nextState.cursorRow,
|
||
);
|
||
|
||
const newPastedContent = { ...nextState.pastedContent };
|
||
if (isDetached && nextState.expandedPaste) {
|
||
delete newPastedContent[nextState.expandedPaste.id];
|
||
}
|
||
|
||
return {
|
||
...nextState,
|
||
lines: newLines,
|
||
cursorRow: newCursorRow,
|
||
cursorCol: newCursorCol,
|
||
preferredCol: null,
|
||
pastedContent: newPastedContent,
|
||
expandedPaste: newExpandedPaste,
|
||
};
|
||
}
|
||
|
||
case 'set_viewport': {
|
||
const { width, height } = action.payload;
|
||
if (width === state.viewportWidth && height === state.viewportHeight) {
|
||
return state;
|
||
}
|
||
return {
|
||
...state,
|
||
viewportWidth: width,
|
||
viewportHeight: height,
|
||
};
|
||
}
|
||
|
||
case 'move': {
|
||
const { dir } = action.payload;
|
||
const { cursorRow, cursorCol, lines, visualLayout, preferredCol } = state;
|
||
|
||
// Visual movements
|
||
if (
|
||
dir === 'left' ||
|
||
dir === 'right' ||
|
||
dir === 'up' ||
|
||
dir === 'down' ||
|
||
dir === 'home' ||
|
||
dir === 'end'
|
||
) {
|
||
const visualCursor = calculateVisualCursorFromLayout(visualLayout, [
|
||
cursorRow,
|
||
cursorCol,
|
||
]);
|
||
const { visualLines, visualToLogicalMap } = visualLayout;
|
||
|
||
let newVisualRow = visualCursor[0];
|
||
let newVisualCol = visualCursor[1];
|
||
let newPreferredCol = preferredCol;
|
||
|
||
const currentVisLineLen = cpLen(visualLines[newVisualRow] ?? '');
|
||
|
||
switch (dir) {
|
||
case 'left':
|
||
newPreferredCol = null;
|
||
if (newVisualCol > 0) {
|
||
newVisualCol--;
|
||
} else if (newVisualRow > 0) {
|
||
newVisualRow--;
|
||
newVisualCol = cpLen(visualLines[newVisualRow] ?? '');
|
||
}
|
||
break;
|
||
case 'right':
|
||
newPreferredCol = null;
|
||
if (newVisualCol < currentVisLineLen) {
|
||
newVisualCol++;
|
||
} else if (newVisualRow < visualLines.length - 1) {
|
||
newVisualRow++;
|
||
newVisualCol = 0;
|
||
}
|
||
break;
|
||
case 'up':
|
||
if (newVisualRow > 0) {
|
||
if (newPreferredCol === null) newPreferredCol = newVisualCol;
|
||
newVisualRow--;
|
||
newVisualCol = clamp(
|
||
newPreferredCol,
|
||
0,
|
||
cpLen(visualLines[newVisualRow] ?? ''),
|
||
);
|
||
}
|
||
break;
|
||
case 'down':
|
||
if (newVisualRow < visualLines.length - 1) {
|
||
if (newPreferredCol === null) newPreferredCol = newVisualCol;
|
||
newVisualRow++;
|
||
newVisualCol = clamp(
|
||
newPreferredCol,
|
||
0,
|
||
cpLen(visualLines[newVisualRow] ?? ''),
|
||
);
|
||
}
|
||
break;
|
||
case 'home':
|
||
newPreferredCol = null;
|
||
newVisualCol = 0;
|
||
break;
|
||
case 'end':
|
||
newPreferredCol = null;
|
||
newVisualCol = currentVisLineLen;
|
||
break;
|
||
default: {
|
||
const exhaustiveCheck: never = dir;
|
||
debugLogger.error(
|
||
`Unknown visual movement direction: ${exhaustiveCheck}`,
|
||
);
|
||
return state;
|
||
}
|
||
}
|
||
|
||
if (visualToLogicalMap[newVisualRow]) {
|
||
const [logRow, logicalStartCol] = visualToLogicalMap[newVisualRow];
|
||
const transformedToLogicalMap =
|
||
visualLayout.transformedToLogicalMaps?.[logRow] ?? [];
|
||
let transformedStartCol = 0;
|
||
while (
|
||
transformedStartCol < transformedToLogicalMap.length &&
|
||
transformedToLogicalMap[transformedStartCol] < logicalStartCol
|
||
) {
|
||
transformedStartCol++;
|
||
}
|
||
const clampedTransformedCol = Math.min(
|
||
transformedStartCol + newVisualCol,
|
||
Math.max(0, transformedToLogicalMap.length - 1),
|
||
);
|
||
const newLogicalCol =
|
||
transformedToLogicalMap[clampedTransformedCol] ??
|
||
cpLen(lines[logRow] ?? '');
|
||
return {
|
||
...state,
|
||
cursorRow: logRow,
|
||
cursorCol: newLogicalCol,
|
||
preferredCol: newPreferredCol,
|
||
};
|
||
}
|
||
return state;
|
||
}
|
||
|
||
// Logical movements
|
||
switch (dir) {
|
||
case 'wordLeft': {
|
||
if (cursorCol === 0 && cursorRow === 0) return state;
|
||
|
||
let newCursorRow = cursorRow;
|
||
let newCursorCol = cursorCol;
|
||
|
||
if (cursorCol === 0) {
|
||
newCursorRow--;
|
||
newCursorCol = cpLen(lines[newCursorRow] ?? '');
|
||
} else {
|
||
const lineContent = lines[cursorRow];
|
||
newCursorCol = findPrevWordBoundary(lineContent, cursorCol);
|
||
}
|
||
return {
|
||
...state,
|
||
cursorRow: newCursorRow,
|
||
cursorCol: newCursorCol,
|
||
preferredCol: null,
|
||
};
|
||
}
|
||
case 'wordRight': {
|
||
const lineContent = lines[cursorRow] ?? '';
|
||
if (
|
||
cursorRow === lines.length - 1 &&
|
||
cursorCol === cpLen(lineContent)
|
||
) {
|
||
return state;
|
||
}
|
||
|
||
let newCursorRow = cursorRow;
|
||
let newCursorCol = cursorCol;
|
||
const lineLen = cpLen(lineContent);
|
||
|
||
if (cursorCol >= lineLen) {
|
||
newCursorRow++;
|
||
newCursorCol = 0;
|
||
} else {
|
||
newCursorCol = findNextWordBoundary(lineContent, cursorCol);
|
||
}
|
||
return {
|
||
...state,
|
||
cursorRow: newCursorRow,
|
||
cursorCol: newCursorCol,
|
||
preferredCol: null,
|
||
};
|
||
}
|
||
default:
|
||
return state;
|
||
}
|
||
}
|
||
|
||
case 'set_cursor': {
|
||
return {
|
||
...state,
|
||
...action.payload,
|
||
};
|
||
}
|
||
|
||
case 'delete': {
|
||
const stateWithUndo = pushUndoLocal(state);
|
||
const currentState = detachExpandedPaste(stateWithUndo);
|
||
const { cursorRow, cursorCol, lines, transformationsByLine } =
|
||
currentState;
|
||
|
||
// Check if cursor is at start of an atomic placeholder
|
||
const transformations = transformationsByLine[cursorRow] ?? [];
|
||
const placeholder = findAtomicPlaceholderForDelete(
|
||
lines[cursorRow],
|
||
cursorCol,
|
||
transformations,
|
||
);
|
||
|
||
if (placeholder) {
|
||
const nextState = currentState;
|
||
const newLines = [...nextState.lines];
|
||
newLines[cursorRow] =
|
||
cpSlice(newLines[cursorRow], 0, placeholder.start) +
|
||
cpSlice(newLines[cursorRow], placeholder.end);
|
||
|
||
// Recalculate transformations for the modified line
|
||
const newTransformations = [...nextState.transformationsByLine];
|
||
newTransformations[cursorRow] = calculateTransformationsForLine(
|
||
newLines[cursorRow],
|
||
);
|
||
|
||
// Clean up pastedContent if this was a paste placeholder
|
||
let newPastedContent = nextState.pastedContent;
|
||
if (placeholder.type === 'paste' && placeholder.id) {
|
||
const { [placeholder.id]: _, ...remaining } = nextState.pastedContent;
|
||
newPastedContent = remaining;
|
||
}
|
||
|
||
return {
|
||
...nextState,
|
||
lines: newLines,
|
||
// cursorCol stays the same
|
||
preferredCol: null,
|
||
transformationsByLine: newTransformations,
|
||
pastedContent: newPastedContent,
|
||
};
|
||
}
|
||
|
||
// Standard delete logic
|
||
const lineContent = currentLine(cursorRow);
|
||
let lineDelta = 0;
|
||
const nextState = currentState;
|
||
const newLines = [...nextState.lines];
|
||
|
||
if (cursorCol < currentLineLen(cursorRow)) {
|
||
newLines[cursorRow] =
|
||
cpSlice(lineContent, 0, cursorCol) +
|
||
cpSlice(lineContent, cursorCol + 1);
|
||
} else if (cursorRow < lines.length - 1) {
|
||
const nextLineContent = currentLine(cursorRow + 1);
|
||
newLines[cursorRow] = lineContent + nextLineContent;
|
||
newLines.splice(cursorRow + 1, 1);
|
||
lineDelta = -1;
|
||
} else {
|
||
return currentState;
|
||
}
|
||
|
||
const { newInfo: newExpandedPaste, isDetached } = shiftExpandedRegions(
|
||
nextState.expandedPaste,
|
||
nextState.cursorRow,
|
||
lineDelta,
|
||
nextState.cursorRow + (lineDelta < 0 ? 1 : 0),
|
||
);
|
||
|
||
const newPastedContent = { ...nextState.pastedContent };
|
||
if (isDetached && nextState.expandedPaste) {
|
||
delete newPastedContent[nextState.expandedPaste.id];
|
||
}
|
||
|
||
return {
|
||
...nextState,
|
||
lines: newLines,
|
||
preferredCol: null,
|
||
pastedContent: newPastedContent,
|
||
expandedPaste: newExpandedPaste,
|
||
};
|
||
}
|
||
|
||
case 'delete_word_left': {
|
||
const stateWithUndo = pushUndoLocal(state);
|
||
const currentState = detachExpandedPaste(stateWithUndo);
|
||
const { cursorRow, cursorCol } = currentState;
|
||
if (cursorCol === 0 && cursorRow === 0) return currentState;
|
||
|
||
const nextState = currentState;
|
||
const newLines = [...nextState.lines];
|
||
let newCursorRow = cursorRow;
|
||
let newCursorCol = cursorCol;
|
||
|
||
if (newCursorCol > 0) {
|
||
const lineContent = currentLine(newCursorRow);
|
||
const prevWordStart = findPrevWordStartInLine(
|
||
lineContent,
|
||
newCursorCol,
|
||
);
|
||
const start = prevWordStart === null ? 0 : prevWordStart;
|
||
newLines[newCursorRow] =
|
||
cpSlice(lineContent, 0, start) + cpSlice(lineContent, newCursorCol);
|
||
newCursorCol = start;
|
||
} else {
|
||
// Act as a backspace
|
||
const prevLineContent = currentLine(cursorRow - 1);
|
||
const currentLineContentVal = currentLine(cursorRow);
|
||
const newCol = cpLen(prevLineContent);
|
||
newLines[cursorRow - 1] = prevLineContent + currentLineContentVal;
|
||
newLines.splice(cursorRow, 1);
|
||
newCursorRow--;
|
||
newCursorCol = newCol;
|
||
}
|
||
|
||
return {
|
||
...nextState,
|
||
lines: newLines,
|
||
cursorRow: newCursorRow,
|
||
cursorCol: newCursorCol,
|
||
preferredCol: null,
|
||
};
|
||
}
|
||
|
||
case 'delete_word_right': {
|
||
const stateWithUndo = pushUndoLocal(state);
|
||
const currentState = detachExpandedPaste(stateWithUndo);
|
||
const { cursorRow, cursorCol, lines } = currentState;
|
||
const lineContent = currentLine(cursorRow);
|
||
const lineLen = cpLen(lineContent);
|
||
|
||
if (cursorCol >= lineLen && cursorRow === lines.length - 1) {
|
||
return currentState;
|
||
}
|
||
|
||
const nextState = currentState;
|
||
const newLines = [...nextState.lines];
|
||
|
||
if (cursorCol >= lineLen) {
|
||
// Act as a delete, joining with the next line
|
||
const nextLineContent = currentLine(cursorRow + 1);
|
||
newLines[cursorRow] = lineContent + nextLineContent;
|
||
newLines.splice(cursorRow + 1, 1);
|
||
} else {
|
||
const nextWordStart = findNextWordStartInLine(lineContent, cursorCol);
|
||
const end = nextWordStart === null ? lineLen : nextWordStart;
|
||
newLines[cursorRow] =
|
||
cpSlice(lineContent, 0, cursorCol) + cpSlice(lineContent, end);
|
||
}
|
||
|
||
return {
|
||
...nextState,
|
||
lines: newLines,
|
||
preferredCol: null,
|
||
};
|
||
}
|
||
|
||
case 'kill_line_right': {
|
||
const stateWithUndo = pushUndoLocal(state);
|
||
const currentState = detachExpandedPaste(stateWithUndo);
|
||
const { cursorRow, cursorCol, lines } = currentState;
|
||
const lineContent = currentLine(cursorRow);
|
||
if (cursorCol < currentLineLen(cursorRow)) {
|
||
const nextState = currentState;
|
||
const newLines = [...nextState.lines];
|
||
newLines[cursorRow] = cpSlice(lineContent, 0, cursorCol);
|
||
return {
|
||
...nextState,
|
||
lines: newLines,
|
||
};
|
||
} else if (cursorRow < lines.length - 1) {
|
||
// Act as a delete
|
||
const nextState = currentState;
|
||
const nextLineContent = currentLine(cursorRow + 1);
|
||
const newLines = [...nextState.lines];
|
||
newLines[cursorRow] = lineContent + nextLineContent;
|
||
newLines.splice(cursorRow + 1, 1);
|
||
return {
|
||
...nextState,
|
||
lines: newLines,
|
||
preferredCol: null,
|
||
};
|
||
}
|
||
return currentState;
|
||
}
|
||
|
||
case 'kill_line_left': {
|
||
const stateWithUndo = pushUndoLocal(state);
|
||
const currentState = detachExpandedPaste(stateWithUndo);
|
||
const { cursorRow, cursorCol } = currentState;
|
||
if (cursorCol > 0) {
|
||
const nextState = currentState;
|
||
const lineContent = currentLine(cursorRow);
|
||
const newLines = [...nextState.lines];
|
||
newLines[cursorRow] = cpSlice(lineContent, cursorCol);
|
||
return {
|
||
...nextState,
|
||
lines: newLines,
|
||
cursorCol: 0,
|
||
preferredCol: null,
|
||
};
|
||
}
|
||
return currentState;
|
||
}
|
||
|
||
case 'undo': {
|
||
const stateToRestore = state.undoStack[state.undoStack.length - 1];
|
||
if (!stateToRestore) return state;
|
||
|
||
const currentSnapshot: UndoHistoryEntry = {
|
||
lines: [...state.lines],
|
||
cursorRow: state.cursorRow,
|
||
cursorCol: state.cursorCol,
|
||
pastedContent: { ...state.pastedContent },
|
||
expandedPaste: state.expandedPaste ? { ...state.expandedPaste } : null,
|
||
};
|
||
return {
|
||
...state,
|
||
...stateToRestore,
|
||
undoStack: state.undoStack.slice(0, -1),
|
||
redoStack: [...state.redoStack, currentSnapshot],
|
||
};
|
||
}
|
||
|
||
case 'redo': {
|
||
const stateToRestore = state.redoStack[state.redoStack.length - 1];
|
||
if (!stateToRestore) return state;
|
||
|
||
const currentSnapshot: UndoHistoryEntry = {
|
||
lines: [...state.lines],
|
||
cursorRow: state.cursorRow,
|
||
cursorCol: state.cursorCol,
|
||
pastedContent: { ...state.pastedContent },
|
||
expandedPaste: state.expandedPaste ? { ...state.expandedPaste } : null,
|
||
};
|
||
return {
|
||
...state,
|
||
...stateToRestore,
|
||
redoStack: state.redoStack.slice(0, -1),
|
||
undoStack: [...state.undoStack, currentSnapshot],
|
||
};
|
||
}
|
||
|
||
case 'replace_range': {
|
||
const { startRow, startCol, endRow, endCol, text } = action.payload;
|
||
const nextState = pushUndoLocal(state);
|
||
const newState = replaceRangeInternal(
|
||
nextState,
|
||
startRow,
|
||
startCol,
|
||
endRow,
|
||
endCol,
|
||
text,
|
||
);
|
||
|
||
const oldLineCount = endRow - startRow + 1;
|
||
const newLineCount =
|
||
newState.lines.length - (nextState.lines.length - oldLineCount);
|
||
const lineDelta = newLineCount - oldLineCount;
|
||
|
||
const { newInfo: newExpandedPaste, isDetached } = shiftExpandedRegions(
|
||
nextState.expandedPaste,
|
||
startRow,
|
||
lineDelta,
|
||
endRow,
|
||
);
|
||
|
||
const newPastedContent = { ...newState.pastedContent };
|
||
if (isDetached && nextState.expandedPaste) {
|
||
delete newPastedContent[nextState.expandedPaste.id];
|
||
}
|
||
|
||
return {
|
||
...newState,
|
||
pastedContent: newPastedContent,
|
||
expandedPaste: newExpandedPaste,
|
||
};
|
||
}
|
||
|
||
case 'move_to_offset': {
|
||
const { offset } = action.payload;
|
||
const [newRow, newCol] = offsetToLogicalPos(
|
||
state.lines.join('\n'),
|
||
offset,
|
||
);
|
||
return {
|
||
...state,
|
||
cursorRow: newRow,
|
||
cursorCol: newCol,
|
||
preferredCol: null,
|
||
};
|
||
}
|
||
|
||
case 'create_undo_snapshot': {
|
||
return pushUndoLocal(state);
|
||
}
|
||
|
||
// Vim-specific operations
|
||
case 'vim_delete_word_forward':
|
||
case 'vim_delete_word_backward':
|
||
case 'vim_delete_word_end':
|
||
case 'vim_change_word_forward':
|
||
case 'vim_change_word_backward':
|
||
case 'vim_change_word_end':
|
||
case 'vim_delete_line':
|
||
case 'vim_change_line':
|
||
case 'vim_delete_to_end_of_line':
|
||
case 'vim_change_to_end_of_line':
|
||
case 'vim_change_movement':
|
||
case 'vim_move_left':
|
||
case 'vim_move_right':
|
||
case 'vim_move_up':
|
||
case 'vim_move_down':
|
||
case 'vim_move_word_forward':
|
||
case 'vim_move_word_backward':
|
||
case 'vim_move_word_end':
|
||
case 'vim_delete_char':
|
||
case 'vim_insert_at_cursor':
|
||
case 'vim_append_at_cursor':
|
||
case 'vim_open_line_below':
|
||
case 'vim_open_line_above':
|
||
case 'vim_append_at_line_end':
|
||
case 'vim_insert_at_line_start':
|
||
case 'vim_move_to_line_start':
|
||
case 'vim_move_to_line_end':
|
||
case 'vim_move_to_first_nonwhitespace':
|
||
case 'vim_move_to_first_line':
|
||
case 'vim_move_to_last_line':
|
||
case 'vim_move_to_line':
|
||
case 'vim_escape_insert_mode':
|
||
return handleVimAction(state, action as VimAction);
|
||
|
||
case 'toggle_paste_expansion': {
|
||
const { id, row, col } = action.payload;
|
||
const expandedPaste = state.expandedPaste;
|
||
|
||
if (expandedPaste && expandedPaste.id === id) {
|
||
const nextState = pushUndoLocal(state);
|
||
// COLLAPSE: Restore original line with placeholder
|
||
const newLines = [...nextState.lines];
|
||
newLines.splice(
|
||
expandedPaste.startLine,
|
||
expandedPaste.lineCount,
|
||
expandedPaste.prefix + id + expandedPaste.suffix,
|
||
);
|
||
|
||
// Move cursor to end of collapsed placeholder
|
||
const newCursorRow = expandedPaste.startLine;
|
||
const newCursorCol = cpLen(expandedPaste.prefix) + cpLen(id);
|
||
|
||
return {
|
||
...nextState,
|
||
lines: newLines,
|
||
cursorRow: newCursorRow,
|
||
cursorCol: newCursorCol,
|
||
preferredCol: null,
|
||
expandedPaste: null,
|
||
};
|
||
} else {
|
||
// EXPAND: Replace placeholder with content
|
||
|
||
// Collapse any existing expanded paste first
|
||
let currentState = state;
|
||
let targetRow = row;
|
||
if (state.expandedPaste) {
|
||
const existingInfo = state.expandedPaste;
|
||
const lineDelta = 1 - existingInfo.lineCount;
|
||
|
||
if (targetRow !== undefined && targetRow > existingInfo.startLine) {
|
||
// If we collapsed something above our target, our target row shifted up
|
||
targetRow += lineDelta;
|
||
}
|
||
|
||
currentState = textBufferReducerLogic(state, {
|
||
type: 'toggle_paste_expansion',
|
||
payload: {
|
||
id: existingInfo.id,
|
||
row: existingInfo.startLine,
|
||
col: 0,
|
||
},
|
||
});
|
||
// Update transformations because they are needed for finding the next placeholder
|
||
currentState.transformationsByLine = calculateTransformations(
|
||
currentState.lines,
|
||
);
|
||
}
|
||
|
||
const content = currentState.pastedContent[id];
|
||
if (!content) return currentState;
|
||
|
||
// Find line and position containing exactly this placeholder
|
||
let lineIndex = -1;
|
||
let placeholderStart = -1;
|
||
|
||
const tryFindOnLine = (idx: number) => {
|
||
const transforms = currentState.transformationsByLine[idx] ?? [];
|
||
|
||
// Precise match by col
|
||
let transform = transforms.find(
|
||
(t) =>
|
||
t.type === 'paste' &&
|
||
t.id === id &&
|
||
col >= t.logStart &&
|
||
col <= t.logEnd,
|
||
);
|
||
|
||
if (!transform) {
|
||
// Fallback to first match on line
|
||
transform = transforms.find(
|
||
(t) => t.type === 'paste' && t.id === id,
|
||
);
|
||
}
|
||
|
||
if (transform) {
|
||
lineIndex = idx;
|
||
placeholderStart = transform.logStart;
|
||
return true;
|
||
}
|
||
return false;
|
||
};
|
||
|
||
// Try provided row first for precise targeting
|
||
if (targetRow >= 0 && targetRow < currentState.lines.length) {
|
||
tryFindOnLine(targetRow);
|
||
}
|
||
|
||
if (lineIndex === -1) {
|
||
for (let i = 0; i < currentState.lines.length; i++) {
|
||
if (tryFindOnLine(i)) break;
|
||
}
|
||
}
|
||
|
||
if (lineIndex === -1) return currentState;
|
||
|
||
const nextState = pushUndoLocal(currentState);
|
||
|
||
const line = nextState.lines[lineIndex];
|
||
const prefix = cpSlice(line, 0, placeholderStart);
|
||
const suffix = cpSlice(line, placeholderStart + cpLen(id));
|
||
|
||
// Split content into lines
|
||
const contentLines = content.split('\n');
|
||
const newLines = [...nextState.lines];
|
||
|
||
let expandedLines: string[];
|
||
if (contentLines.length === 1) {
|
||
// Single-line content
|
||
expandedLines = [prefix + contentLines[0] + suffix];
|
||
} else {
|
||
// Multi-line content
|
||
expandedLines = [
|
||
prefix + contentLines[0],
|
||
...contentLines.slice(1, -1),
|
||
contentLines[contentLines.length - 1] + suffix,
|
||
];
|
||
}
|
||
|
||
newLines.splice(lineIndex, 1, ...expandedLines);
|
||
|
||
// Move cursor to end of expanded content (before suffix)
|
||
const newCursorRow = lineIndex + expandedLines.length - 1;
|
||
const lastExpandedLine = expandedLines[expandedLines.length - 1];
|
||
const newCursorCol = cpLen(lastExpandedLine) - cpLen(suffix);
|
||
|
||
return {
|
||
...nextState,
|
||
lines: newLines,
|
||
cursorRow: newCursorRow,
|
||
cursorCol: newCursorCol,
|
||
preferredCol: null,
|
||
expandedPaste: {
|
||
id,
|
||
startLine: lineIndex,
|
||
lineCount: expandedLines.length,
|
||
prefix,
|
||
suffix,
|
||
},
|
||
};
|
||
}
|
||
}
|
||
|
||
default: {
|
||
const exhaustiveCheck: never = action;
|
||
debugLogger.error(`Unknown action encountered: ${exhaustiveCheck}`);
|
||
return state;
|
||
}
|
||
}
|
||
}
|
||
|
||
export function textBufferReducer(
|
||
state: TextBufferState,
|
||
action: TextBufferAction,
|
||
options: TextBufferOptions = {},
|
||
): TextBufferState {
|
||
const newState = textBufferReducerLogic(state, action, options);
|
||
|
||
const newTransformedLines =
|
||
newState.lines !== state.lines
|
||
? calculateTransformations(newState.lines)
|
||
: state.transformationsByLine;
|
||
|
||
const oldTransform = getTransformUnderCursor(
|
||
state.cursorRow,
|
||
state.cursorCol,
|
||
state.transformationsByLine,
|
||
);
|
||
const newTransform = getTransformUnderCursor(
|
||
newState.cursorRow,
|
||
newState.cursorCol,
|
||
newTransformedLines,
|
||
);
|
||
const oldInside = oldTransform !== null;
|
||
const newInside = newTransform !== null;
|
||
const movedBetweenTransforms =
|
||
oldTransform !== newTransform &&
|
||
(oldTransform !== null || newTransform !== null);
|
||
|
||
if (
|
||
newState.lines !== state.lines ||
|
||
newState.viewportWidth !== state.viewportWidth ||
|
||
oldInside !== newInside ||
|
||
movedBetweenTransforms
|
||
) {
|
||
const shouldResetPreferred =
|
||
oldInside !== newInside || movedBetweenTransforms;
|
||
|
||
return {
|
||
...newState,
|
||
preferredCol: shouldResetPreferred ? null : newState.preferredCol,
|
||
visualLayout: calculateLayout(newState.lines, newState.viewportWidth, [
|
||
newState.cursorRow,
|
||
newState.cursorCol,
|
||
]),
|
||
transformationsByLine: newTransformedLines,
|
||
};
|
||
}
|
||
|
||
return newState;
|
||
}
|
||
|
||
// --- End of reducer logic ---
|
||
|
||
export function useTextBuffer({
|
||
initialText = '',
|
||
initialCursorOffset = 0,
|
||
viewport,
|
||
stdin,
|
||
setRawMode,
|
||
onChange,
|
||
isValidPath,
|
||
shellModeActive = false,
|
||
inputFilter,
|
||
singleLine = false,
|
||
getPreferredEditor,
|
||
}: UseTextBufferProps): TextBuffer {
|
||
const initialState = useMemo((): TextBufferState => {
|
||
const lines = initialText.split('\n');
|
||
const [initialCursorRow, initialCursorCol] = calculateInitialCursorPosition(
|
||
lines.length === 0 ? [''] : lines,
|
||
initialCursorOffset,
|
||
);
|
||
const transformationsByLine = calculateTransformations(
|
||
lines.length === 0 ? [''] : lines,
|
||
);
|
||
const visualLayout = calculateLayout(
|
||
lines.length === 0 ? [''] : lines,
|
||
viewport.width,
|
||
[initialCursorRow, initialCursorCol],
|
||
);
|
||
return {
|
||
lines: lines.length === 0 ? [''] : lines,
|
||
cursorRow: initialCursorRow,
|
||
cursorCol: initialCursorCol,
|
||
transformationsByLine,
|
||
preferredCol: null,
|
||
undoStack: [],
|
||
redoStack: [],
|
||
clipboard: null,
|
||
selectionAnchor: null,
|
||
viewportWidth: viewport.width,
|
||
viewportHeight: viewport.height,
|
||
visualLayout,
|
||
pastedContent: {},
|
||
expandedPaste: null,
|
||
};
|
||
}, [initialText, initialCursorOffset, viewport.width, viewport.height]);
|
||
|
||
const [state, dispatch] = useReducer(
|
||
(s: TextBufferState, a: TextBufferAction) =>
|
||
textBufferReducer(s, a, { inputFilter, singleLine }),
|
||
initialState,
|
||
);
|
||
const {
|
||
lines,
|
||
cursorRow,
|
||
cursorCol,
|
||
preferredCol,
|
||
selectionAnchor,
|
||
visualLayout,
|
||
transformationsByLine,
|
||
pastedContent,
|
||
expandedPaste,
|
||
} = state;
|
||
|
||
const text = useMemo(() => lines.join('\n'), [lines]);
|
||
|
||
const visualCursor = useMemo(
|
||
() => calculateVisualCursorFromLayout(visualLayout, [cursorRow, cursorCol]),
|
||
[visualLayout, cursorRow, cursorCol],
|
||
);
|
||
|
||
const {
|
||
visualLines,
|
||
visualToLogicalMap,
|
||
transformedToLogicalMaps,
|
||
visualToTransformedMap,
|
||
} = visualLayout;
|
||
|
||
const [scrollRowState, setScrollRowState] = useState<number>(0);
|
||
|
||
useEffect(() => {
|
||
if (onChange) {
|
||
onChange(text);
|
||
}
|
||
}, [text, onChange]);
|
||
|
||
useEffect(() => {
|
||
dispatch({
|
||
type: 'set_viewport',
|
||
payload: { width: viewport.width, height: viewport.height },
|
||
});
|
||
}, [viewport.width, viewport.height]);
|
||
|
||
// Update visual scroll (vertical)
|
||
useEffect(() => {
|
||
const { height } = viewport;
|
||
const totalVisualLines = visualLines.length;
|
||
const maxScrollStart = Math.max(0, totalVisualLines - height);
|
||
let newVisualScrollRow = scrollRowState;
|
||
|
||
if (visualCursor[0] < scrollRowState) {
|
||
newVisualScrollRow = visualCursor[0];
|
||
} else if (visualCursor[0] >= scrollRowState + height) {
|
||
newVisualScrollRow = visualCursor[0] - height + 1;
|
||
}
|
||
|
||
// When the number of visual lines shrinks (e.g., after widening the viewport),
|
||
// ensure scroll never starts beyond the last valid start so we can render a full window.
|
||
newVisualScrollRow = clamp(newVisualScrollRow, 0, maxScrollStart);
|
||
|
||
if (newVisualScrollRow !== scrollRowState) {
|
||
setScrollRowState(newVisualScrollRow);
|
||
}
|
||
}, [visualCursor, scrollRowState, viewport, visualLines.length]);
|
||
|
||
const insert = useCallback(
|
||
(ch: string, { paste = false }: { paste?: boolean } = {}): void => {
|
||
if (typeof ch !== 'string') {
|
||
return;
|
||
}
|
||
|
||
let textToInsert = ch;
|
||
const minLengthToInferAsDragDrop = 3;
|
||
if (
|
||
ch.length >= minLengthToInferAsDragDrop &&
|
||
!shellModeActive &&
|
||
paste
|
||
) {
|
||
let potentialPath = ch.trim();
|
||
const quoteMatch = potentialPath.match(/^'(.*)'$/);
|
||
if (quoteMatch) {
|
||
potentialPath = quoteMatch[1];
|
||
}
|
||
|
||
potentialPath = potentialPath.trim();
|
||
|
||
const processed = parsePastedPaths(potentialPath, isValidPath);
|
||
if (processed) {
|
||
textToInsert = processed;
|
||
}
|
||
}
|
||
|
||
let currentText = '';
|
||
for (const char of toCodePoints(textToInsert)) {
|
||
if (char.codePointAt(0) === 127) {
|
||
if (currentText.length > 0) {
|
||
dispatch({ type: 'insert', payload: currentText, isPaste: paste });
|
||
currentText = '';
|
||
}
|
||
dispatch({ type: 'backspace' });
|
||
} else {
|
||
currentText += char;
|
||
}
|
||
}
|
||
if (currentText.length > 0) {
|
||
dispatch({ type: 'insert', payload: currentText, isPaste: paste });
|
||
}
|
||
},
|
||
[isValidPath, shellModeActive],
|
||
);
|
||
|
||
const newline = useCallback((): void => {
|
||
if (singleLine) {
|
||
return;
|
||
}
|
||
dispatch({ type: 'insert', payload: '\n' });
|
||
}, [singleLine]);
|
||
|
||
const backspace = useCallback((): void => {
|
||
dispatch({ type: 'backspace' });
|
||
}, []);
|
||
|
||
const del = useCallback((): void => {
|
||
dispatch({ type: 'delete' });
|
||
}, []);
|
||
|
||
const move = useCallback(
|
||
(dir: Direction): void => {
|
||
dispatch({ type: 'move', payload: { dir } });
|
||
},
|
||
[dispatch],
|
||
);
|
||
|
||
const undo = useCallback((): void => {
|
||
dispatch({ type: 'undo' });
|
||
}, []);
|
||
|
||
const redo = useCallback((): void => {
|
||
dispatch({ type: 'redo' });
|
||
}, []);
|
||
|
||
const setText = useCallback((newText: string): void => {
|
||
dispatch({ type: 'set_text', payload: newText });
|
||
}, []);
|
||
|
||
const deleteWordLeft = useCallback((): void => {
|
||
dispatch({ type: 'delete_word_left' });
|
||
}, []);
|
||
|
||
const deleteWordRight = useCallback((): void => {
|
||
dispatch({ type: 'delete_word_right' });
|
||
}, []);
|
||
|
||
const killLineRight = useCallback((): void => {
|
||
dispatch({ type: 'kill_line_right' });
|
||
}, []);
|
||
|
||
const killLineLeft = useCallback((): void => {
|
||
dispatch({ type: 'kill_line_left' });
|
||
}, []);
|
||
|
||
// Vim-specific operations
|
||
const vimDeleteWordForward = useCallback((count: number): void => {
|
||
dispatch({ type: 'vim_delete_word_forward', payload: { count } });
|
||
}, []);
|
||
|
||
const vimDeleteWordBackward = useCallback((count: number): void => {
|
||
dispatch({ type: 'vim_delete_word_backward', payload: { count } });
|
||
}, []);
|
||
|
||
const vimDeleteWordEnd = useCallback((count: number): void => {
|
||
dispatch({ type: 'vim_delete_word_end', payload: { count } });
|
||
}, []);
|
||
|
||
const vimChangeWordForward = useCallback((count: number): void => {
|
||
dispatch({ type: 'vim_change_word_forward', payload: { count } });
|
||
}, []);
|
||
|
||
const vimChangeWordBackward = useCallback((count: number): void => {
|
||
dispatch({ type: 'vim_change_word_backward', payload: { count } });
|
||
}, []);
|
||
|
||
const vimChangeWordEnd = useCallback((count: number): void => {
|
||
dispatch({ type: 'vim_change_word_end', payload: { count } });
|
||
}, []);
|
||
|
||
const vimDeleteLine = useCallback((count: number): void => {
|
||
dispatch({ type: 'vim_delete_line', payload: { count } });
|
||
}, []);
|
||
|
||
const vimChangeLine = useCallback((count: number): void => {
|
||
dispatch({ type: 'vim_change_line', payload: { count } });
|
||
}, []);
|
||
|
||
const vimDeleteToEndOfLine = useCallback((): void => {
|
||
dispatch({ type: 'vim_delete_to_end_of_line' });
|
||
}, []);
|
||
|
||
const vimChangeToEndOfLine = useCallback((): void => {
|
||
dispatch({ type: 'vim_change_to_end_of_line' });
|
||
}, []);
|
||
|
||
const vimChangeMovement = useCallback(
|
||
(movement: 'h' | 'j' | 'k' | 'l', count: number): void => {
|
||
dispatch({ type: 'vim_change_movement', payload: { movement, count } });
|
||
},
|
||
[],
|
||
);
|
||
|
||
// New vim navigation and operation methods
|
||
const vimMoveLeft = useCallback((count: number): void => {
|
||
dispatch({ type: 'vim_move_left', payload: { count } });
|
||
}, []);
|
||
|
||
const vimMoveRight = useCallback((count: number): void => {
|
||
dispatch({ type: 'vim_move_right', payload: { count } });
|
||
}, []);
|
||
|
||
const vimMoveUp = useCallback((count: number): void => {
|
||
dispatch({ type: 'vim_move_up', payload: { count } });
|
||
}, []);
|
||
|
||
const vimMoveDown = useCallback((count: number): void => {
|
||
dispatch({ type: 'vim_move_down', payload: { count } });
|
||
}, []);
|
||
|
||
const vimMoveWordForward = useCallback((count: number): void => {
|
||
dispatch({ type: 'vim_move_word_forward', payload: { count } });
|
||
}, []);
|
||
|
||
const vimMoveWordBackward = useCallback((count: number): void => {
|
||
dispatch({ type: 'vim_move_word_backward', payload: { count } });
|
||
}, []);
|
||
|
||
const vimMoveWordEnd = useCallback((count: number): void => {
|
||
dispatch({ type: 'vim_move_word_end', payload: { count } });
|
||
}, []);
|
||
|
||
const vimDeleteChar = useCallback((count: number): void => {
|
||
dispatch({ type: 'vim_delete_char', payload: { count } });
|
||
}, []);
|
||
|
||
const vimInsertAtCursor = useCallback((): void => {
|
||
dispatch({ type: 'vim_insert_at_cursor' });
|
||
}, []);
|
||
|
||
const vimAppendAtCursor = useCallback((): void => {
|
||
dispatch({ type: 'vim_append_at_cursor' });
|
||
}, []);
|
||
|
||
const vimOpenLineBelow = useCallback((): void => {
|
||
dispatch({ type: 'vim_open_line_below' });
|
||
}, []);
|
||
|
||
const vimOpenLineAbove = useCallback((): void => {
|
||
dispatch({ type: 'vim_open_line_above' });
|
||
}, []);
|
||
|
||
const vimAppendAtLineEnd = useCallback((): void => {
|
||
dispatch({ type: 'vim_append_at_line_end' });
|
||
}, []);
|
||
|
||
const vimInsertAtLineStart = useCallback((): void => {
|
||
dispatch({ type: 'vim_insert_at_line_start' });
|
||
}, []);
|
||
|
||
const vimMoveToLineStart = useCallback((): void => {
|
||
dispatch({ type: 'vim_move_to_line_start' });
|
||
}, []);
|
||
|
||
const vimMoveToLineEnd = useCallback((): void => {
|
||
dispatch({ type: 'vim_move_to_line_end' });
|
||
}, []);
|
||
|
||
const vimMoveToFirstNonWhitespace = useCallback((): void => {
|
||
dispatch({ type: 'vim_move_to_first_nonwhitespace' });
|
||
}, []);
|
||
|
||
const vimMoveToFirstLine = useCallback((): void => {
|
||
dispatch({ type: 'vim_move_to_first_line' });
|
||
}, []);
|
||
|
||
const vimMoveToLastLine = useCallback((): void => {
|
||
dispatch({ type: 'vim_move_to_last_line' });
|
||
}, []);
|
||
|
||
const vimMoveToLine = useCallback((lineNumber: number): void => {
|
||
dispatch({ type: 'vim_move_to_line', payload: { lineNumber } });
|
||
}, []);
|
||
|
||
const vimEscapeInsertMode = useCallback((): void => {
|
||
dispatch({ type: 'vim_escape_insert_mode' });
|
||
}, []);
|
||
|
||
const openInExternalEditor = useCallback(async (): Promise<void> => {
|
||
const tmpDir = fs.mkdtempSync(pathMod.join(os.tmpdir(), 'gemini-edit-'));
|
||
const filePath = pathMod.join(tmpDir, 'buffer.txt');
|
||
// Expand paste placeholders so user sees full content in editor
|
||
const expandedText = text.replace(
|
||
PASTED_TEXT_PLACEHOLDER_REGEX,
|
||
(match) => pastedContent[match] || match,
|
||
);
|
||
fs.writeFileSync(filePath, expandedText, 'utf8');
|
||
|
||
let command: string | undefined = undefined;
|
||
const args = [filePath];
|
||
|
||
const preferredEditorType = getPreferredEditor?.();
|
||
if (!command && preferredEditorType) {
|
||
command = getEditorCommand(preferredEditorType);
|
||
if (isGuiEditor(preferredEditorType)) {
|
||
args.unshift('--wait');
|
||
}
|
||
}
|
||
|
||
if (!command) {
|
||
command =
|
||
process.env['VISUAL'] ??
|
||
process.env['EDITOR'] ??
|
||
(process.platform === 'win32' ? 'notepad' : 'vi');
|
||
}
|
||
|
||
dispatch({ type: 'create_undo_snapshot' });
|
||
|
||
const wasRaw = stdin?.isRaw ?? false;
|
||
try {
|
||
setRawMode?.(false);
|
||
const { status, error } = spawnSync(command, args, {
|
||
stdio: 'inherit',
|
||
});
|
||
if (error) throw error;
|
||
if (typeof status === 'number' && status !== 0)
|
||
throw new Error(`External editor exited with status ${status}`);
|
||
|
||
let newText = fs.readFileSync(filePath, 'utf8');
|
||
newText = newText.replace(/\r\n?/g, '\n');
|
||
|
||
// Attempt to re-collapse unchanged pasted content back into placeholders
|
||
const sortedPlaceholders = Object.entries(pastedContent).sort(
|
||
(a, b) => b[1].length - a[1].length,
|
||
);
|
||
for (const [id, content] of sortedPlaceholders) {
|
||
if (newText.includes(content)) {
|
||
newText = newText.replace(content, id);
|
||
}
|
||
}
|
||
|
||
dispatch({ type: 'set_text', payload: newText, pushToUndo: false });
|
||
} catch (err) {
|
||
coreEvents.emitFeedback(
|
||
'error',
|
||
'[useTextBuffer] external editor error',
|
||
err,
|
||
);
|
||
} finally {
|
||
coreEvents.emit(CoreEvent.ExternalEditorClosed);
|
||
if (wasRaw) setRawMode?.(true);
|
||
try {
|
||
fs.unlinkSync(filePath);
|
||
} catch {
|
||
/* ignore */
|
||
}
|
||
try {
|
||
fs.rmdirSync(tmpDir);
|
||
} catch {
|
||
/* ignore */
|
||
}
|
||
}
|
||
}, [text, pastedContent, stdin, setRawMode, getPreferredEditor]);
|
||
|
||
const handleInput = useCallback(
|
||
(key: Key): boolean => {
|
||
const { sequence: input } = key;
|
||
|
||
if (key.name === 'paste') {
|
||
insert(input, { paste: true });
|
||
return true;
|
||
}
|
||
if (keyMatchers[Command.RETURN](key)) {
|
||
if (singleLine) {
|
||
return false;
|
||
}
|
||
newline();
|
||
return true;
|
||
}
|
||
if (keyMatchers[Command.NEWLINE](key)) {
|
||
if (singleLine) {
|
||
return false;
|
||
}
|
||
newline();
|
||
return true;
|
||
}
|
||
if (keyMatchers[Command.MOVE_LEFT](key)) {
|
||
if (cursorRow === 0 && cursorCol === 0) return false;
|
||
move('left');
|
||
return true;
|
||
}
|
||
if (keyMatchers[Command.MOVE_RIGHT](key)) {
|
||
const lastLineIdx = lines.length - 1;
|
||
if (
|
||
cursorRow === lastLineIdx &&
|
||
cursorCol === cpLen(lines[lastLineIdx] ?? '')
|
||
) {
|
||
return false;
|
||
}
|
||
move('right');
|
||
return true;
|
||
}
|
||
if (keyMatchers[Command.MOVE_UP](key)) {
|
||
if (visualCursor[0] === 0) return false;
|
||
move('up');
|
||
return true;
|
||
}
|
||
if (keyMatchers[Command.MOVE_DOWN](key)) {
|
||
if (visualCursor[0] === visualLines.length - 1) return false;
|
||
move('down');
|
||
return true;
|
||
}
|
||
if (keyMatchers[Command.MOVE_WORD_LEFT](key)) {
|
||
move('wordLeft');
|
||
return true;
|
||
}
|
||
if (keyMatchers[Command.MOVE_WORD_RIGHT](key)) {
|
||
move('wordRight');
|
||
return true;
|
||
}
|
||
if (keyMatchers[Command.HOME](key)) {
|
||
move('home');
|
||
return true;
|
||
}
|
||
if (keyMatchers[Command.END](key)) {
|
||
move('end');
|
||
return true;
|
||
}
|
||
if (keyMatchers[Command.DELETE_WORD_BACKWARD](key)) {
|
||
deleteWordLeft();
|
||
return true;
|
||
}
|
||
if (keyMatchers[Command.DELETE_WORD_FORWARD](key)) {
|
||
deleteWordRight();
|
||
return true;
|
||
}
|
||
if (keyMatchers[Command.DELETE_CHAR_LEFT](key)) {
|
||
backspace();
|
||
return true;
|
||
}
|
||
if (keyMatchers[Command.DELETE_CHAR_RIGHT](key)) {
|
||
del();
|
||
return true;
|
||
}
|
||
if (keyMatchers[Command.UNDO](key)) {
|
||
undo();
|
||
return true;
|
||
}
|
||
if (keyMatchers[Command.REDO](key)) {
|
||
redo();
|
||
return true;
|
||
}
|
||
if (key.insertable) {
|
||
insert(input, { paste: false });
|
||
return true;
|
||
}
|
||
return false;
|
||
},
|
||
[
|
||
newline,
|
||
move,
|
||
deleteWordLeft,
|
||
deleteWordRight,
|
||
backspace,
|
||
del,
|
||
insert,
|
||
undo,
|
||
redo,
|
||
cursorRow,
|
||
cursorCol,
|
||
lines,
|
||
singleLine,
|
||
visualCursor,
|
||
visualLines,
|
||
],
|
||
);
|
||
|
||
const visualScrollRow = useMemo(() => {
|
||
const totalVisualLines = visualLines.length;
|
||
return Math.min(
|
||
scrollRowState,
|
||
Math.max(0, totalVisualLines - viewport.height),
|
||
);
|
||
}, [visualLines.length, scrollRowState, viewport.height]);
|
||
|
||
const renderedVisualLines = useMemo(
|
||
() => visualLines.slice(visualScrollRow, visualScrollRow + viewport.height),
|
||
[visualLines, visualScrollRow, viewport.height],
|
||
);
|
||
|
||
const replaceRange = useCallback(
|
||
(
|
||
startRow: number,
|
||
startCol: number,
|
||
endRow: number,
|
||
endCol: number,
|
||
text: string,
|
||
): void => {
|
||
dispatch({
|
||
type: 'replace_range',
|
||
payload: { startRow, startCol, endRow, endCol, text },
|
||
});
|
||
},
|
||
[],
|
||
);
|
||
|
||
const replaceRangeByOffset = useCallback(
|
||
(startOffset: number, endOffset: number, replacementText: string): void => {
|
||
const [startRow, startCol] = offsetToLogicalPos(text, startOffset);
|
||
const [endRow, endCol] = offsetToLogicalPos(text, endOffset);
|
||
replaceRange(startRow, startCol, endRow, endCol, replacementText);
|
||
},
|
||
[text, replaceRange],
|
||
);
|
||
|
||
const moveToOffset = useCallback((offset: number): void => {
|
||
dispatch({ type: 'move_to_offset', payload: { offset } });
|
||
}, []);
|
||
|
||
const moveToVisualPosition = useCallback(
|
||
(visRow: number, visCol: number): void => {
|
||
const {
|
||
visualLines,
|
||
visualToLogicalMap,
|
||
transformedToLogicalMaps,
|
||
visualToTransformedMap,
|
||
} = visualLayout;
|
||
// Clamp visRow to valid range
|
||
const clampedVisRow = Math.max(
|
||
0,
|
||
Math.min(visRow, visualLines.length - 1),
|
||
);
|
||
const visualLine = visualLines[clampedVisRow] || '';
|
||
|
||
if (visualToLogicalMap[clampedVisRow]) {
|
||
const [logRow] = visualToLogicalMap[clampedVisRow];
|
||
const transformedToLogicalMap =
|
||
transformedToLogicalMaps?.[logRow] ?? [];
|
||
|
||
// Where does this visual line begin within the transformed line?
|
||
const startColInTransformed =
|
||
visualToTransformedMap?.[clampedVisRow] ?? 0;
|
||
|
||
// Handle wide characters: convert visual X position to character offset
|
||
const codePoints = toCodePoints(visualLine);
|
||
let currentVisX = 0;
|
||
let charOffset = 0;
|
||
|
||
for (const char of codePoints) {
|
||
const charWidth = getCachedStringWidth(char);
|
||
// If the click is within this character
|
||
if (visCol < currentVisX + charWidth) {
|
||
// Check if we clicked the second half of a wide character
|
||
if (charWidth > 1 && visCol >= currentVisX + charWidth / 2) {
|
||
charOffset++;
|
||
}
|
||
break;
|
||
}
|
||
currentVisX += charWidth;
|
||
charOffset++;
|
||
}
|
||
|
||
// Clamp charOffset to length
|
||
charOffset = Math.min(charOffset, codePoints.length);
|
||
|
||
// Map character offset through transformations to get logical position
|
||
const transformedCol = Math.min(
|
||
startColInTransformed + charOffset,
|
||
Math.max(0, transformedToLogicalMap.length - 1),
|
||
);
|
||
|
||
const newCursorRow = logRow;
|
||
const newCursorCol =
|
||
transformedToLogicalMap[transformedCol] ?? cpLen(lines[logRow] ?? '');
|
||
|
||
dispatch({
|
||
type: 'set_cursor',
|
||
payload: {
|
||
cursorRow: newCursorRow,
|
||
cursorCol: newCursorCol,
|
||
preferredCol: charOffset,
|
||
},
|
||
});
|
||
}
|
||
},
|
||
[visualLayout, lines],
|
||
);
|
||
|
||
const getLogicalPositionFromVisual = useCallback(
|
||
(visRow: number, visCol: number): { row: number; col: number } | null => {
|
||
const {
|
||
visualLines,
|
||
visualToLogicalMap,
|
||
transformedToLogicalMaps,
|
||
visualToTransformedMap,
|
||
} = visualLayout;
|
||
|
||
// Clamp visRow to valid range
|
||
const clampedVisRow = Math.max(
|
||
0,
|
||
Math.min(visRow, visualLines.length - 1),
|
||
);
|
||
const visualLine = visualLines[clampedVisRow] || '';
|
||
|
||
if (!visualToLogicalMap[clampedVisRow]) {
|
||
return null;
|
||
}
|
||
|
||
const [logRow] = visualToLogicalMap[clampedVisRow];
|
||
const transformedToLogicalMap = transformedToLogicalMaps?.[logRow] ?? [];
|
||
|
||
// Where does this visual line begin within the transformed line?
|
||
const startColInTransformed =
|
||
visualToTransformedMap?.[clampedVisRow] ?? 0;
|
||
|
||
// Handle wide characters: convert visual X position to character offset
|
||
const codePoints = toCodePoints(visualLine);
|
||
let currentVisX = 0;
|
||
let charOffset = 0;
|
||
|
||
for (const char of codePoints) {
|
||
const charWidth = getCachedStringWidth(char);
|
||
if (visCol < currentVisX + charWidth) {
|
||
if (charWidth > 1 && visCol >= currentVisX + charWidth / 2) {
|
||
charOffset++;
|
||
}
|
||
break;
|
||
}
|
||
currentVisX += charWidth;
|
||
charOffset++;
|
||
}
|
||
|
||
charOffset = Math.min(charOffset, codePoints.length);
|
||
|
||
const transformedCol = Math.min(
|
||
startColInTransformed + charOffset,
|
||
Math.max(0, transformedToLogicalMap.length - 1),
|
||
);
|
||
|
||
const row = logRow;
|
||
const col =
|
||
transformedToLogicalMap[transformedCol] ?? cpLen(lines[logRow] ?? '');
|
||
|
||
return { row, col };
|
||
},
|
||
[visualLayout, lines],
|
||
);
|
||
|
||
const getOffset = useCallback(
|
||
(): number => logicalPosToOffset(lines, cursorRow, cursorCol),
|
||
[lines, cursorRow, cursorCol],
|
||
);
|
||
|
||
const togglePasteExpansion = useCallback(
|
||
(id: string, row: number, col: number): void => {
|
||
dispatch({ type: 'toggle_paste_expansion', payload: { id, row, col } });
|
||
},
|
||
[],
|
||
);
|
||
|
||
const getExpandedPasteAtLineCallback = useCallback(
|
||
(lineIndex: number): string | null =>
|
||
getExpandedPasteAtLine(lineIndex, expandedPaste),
|
||
[expandedPaste],
|
||
);
|
||
|
||
const returnValue: TextBuffer = useMemo(
|
||
() => ({
|
||
lines,
|
||
text,
|
||
cursor: [cursorRow, cursorCol],
|
||
preferredCol,
|
||
selectionAnchor,
|
||
pastedContent,
|
||
|
||
allVisualLines: visualLines,
|
||
viewportVisualLines: renderedVisualLines,
|
||
visualCursor,
|
||
visualScrollRow,
|
||
visualToLogicalMap,
|
||
transformedToLogicalMaps,
|
||
visualToTransformedMap,
|
||
transformationsByLine,
|
||
visualLayout,
|
||
setText,
|
||
insert,
|
||
newline,
|
||
backspace,
|
||
del,
|
||
move,
|
||
undo,
|
||
redo,
|
||
replaceRange,
|
||
replaceRangeByOffset,
|
||
moveToOffset,
|
||
getOffset,
|
||
moveToVisualPosition,
|
||
getLogicalPositionFromVisual,
|
||
getExpandedPasteAtLine: getExpandedPasteAtLineCallback,
|
||
togglePasteExpansion,
|
||
expandedPaste,
|
||
deleteWordLeft,
|
||
deleteWordRight,
|
||
|
||
killLineRight,
|
||
killLineLeft,
|
||
handleInput,
|
||
openInExternalEditor,
|
||
// Vim-specific operations
|
||
vimDeleteWordForward,
|
||
vimDeleteWordBackward,
|
||
vimDeleteWordEnd,
|
||
vimChangeWordForward,
|
||
vimChangeWordBackward,
|
||
vimChangeWordEnd,
|
||
vimDeleteLine,
|
||
vimChangeLine,
|
||
vimDeleteToEndOfLine,
|
||
vimChangeToEndOfLine,
|
||
vimChangeMovement,
|
||
vimMoveLeft,
|
||
vimMoveRight,
|
||
vimMoveUp,
|
||
vimMoveDown,
|
||
vimMoveWordForward,
|
||
vimMoveWordBackward,
|
||
vimMoveWordEnd,
|
||
vimDeleteChar,
|
||
vimInsertAtCursor,
|
||
vimAppendAtCursor,
|
||
vimOpenLineBelow,
|
||
vimOpenLineAbove,
|
||
vimAppendAtLineEnd,
|
||
vimInsertAtLineStart,
|
||
vimMoveToLineStart,
|
||
vimMoveToLineEnd,
|
||
vimMoveToFirstNonWhitespace,
|
||
vimMoveToFirstLine,
|
||
vimMoveToLastLine,
|
||
vimMoveToLine,
|
||
vimEscapeInsertMode,
|
||
}),
|
||
[
|
||
lines,
|
||
text,
|
||
cursorRow,
|
||
cursorCol,
|
||
preferredCol,
|
||
selectionAnchor,
|
||
pastedContent,
|
||
visualLines,
|
||
renderedVisualLines,
|
||
visualCursor,
|
||
visualScrollRow,
|
||
visualToLogicalMap,
|
||
transformedToLogicalMaps,
|
||
visualToTransformedMap,
|
||
transformationsByLine,
|
||
visualLayout,
|
||
setText,
|
||
insert,
|
||
newline,
|
||
backspace,
|
||
del,
|
||
move,
|
||
undo,
|
||
redo,
|
||
replaceRange,
|
||
replaceRangeByOffset,
|
||
moveToOffset,
|
||
getOffset,
|
||
moveToVisualPosition,
|
||
getLogicalPositionFromVisual,
|
||
getExpandedPasteAtLineCallback,
|
||
togglePasteExpansion,
|
||
expandedPaste,
|
||
deleteWordLeft,
|
||
deleteWordRight,
|
||
killLineRight,
|
||
killLineLeft,
|
||
handleInput,
|
||
openInExternalEditor,
|
||
vimDeleteWordForward,
|
||
vimDeleteWordBackward,
|
||
vimDeleteWordEnd,
|
||
vimChangeWordForward,
|
||
vimChangeWordBackward,
|
||
vimChangeWordEnd,
|
||
vimDeleteLine,
|
||
vimChangeLine,
|
||
vimDeleteToEndOfLine,
|
||
vimChangeToEndOfLine,
|
||
vimChangeMovement,
|
||
vimMoveLeft,
|
||
vimMoveRight,
|
||
vimMoveUp,
|
||
vimMoveDown,
|
||
vimMoveWordForward,
|
||
vimMoveWordBackward,
|
||
vimMoveWordEnd,
|
||
vimDeleteChar,
|
||
vimInsertAtCursor,
|
||
vimAppendAtCursor,
|
||
vimOpenLineBelow,
|
||
vimOpenLineAbove,
|
||
vimAppendAtLineEnd,
|
||
vimInsertAtLineStart,
|
||
vimMoveToLineStart,
|
||
vimMoveToLineEnd,
|
||
vimMoveToFirstNonWhitespace,
|
||
vimMoveToFirstLine,
|
||
vimMoveToLastLine,
|
||
vimMoveToLine,
|
||
vimEscapeInsertMode,
|
||
],
|
||
);
|
||
return returnValue;
|
||
}
|
||
|
||
export interface TextBuffer {
|
||
// State
|
||
lines: string[]; // Logical lines
|
||
text: string;
|
||
cursor: [number, number]; // Logical cursor [row, col]
|
||
/**
|
||
* When the user moves the caret vertically we try to keep their original
|
||
* horizontal column even when passing through shorter lines. We remember
|
||
* that *preferred* column in this field while the user is still travelling
|
||
* vertically. Any explicit horizontal movement resets the preference.
|
||
*/
|
||
preferredCol: number | null; // Preferred visual column
|
||
selectionAnchor: [number, number] | null; // Logical selection anchor
|
||
pastedContent: Record<string, string>;
|
||
|
||
// Visual state (handles wrapping)
|
||
allVisualLines: string[]; // All visual lines for the current text and viewport width.
|
||
viewportVisualLines: string[]; // The subset of visual lines to be rendered based on visualScrollRow and viewport.height
|
||
visualCursor: [number, number]; // Visual cursor [row, col] relative to the start of all visualLines
|
||
visualScrollRow: number; // Scroll position for visual lines (index of the first visible visual line)
|
||
/**
|
||
* For each visual line (by absolute index in allVisualLines) provides a tuple
|
||
* [logicalLineIndex, startColInLogical] that maps where that visual line
|
||
* begins within the logical buffer. Indices are code-point based.
|
||
*/
|
||
visualToLogicalMap: Array<[number, number]>;
|
||
/**
|
||
* For each logical line, an array mapping transformed positions (in the transformed
|
||
* line) back to logical column indices.
|
||
*/
|
||
transformedToLogicalMaps: number[][];
|
||
/**
|
||
* For each visual line (absolute index across all visual lines), the start index
|
||
* within that logical line's transformed content.
|
||
*/
|
||
visualToTransformedMap: number[];
|
||
/** Cached transformations per logical line */
|
||
transformationsByLine: Transformation[][];
|
||
visualLayout: VisualLayout;
|
||
|
||
// Actions
|
||
|
||
/**
|
||
* Replaces the entire buffer content with the provided text.
|
||
* The operation is undoable.
|
||
*/
|
||
setText: (text: string) => void;
|
||
/**
|
||
* Insert a single character or string without newlines.
|
||
*/
|
||
insert: (ch: string, opts?: { paste?: boolean }) => void;
|
||
newline: () => void;
|
||
backspace: () => void;
|
||
del: () => void;
|
||
move: (dir: Direction) => void;
|
||
undo: () => void;
|
||
redo: () => void;
|
||
/**
|
||
* Replaces the text within the specified range with new text.
|
||
* Handles both single-line and multi-line ranges.
|
||
*
|
||
* @param startRow The starting row index (inclusive).
|
||
* @param startCol The starting column index (inclusive, code-point based).
|
||
* @param endRow The ending row index (inclusive).
|
||
* @param endCol The ending column index (exclusive, code-point based).
|
||
* @param text The new text to insert.
|
||
* @returns True if the buffer was modified, false otherwise.
|
||
*/
|
||
replaceRange: (
|
||
startRow: number,
|
||
startCol: number,
|
||
endRow: number,
|
||
endCol: number,
|
||
text: string,
|
||
) => void;
|
||
/**
|
||
* Delete the word to the *left* of the caret, mirroring common
|
||
* Ctrl/Alt+Backspace behaviour in editors & terminals. Both the adjacent
|
||
* whitespace *and* the word characters immediately preceding the caret are
|
||
* removed. If the caret is already at column‑0 this becomes a no-op.
|
||
*/
|
||
deleteWordLeft: () => void;
|
||
/**
|
||
* Delete the word to the *right* of the caret, akin to many editors'
|
||
* Ctrl/Alt+Delete shortcut. Removes any whitespace/punctuation that
|
||
* follows the caret and the next contiguous run of word characters.
|
||
*/
|
||
deleteWordRight: () => void;
|
||
|
||
/**
|
||
* Deletes text from the cursor to the end of the current line.
|
||
*/
|
||
killLineRight: () => void;
|
||
/**
|
||
* Deletes text from the start of the current line to the cursor.
|
||
*/
|
||
killLineLeft: () => void;
|
||
/**
|
||
* High level "handleInput" – receives what Ink gives us.
|
||
*/
|
||
handleInput: (key: Key) => void;
|
||
/**
|
||
* Opens the current buffer contents in the user's preferred terminal text
|
||
* editor ($VISUAL or $EDITOR, falling back to "vi"). The method blocks
|
||
* until the editor exits, then reloads the file and replaces the in‑memory
|
||
* buffer with whatever the user saved.
|
||
*
|
||
* The operation is treated as a single undoable edit – we snapshot the
|
||
* previous state *once* before launching the editor so one `undo()` will
|
||
* revert the entire change set.
|
||
*
|
||
* Note: We purposefully rely on the *synchronous* spawn API so that the
|
||
* calling process genuinely waits for the editor to close before
|
||
* continuing. This mirrors Git's behaviour and simplifies downstream
|
||
* control‑flow (callers can simply `await` the Promise).
|
||
*/
|
||
openInExternalEditor: () => Promise<void>;
|
||
|
||
replaceRangeByOffset: (
|
||
startOffset: number,
|
||
endOffset: number,
|
||
replacementText: string,
|
||
) => void;
|
||
getOffset: () => number;
|
||
moveToOffset(offset: number): void;
|
||
moveToVisualPosition(visualRow: number, visualCol: number): void;
|
||
/**
|
||
* Convert visual coordinates to logical position without moving cursor.
|
||
* Returns null if the position is out of bounds.
|
||
*/
|
||
getLogicalPositionFromVisual(
|
||
visualRow: number,
|
||
visualCol: number,
|
||
): { row: number; col: number } | null;
|
||
/**
|
||
* Check if a line index falls within an expanded paste region.
|
||
* Returns the paste placeholder ID if found, null otherwise.
|
||
*/
|
||
getExpandedPasteAtLine(lineIndex: number): string | null;
|
||
/**
|
||
* Toggle expansion state for a paste placeholder.
|
||
* If collapsed, expands to show full content inline.
|
||
* If expanded, collapses back to placeholder.
|
||
*/
|
||
togglePasteExpansion(id: string, row: number, col: number): void;
|
||
/**
|
||
* The current expanded paste info (read-only).
|
||
*/
|
||
expandedPaste: ExpandedPasteInfo | null;
|
||
|
||
// Vim-specific operations
|
||
/**
|
||
* Delete N words forward from cursor position (vim 'dw' command)
|
||
*/
|
||
vimDeleteWordForward: (count: number) => void;
|
||
/**
|
||
* Delete N words backward from cursor position (vim 'db' command)
|
||
*/
|
||
vimDeleteWordBackward: (count: number) => void;
|
||
/**
|
||
* Delete to end of N words from cursor position (vim 'de' command)
|
||
*/
|
||
vimDeleteWordEnd: (count: number) => void;
|
||
/**
|
||
* Change N words forward from cursor position (vim 'cw' command)
|
||
*/
|
||
vimChangeWordForward: (count: number) => void;
|
||
/**
|
||
* Change N words backward from cursor position (vim 'cb' command)
|
||
*/
|
||
vimChangeWordBackward: (count: number) => void;
|
||
/**
|
||
* Change to end of N words from cursor position (vim 'ce' command)
|
||
*/
|
||
vimChangeWordEnd: (count: number) => void;
|
||
/**
|
||
* Delete N lines from cursor position (vim 'dd' command)
|
||
*/
|
||
vimDeleteLine: (count: number) => void;
|
||
/**
|
||
* Change N lines from cursor position (vim 'cc' command)
|
||
*/
|
||
vimChangeLine: (count: number) => void;
|
||
/**
|
||
* Delete from cursor to end of line (vim 'D' command)
|
||
*/
|
||
vimDeleteToEndOfLine: () => void;
|
||
/**
|
||
* Change from cursor to end of line (vim 'C' command)
|
||
*/
|
||
vimChangeToEndOfLine: () => void;
|
||
/**
|
||
* Change movement operations (vim 'ch', 'cj', 'ck', 'cl' commands)
|
||
*/
|
||
vimChangeMovement: (movement: 'h' | 'j' | 'k' | 'l', count: number) => void;
|
||
/**
|
||
* Move cursor left N times (vim 'h' command)
|
||
*/
|
||
vimMoveLeft: (count: number) => void;
|
||
/**
|
||
* Move cursor right N times (vim 'l' command)
|
||
*/
|
||
vimMoveRight: (count: number) => void;
|
||
/**
|
||
* Move cursor up N times (vim 'k' command)
|
||
*/
|
||
vimMoveUp: (count: number) => void;
|
||
/**
|
||
* Move cursor down N times (vim 'j' command)
|
||
*/
|
||
vimMoveDown: (count: number) => void;
|
||
/**
|
||
* Move cursor forward N words (vim 'w' command)
|
||
*/
|
||
vimMoveWordForward: (count: number) => void;
|
||
/**
|
||
* Move cursor backward N words (vim 'b' command)
|
||
*/
|
||
vimMoveWordBackward: (count: number) => void;
|
||
/**
|
||
* Move cursor to end of Nth word (vim 'e' command)
|
||
*/
|
||
vimMoveWordEnd: (count: number) => void;
|
||
/**
|
||
* Delete N characters at cursor (vim 'x' command)
|
||
*/
|
||
vimDeleteChar: (count: number) => void;
|
||
/**
|
||
* Enter insert mode at cursor (vim 'i' command)
|
||
*/
|
||
vimInsertAtCursor: () => void;
|
||
/**
|
||
* Enter insert mode after cursor (vim 'a' command)
|
||
*/
|
||
vimAppendAtCursor: () => void;
|
||
/**
|
||
* Open new line below and enter insert mode (vim 'o' command)
|
||
*/
|
||
vimOpenLineBelow: () => void;
|
||
/**
|
||
* Open new line above and enter insert mode (vim 'O' command)
|
||
*/
|
||
vimOpenLineAbove: () => void;
|
||
/**
|
||
* Move to end of line and enter insert mode (vim 'A' command)
|
||
*/
|
||
vimAppendAtLineEnd: () => void;
|
||
/**
|
||
* Move to first non-whitespace and enter insert mode (vim 'I' command)
|
||
*/
|
||
vimInsertAtLineStart: () => void;
|
||
/**
|
||
* Move cursor to beginning of line (vim '0' command)
|
||
*/
|
||
vimMoveToLineStart: () => void;
|
||
/**
|
||
* Move cursor to end of line (vim '$' command)
|
||
*/
|
||
vimMoveToLineEnd: () => void;
|
||
/**
|
||
* Move cursor to first non-whitespace character (vim '^' command)
|
||
*/
|
||
vimMoveToFirstNonWhitespace: () => void;
|
||
/**
|
||
* Move cursor to first line (vim 'gg' command)
|
||
*/
|
||
vimMoveToFirstLine: () => void;
|
||
/**
|
||
* Move cursor to last line (vim 'G' command)
|
||
*/
|
||
vimMoveToLastLine: () => void;
|
||
/**
|
||
* Move cursor to specific line number (vim '[N]G' command)
|
||
*/
|
||
vimMoveToLine: (lineNumber: number) => void;
|
||
/**
|
||
* Handle escape from insert mode (moves cursor left if not at line start)
|
||
*/
|
||
vimEscapeInsertMode: () => void;
|
||
}
|