Fix bugs for ctrl+U

Author: rugvedS07

src/subcommands/chat/react/inputReducer.test.ts

@@ -1966,6 +1966,47 @@ describe("chatInputStateReducers", () => {
       expect(result.cursorOnSegmentIndex).toBe(0);
       expect(result.cursorInSegmentOffset).toBe(0);
     });
+
+    it("preserves text after cursor when line starts with largePaste", () => {
+      const initialState: ChatUserInputState = {
+        segments: [
+          { type: "largePaste", content: "x".repeat(100) },
+          { type: "text", content: "should deleteshould keep" },
+        ],
+        cursorOnSegmentIndex: 1,
+        cursorInSegmentOffset: 13,
+      };
+
+      const result = deleteToLineStart(initialState);
+
+      expect(result.segments.length).toBe(1);
+      expect(result.segments[0].type).toBe("text");
+      expect(result.segments[0].content).toBe("should keep");
+      expect(result.cursorOnSegmentIndex).toBe(0);
+      expect(result.cursorInSegmentOffset).toBe(0);
+    });
+
+    it("deletes largePaste when cursor is at paste start with text on both sides", () => {
+      const initialState: ChatUserInputState = {
+        segments: [
+          { type: "text", content: "before" },
+          { type: "largePaste", content: "x".repeat(100) },
+          { type: "text", content: "after" },
+        ],
+        cursorOnSegmentIndex: 1,
+        cursorInSegmentOffset: 0,
+      };
+
+      const result = deleteToLineStart(initialState);
+
+      expect(result.segments.length).toBe(2);
+      expect(result.segments[0].type).toBe("largePaste");
+      expect(result.segments[0].content).toBe("x".repeat(100));
+      expect(result.segments[1].type).toBe("text");
+      expect(result.segments[1].content).toBe("after");
+      expect(result.cursorOnSegmentIndex).toBe(0);
+      expect(result.cursorInSegmentOffset).toBe(0);
+    });
   });
 
   describe("deleteToLineEnd", () => {
@@ -2068,7 +2109,25 @@ describe("chatInputStateReducers", () => {
       expect(result.cursorOnSegmentIndex).toBe(0);
       expect(result.cursorInSegmentOffset).toBe(9);
     });
+    it("deletes trailing largePaste when cursor is in text before it", () => {
+      const initialState: ChatUserInputState = {
+        segments: [
+          { type: "text", content: "keep this" },
+          { type: "largePaste", content: "x".repeat(100) },
+          { type: "text", content: "" },
+        ],
+        cursorOnSegmentIndex: 0,
+        cursorInSegmentOffset: 9,
+      };
+
+      const result = deleteToLineEnd(initialState);
 
+      expect(result.segments.length).toBe(1);
+      expect(result.segments[0].type).toBe("text");
+      expect(result.segments[0].content).toBe("keep this");
+      expect(result.cursorOnSegmentIndex).toBe(0);
+      expect(result.cursorInSegmentOffset).toBe(9);
+    });
     it("deletes to newline and preserves text after newline", () => {
       const initialState: ChatUserInputState = {
         segments: [{ type: "text", content: "line one\nline two" }],

src/subcommands/chat/react/inputReducer.ts

@@ -1041,18 +1041,44 @@ export function deleteToLineStart(state: ChatUserInputState): ChatUserInputState
     const lineStartSegment = draft.segments[lineStartPosition.segmentIndex];
     const currentSegment = draft.segments[currentSegmentIndex];
 
-    if (lineStartSegment !== undefined && lineStartSegment.type === "text") {
-      // Keep content before line start
-      lineStartSegment.content = lineStartSegment.content.slice(0, lineStartPosition.offset);
-    }
+    // If cursor is at start of current segment (offset 0), preserve current segment
+    if (currentOffset === 0) {
+      // Delete segments between line start and current (exclusive of current)
+      const segmentsToRemove = currentSegmentIndex - lineStartPosition.segmentIndex - 1;
+      if (segmentsToRemove > 0) {
+        draft.segments.splice(lineStartPosition.segmentIndex + 1, segmentsToRemove);
+        draft.cursorOnSegmentIndex = lineStartPosition.segmentIndex + 1;
+      }
 
-    if (currentSegment !== undefined && currentSegment.type === "text") {
-      // Keep content after cursor
-      const contentAfterCursor = currentSegment.content.slice(currentOffset);
-      // Merge into line start segment if it's text
+      // Truncate line start segment at line start position
       if (lineStartSegment !== undefined && lineStartSegment.type === "text") {
-        lineStartSegment.content += contentAfterCursor;
+        lineStartSegment.content = lineStartSegment.content.slice(0, lineStartPosition.offset);
+        if (lineStartSegment.content.length === 0) {
+          // Remove empty line start segment
+          draft.segments.splice(lineStartPosition.segmentIndex, 1);
+          draft.cursorOnSegmentIndex = lineStartPosition.segmentIndex;
+        }
       }
+
+      draft.cursorInSegmentOffset = 0;
+      return;
+    }
+
+    let contentAfterCursor = "";
+    if (currentSegment !== undefined && currentSegment.type === "text") {
+      contentAfterCursor = currentSegment.content.slice(currentOffset);
+    }
+
+    if (lineStartSegment !== undefined && lineStartSegment.type === "text") {
+      // Keep content before line start and merge content after cursor
+      lineStartSegment.content =
+        lineStartSegment.content.slice(0, lineStartPosition.offset) + contentAfterCursor;
+    } else if (contentAfterCursor.length > 0) {
+      // Line starts with non-text segment, create new text segment for content after cursor
+      draft.segments[lineStartPosition.segmentIndex] = {
+        type: "text",
+        content: contentAfterCursor,
+      };
     }
 
     // Remove segments between line start and cursor (inclusive of current if different)
@@ -1063,7 +1089,10 @@ export function deleteToLineStart(state: ChatUserInputState): ChatUserInputState
 
     // Position cursor at line start
     draft.cursorOnSegmentIndex = lineStartPosition.segmentIndex;
-    draft.cursorInSegmentOffset = lineStartPosition.offset;
+    draft.cursorInSegmentOffset =
+      lineStartSegment !== undefined && lineStartSegment.type === "text"
+        ? lineStartPosition.offset
+        : 0;
   });
 }