cancel edit, don't regenerate when the same content

3.15.3
twinnydotdev · Aug 16, 2024 · 36e648b · 36e648b
1 parent 744f442
commit 36e648b
Show file tree

Hide file tree

Showing 3 changed files with 27 additions and 10 deletions.
diff --git a/package-lock.json b/package-lock.json
diff --git a/package.json b/package.json
@@ -2,7 +2,7 @@
   "name": "twinny",
   "displayName": "twinny - AI Code Completion and Chat",
   "description": "Locally hosted AI code completion plugin for vscode",
-  "version": "3.15.2",
+  "version": "3.15.3",
   "icon": "assets/icon.png",
   "keywords": [
     "code-inference",

diff --git a/src/webview/message.tsx b/src/webview/message.tsx
@@ -66,8 +66,16 @@ export const Message = ({
 
   const handleRegenerate = () => onRegenerate?.(index)
 
+  const handleToggleCancel = () => {
+    setEditing(false)
+    editor?.commands.setContent(message.content as string)
+  }
+
   const handleToggleSave = () => {
     const content = editor?.storage.markdown.getMarkdown()
+    if (message.content === content) {
+      return setEditing(false)
+    }
     onUpdate?.(content || '', index)
     setEditing(false)
   }
@@ -105,13 +113,22 @@ export const Message = ({
           <span>{message.role === ASSISTANT ? TWINNY : YOU}</span>
           <div className={styles.messageOptions}>
             {editing && !isAssistant && (
-              <VSCodeButton
-                title="Save message"
-                appearance="icon"
-                onClick={handleToggleSave}
-              >
-                <span className="codicon codicon-check"></span>
-              </VSCodeButton>
+              <>
+                <VSCodeButton
+                  title="Cancel edit"
+                  appearance="icon"
+                  onClick={handleToggleCancel}
+                >
+                  <span className="codicon codicon-close"></span>
+                </VSCodeButton>
+                <VSCodeButton
+                  title="Save message"
+                  appearance="icon"
+                  onClick={handleToggleSave}
+                >
+                  <span className="codicon codicon-check"></span>
+                </VSCodeButton>
+              </>
             )}
             {!editing && !isAssistant && (
               <>