fix removeFigure function

KonnorRogers · Apr 23, 2024 · 6229136 · 6229136
1 parent 45e055e
commit 6229136
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/src/exports/extensions/attachment.ts b/src/exports/extensions/attachment.ts
@@ -712,7 +712,7 @@ export const Attachment = Node.create<AttachmentOptions>({
         }
       };
 
-      const removeFigure = () => {
+      function removeFigure (this: HTMLElement) {
         if (typeof getPos === "function") {
           const { view } = editor;
 
@@ -722,6 +722,12 @@ export const Attachment = Node.create<AttachmentOptions>({
           tr.delete(pos, pos + 1);
           view.dispatch(tr);
         }
+
+        // For some reason it doesnt always delete the attachment, so this is some extra insurance.
+        const closestAttachment = this.closest(".attachment")
+        if (closestAttachment) {
+          closestAttachment.remove()
+        }
       };
 
       const template = html`