fix: text buffer handling

Author: tomasz-stefaniak

extensions/cli/src/ui/EditMessageSelector.test.tsx

@@ -7,19 +7,7 @@ import React from "react";
 import { describe, it, expect, vi, beforeEach } from "vitest";
 
 import { EditMessageSelector } from "./EditMessageSelector.js";
-
-// Mock TextBuffer for controlled testing
-vi.mock("./TextBuffer.js", () => {
-  return {
-    TextBuffer: vi.fn().mockImplementation(() => ({
-      text: "",
-      cursor: 0,
-      setText: vi.fn(),
-      setCursor: vi.fn(),
-      handleInput: vi.fn(),
-    })),
-  };
-});
+import { TextBuffer } from "./TextBuffer.js";
 
 describe("EditMessageSelector", () => {
   const createMockChatHistory = (count: number): ChatHistoryItem[] => {

extensions/cli/src/ui/EditMessageSelector.tsx

@@ -36,6 +36,22 @@ export function EditMessageSelector({
   const [editText, setEditText] = useState("");
   const [cursorPosition, setCursorPosition] = useState(0);
 
+  // Stable callback for TextBuffer state changes (e.g., paste finalization)
+  const onStateChange = React.useCallback(() => {
+    setEditText(textBuffer.text);
+    setCursorPosition(textBuffer.cursor);
+  }, [textBuffer]);
+
+  // Set up callback for when TextBuffer state changes asynchronously
+  React.useEffect(() => {
+    textBuffer.setStateChangeCallback(onStateChange);
+
+    return () => {
+      // Clear any pending timers on unmount
+      textBuffer.clear();
+    };
+  }, [textBuffer, onStateChange]);
+
   // Initialize edit text when selection changes
   React.useEffect(() => {
     if (!isEditing && userMessages[selectedIndex]) {