moved background removal code into ImageUtils

1 files changed, 48 insertions, 0 deletions

diff --git a/src/client/util/Import & Export/ImageUtils.ts b/src/client/util/Import & Export/ImageUtils.ts
index eae3460b3..e2eab9010 100644
--- a/src/client/util/Import & Export/ImageUtils.ts
+++ b/src/client/util/Import & Export/ImageUtils.ts
@@ -1,9 +1,11 @@
 import { ClientUtils } from '../../../ClientUtils';
 import { Doc } from '../../../fields/Doc';
+import { DocData } from '../../../fields/DocSymbols';
 import { Id } from '../../../fields/FieldSymbols';
 import { Cast, NumCast, StrCast } from '../../../fields/Types';
 import { ImageField } from '../../../fields/URLField';
 import { Upload } from '../../../server/SharedMediaTypes';
+import { Docs, DocumentOptions } from '../../documents/Documents';
 import { Networking } from '../../Network';
 
 export namespace ImageUtils {
@@ -31,4 +33,50 @@ export namespace ImageUtils {
         a.download = `Dash Export [${StrCast(collection.title)}].zip`;
         a.click();
     };
+
+    export function convertImgBlobToDataURL(blob: unknown) {
+        return new Promise<string | ArrayBuffer | null>((resolve, reject) => {
+            const reader = new FileReader();
+            reader.onload = () => resolve(reader.result);
+            reader.onerror = e => reject(e);
+            reader.readAsDataURL(blob as Blob);
+        });
+    };
+    export function createImageDocFromBlob(blob: Blob, options: DocumentOptions, docId: string, nativeWidth: number, nativeHeight: number): Promise<Doc> {
+        return new Promise((resolve, reject) => {
+            convertImgBlobToDataURL(blob).then(durl => {
+                ClientUtils.convertDataUri(durl as string, docId + '_noBgd').then(url => {
+                    const imageSnapshot = Docs.Create.ImageDocument(url, options);
+                    Doc.SetNativeWidth(imageSnapshot[DocData], nativeWidth);
+                    Doc.SetNativeHeight(imageSnapshot[DocData], nativeHeight);
+                    resolve(imageSnapshot);
+                }).catch(reject);
+            }).catch(reject);
+        });
+    }
+    let backgroundRemovalWorker: Worker|undefined;
+    const workerCallbackMap = new Map<string, (result: Doc) => void>();
+    function getBackgroundRemovalWorker()  {
+        if (backgroundRemovalWorker) return backgroundRemovalWorker;
+        backgroundRemovalWorker = new Worker('/image.worker.js', { type: 'module' });
+        backgroundRemovalWorker.onerror = (e: ErrorEvent) => console.error('Worker failed:', e); //  worker.terminate();
+        backgroundRemovalWorker.onmessage = async (event: MessageEvent) => {
+            if (event.data.success) {
+                createImageDocFromBlob(event.data.result, event.data.options, event.data.docId, event.data.nativeWidth, event.data.nativeHeight)
+                    .then(imageSnapshot => {
+                        workerCallbackMap.get(event.data.docId)?.(imageSnapshot);
+                        workerCallbackMap.delete(event.data.docId);
+                    })
+            } else {
+                console.error('Error in background removal:', event.data.error);
+            }
+            // worker.terminate();
+        };
+        return backgroundRemovalWorker;
+
+    }
+    export function removeImgBackground(options:DocumentOptions, cb: (result: Doc) => void, docId: string, nativeWidth:number, nativeHeight:number, docImgPath: string) {
+        workerCallbackMap.set(docId, cb); // Store the callback by docId (or use a unique requestId)
+        getBackgroundRemovalWorker().postMessage({ imagePath: docImgPath, options, docId, nativeWidth, nativeHeight });
+    };
 }