fix: #32 iframe loading bug with BBT

windingwind · Oct 17, 2022 · 2357604 · 2357604
1 parent e2ca919
commit 2357604
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/src/events.ts b/src/events.ts
@@ -260,9 +260,10 @@ class AddonEvents extends AddonModule {
       const tabIndex = tabbox.selectedIndex;
       tabbox.selectedIndex =
         tabIndex === tabbox.childNodes.length - 1 ? 1 : tabIndex + 1;
-      setTimeout(() => {
+      setTimeout(async () => {
         tabbox.selectedIndex = tabIndex;
-        this.initPreview(PreviewType.info);
+        await this.initPreview(PreviewType.info);
+        this.doPreview(true);
       }, 1);
     }
 
@@ -378,8 +379,10 @@ class AddonEvents extends AddonModule {
     if (!container) {
       return;
     }
+    const iframeId = this._Addon.preview.getPreviewIds(type).iframeId;
+    document.getElementById(iframeId)?.remove();
     const iframe = document.createElement("iframe");
-    iframe.setAttribute("id", this._Addon.preview.getPreviewIds(type).iframeId);
+    iframe.setAttribute("id", iframeId);
     iframe.setAttribute("src", "chrome://PDFPreview/content/previewPDF.html");
     container.appendChild(iframe);
     await this._Addon.preview._initPromise;