From fa34214012d14385d231a1ac4f16006aaf4331fb Mon Sep 17 00:00:00 2001
From: Nate Sesti <sestinj@gmail.com>
Date: Tue, 1 Aug 2023 14:42:48 -0700
Subject: fix: :bug: fix bugs when selecting code to edit

---
 extension/react-app/src/components/PillButton.tsx                  | 4 +++-
 extension/react-app/src/hooks/AbstractContinueGUIClientProtocol.ts | 2 +-
 extension/react-app/src/hooks/ContinueGUIClientProtocol.ts         | 4 ++--
 3 files changed, 6 insertions(+), 4 deletions(-)

(limited to 'extension/react-app/src')

diff --git a/extension/react-app/src/components/PillButton.tsx b/extension/react-app/src/components/PillButton.tsx
index e3d05711..0b1aa23d 100644
--- a/extension/react-app/src/components/PillButton.tsx
+++ b/extension/react-app/src/components/PillButton.tsx
@@ -168,7 +168,9 @@ const PillButton = (props: PillButtonProps) => {
                   data-tooltip-id={`edit-${props.index}`}
                   backgroundColor={"#8800aa55"}
                   onClick={() => {
-                    client?.setEditingAtIndices([props.index]);
+                    client?.setEditingAtIds([
+                      props.item.description.id.item_id,
+                    ]);
                   }}
                 >
                   <PaintBrush
diff --git a/extension/react-app/src/hooks/AbstractContinueGUIClientProtocol.ts b/extension/react-app/src/hooks/AbstractContinueGUIClientProtocol.ts
index 8e3735ec..8d8b7b7e 100644
--- a/extension/react-app/src/hooks/AbstractContinueGUIClientProtocol.ts
+++ b/extension/react-app/src/hooks/AbstractContinueGUIClientProtocol.ts
@@ -23,7 +23,7 @@ abstract class AbstractContinueGUIClientProtocol {
 
   abstract deleteContextWithIds(ids: ContextItemId[]): void;
 
-  abstract setEditingAtIndices(indices: number[]): void;
+  abstract setEditingAtIds(ids: string[]): void;
 
   abstract toggleAddingHighlightedCode(): void;
 
diff --git a/extension/react-app/src/hooks/ContinueGUIClientProtocol.ts b/extension/react-app/src/hooks/ContinueGUIClientProtocol.ts
index 5a5d4c30..b6dd43d9 100644
--- a/extension/react-app/src/hooks/ContinueGUIClientProtocol.ts
+++ b/extension/react-app/src/hooks/ContinueGUIClientProtocol.ts
@@ -78,8 +78,8 @@ class ContinueGUIClientProtocol extends AbstractContinueGUIClientProtocol {
     });
   }
 
-  setEditingAtIndices(indices: number[]) {
-    this.messenger.send("set_editing_at_indices", { indices });
+  setEditingAtIds(ids: string[]) {
+    this.messenger.send("set_editing_at_ids", { ids });
   }
 
   toggleAddingHighlightedCode(): void {
-- 
cgit v1.2.3-70-g09d2