Harden initial GitHub diff rescans

Author: TennyZhuang

package-lock.json

@@ -1,6 +1,6 @@
 {
   "name": "show-me-talk",
-  "version": "0.1.1",
+  "version": "0.1.2",
   "description": "Chrome extension that auto-collapses code in GitHub PRs so only the talk stays open.",
   "type": "module",
   "private": false,

package.json

@@ -22,6 +22,8 @@ export class GitHubPullRequestController {
 
   private followUpScanTimers: number[] = [];
 
+  private bootstrapScanIntervalId: number | null = null;
+
   private settings: ShowMeTalkSettings = DEFAULT_SETTINGS;
 
   private currentUrl = window.location.href;
@@ -39,6 +41,13 @@ export class GitHubPullRequestController {
     this.scheduleScan();
   };
 
+  private readonly boundHandleVisibilityChange = (): void => {
+    if (document.visibilityState === 'visible') {
+      this.scheduleScan();
+      this.startBootstrapScanLoop();
+    }
+  };
+
   private readonly boundHandleClicks = (event: Event): void => {
     const target = event.target;
     if (!(target instanceof HTMLElement)) {
@@ -77,6 +86,7 @@ export class GitHubPullRequestController {
     this.bind();
     this.scheduleScan();
     this.queueFollowUpScans();
+    this.startBootstrapScanLoop();
   }
 
   private bind(): void {
@@ -93,6 +103,7 @@ export class GitHubPullRequestController {
     document.addEventListener('turbo:render', this.boundHandleMutations);
     window.addEventListener('popstate', this.boundHandleMutations);
     window.addEventListener('scroll', this.boundHandleScroll, { passive: true });
+    document.addEventListener('visibilitychange', this.boundHandleVisibilityChange);
     chrome.storage.onChanged.addListener(this.boundHandleStorageChanges);
   }
 
@@ -109,6 +120,7 @@ export class GitHubPullRequestController {
     this.currentUrl = window.location.href;
     this.manualOverrides.clear();
     this.queueFollowUpScans();
+    this.startBootstrapScanLoop();
   }
 
   private scheduleScan(force = false): void {
@@ -138,6 +150,25 @@ export class GitHubPullRequestController {
     );
   }
 
+  private startBootstrapScanLoop(): void {
+    if (this.bootstrapScanIntervalId !== null) {
+      window.clearInterval(this.bootstrapScanIntervalId);
+    }
+
+    const startedAt = Date.now();
+    this.bootstrapScanIntervalId = window.setInterval(() => {
+      if (Date.now() - startedAt > 15_000) {
+        if (this.bootstrapScanIntervalId !== null) {
+          window.clearInterval(this.bootstrapScanIntervalId);
+          this.bootstrapScanIntervalId = null;
+        }
+        return;
+      }
+
+      this.scheduleScan();
+    }, 750);
+  }
+
   private scanPage(): void {
     const url = new URL(window.location.href);
     if (!isPullRequestPage(url) || !this.settings.enabled) {