Merge pull request #26 from cesardeazevedo/remove-stuff

Split removeBlobs up
cesardeazevedo · Feb 11, 2025 · 994b5cf · 994b5cf
2 parents 6cfe4c4 + 4985261
commit 994b5cf
Showing 1 changed file with 20 additions and 17 deletions.
diff --git a/src/extensions/FileUploadExtension.ts b/src/extensions/FileUploadExtension.ts
@@ -15,7 +15,8 @@ declare module '@tiptap/core' {
       addFile: (file: File, pos: number) => ReturnType
       selectFiles: () => ReturnType
       uploadFiles: () => ReturnType
-      removeBlobs: (errorsOnly?: boolean) => ReturnType
+      removeFailedUploads: () => ReturnType
+      removePendingUploads: () => ReturnType
     }
   }
 }
@@ -79,22 +80,24 @@ export const FileUploadExtension = Extension.create<FileUploadOptions, FileUploa
         props.tr.setMeta('uploadFiles', true)
         return true
       },
-      removeBlobs:
-        (errorsOnly = false) =>
-        (props) => {
-          props.state.doc.descendants((node, pos) => {
-            if (!(node.type.name === 'image' || node.type.name === 'video')) {
-              return
-            }
-            if (node.attrs.src.startsWith('blob:')) {
-              if (errorsOnly && !node.attrs.uploadError) {
-                return
-              }
-              props.state.tr.delete(pos, pos + node.nodeSize)
-            }
-          })
-          return true
-        },
+      removePendingUploads: () => (props) => {
+        props.state.doc.descendants((node, pos) => {
+          if (['image', 'video'].includes(node.type.name) && node.attrs.uploading) {
+            props.state.tr.delete(pos, pos + node.nodeSize)
+          }
+        })
+
+        return true
+      },
+      removeFailedUploads: () => (props) => {
+        props.state.doc.descendants((node, pos) => {
+          if (['image', 'video'].includes(node.type.name) && node.attrs.uploadError) {
+            props.state.tr.delete(pos, pos + node.nodeSize)
+          }
+        })
+
+        return true
+      },
     }
   },