feat(srcDoc): added onload arg

diplodoc-platform · Oct 7, 2024 · 8fadb91 · 8fadb91
1 parent 546251e
commit 8fadb91
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/runtime/SrcDocIFrameController.ts b/src/runtime/SrcDocIFrameController.ts
@@ -79,8 +79,9 @@ export class SrcDocIFrameController extends Disposable implements IEmbeddedConte
 
         this.addAnchorLinkHandlers();
         this.handleHashChange();
-        if (this.config.onload) {
-            this.dispose.add(() => this.config.onload?.(this.host));
+        const unload = this.config.onload?.(this.host);
+        if (unload) {
+            this.dispose.add(unload);
         }
     }