Removing queued images when removing folder

wladimiiir · Oct 30, 2024 · cf836b2 · cf836b2
1 parent d339ac7
commit cf836b2
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/src/main/ipcHandlers.ts b/src/main/ipcHandlers.ts
@@ -105,8 +105,10 @@ export const setupIpcHandlers = (webContents: Electron.WebContents, store: Store
 
   ipcMain.handle('remove-images-in-folder', async (_event, folderPath: string) => {
     const images = store.getImages();
+    const imagesToRemove = images.filter((image) => image.id.startsWith(folderPath));
     const updatedImages = images.filter((image) => !image.id.startsWith(folderPath));
     store.setImages(updatedImages);
+    modelManager.removeQueuedImages(imagesToRemove);
     return updatedImages;
   });
 };
diff --git a/src/main/modelManager.ts b/src/main/modelManager.ts
@@ -80,4 +80,9 @@ export class ModelManager {
 
     return modelProvider.generateImageInfo(settings, image);
   }
+
+  removeQueuedImages(imagesToRemove: Image[]): void {
+    const imageIds = new Set(imagesToRemove.map((img) => img.id));
+    this.queue = this.queue.filter((item) => !imageIds.has(item.image.id));
+  }
 }