Fix more bugs

Author: rugvedS07

src/subcommands/chat/react/ChatInput.tsx

@@ -188,14 +188,16 @@ export const ChatInput = ({
         return;
       }
 
-      // For linux, we need to specifically check for Alt+Arrows so we have these here
-      if (key.leftArrow === true) {
-        setUserInputState(previousState => moveCursorWordLeft(previousState));
-        return;
-      }
-      if (key.rightArrow === true) {
-        setUserInputState(previousState => moveCursorWordRight(previousState));
-        return;
+      if (isMac === false) {
+        // For linux, we need to specifically check for Alt+Arrows so we have these here
+        if (key.leftArrow === true) {
+          setUserInputState(previousState => moveCursorWordLeft(previousState));
+          return;
+        }
+        if (key.rightArrow === true) {
+          setUserInputState(previousState => moveCursorWordRight(previousState));
+          return;
+        }
       }
       // When we press option+fn+delete on macOS, it sends meta+d
       // and for linux, alt+d is the alternative to alt+delete for word delete forward

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

@@ -2007,6 +2007,25 @@ describe("chatInputStateReducers", () => {
       expect(result.cursorOnSegmentIndex).toBe(0);
       expect(result.cursorInSegmentOffset).toBe(0);
     });
+
+    it("deletes largePaste when cursor is at trailing placeholder after paste", () => {
+      const initialState: ChatUserInputState = {
+        segments: [
+          { type: "largePaste", content: "x".repeat(100) },
+          { type: "text", content: "" },
+        ],
+        cursorOnSegmentIndex: 1,
+        cursorInSegmentOffset: 0,
+      };
+
+      const result = deleteToLineStart(initialState);
+
+      expect(result.segments.length).toBe(1);
+      expect(result.segments[0].type).toBe("text");
+      expect(result.segments[0].content).toBe("");
+      expect(result.cursorOnSegmentIndex).toBe(0);
+      expect(result.cursorInSegmentOffset).toBe(0);
+    });
   });
 
   describe("deleteToLineEnd", () => {
@@ -2142,5 +2161,24 @@ describe("chatInputStateReducers", () => {
       expect(result.cursorOnSegmentIndex).toBe(0);
       expect(result.cursorInSegmentOffset).toBe(5);
     });
+
+    it("deletes largePaste and following text when cursor is at offset 0 of largePaste", () => {
+      const initialState: ChatUserInputState = {
+        segments: [
+          { type: "largePaste", content: "x".repeat(100) },
+          { type: "text", content: "following text" },
+        ],
+        cursorOnSegmentIndex: 0,
+        cursorInSegmentOffset: 0,
+      };
+
+      const result = deleteToLineEnd(initialState);
+
+      expect(result.segments.length).toBe(1);
+      expect(result.segments[0].type).toBe("text");
+      expect(result.segments[0].content).toBe("");
+      expect(result.cursorOnSegmentIndex).toBe(0);
+      expect(result.cursorInSegmentOffset).toBe(0);
+    });
   });
 });

src/subcommands/chat/react/inputReducer.ts

@@ -1058,6 +1058,10 @@ export function deleteToLineStart(state: ChatUserInputState): ChatUserInputState
           draft.segments.splice(lineStartPosition.segmentIndex, 1);
           draft.cursorOnSegmentIndex = lineStartPosition.segmentIndex;
         }
+      } else if (lineStartSegment !== undefined) {
+        // Line start segment is non-text (e.g., largePaste) - remove it
+        draft.segments.splice(lineStartPosition.segmentIndex, 1);
+        draft.cursorOnSegmentIndex = lineStartPosition.segmentIndex;
       }
 
       draft.cursorInSegmentOffset = 0;
@@ -1141,6 +1145,13 @@ export function deleteToLineEnd(state: ChatUserInputState): ChatUserInputState {
       } else {
         currentSegment.content = contentBeforeCursor;
       }
+    } else if (currentOffset === 0) {
+      // At start of non-text segment - remove it along with everything to line end
+      draft.segments.splice(
+        currentSegmentIndex,
+        lineEndPosition.segmentIndex - currentSegmentIndex + 1,
+      );
+      return;
     }
 
     // Remove segments between cursor and line end (inclusive of line end if different)