catch edge case where the _biloba object isn't available because the …

…browser is in the middle of a redirect
onsi · Sep 18, 2023 · d9df233 · d9df233
1 parent 20acefd
commit d9df233
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/biloba.go b/biloba.go
@@ -533,6 +533,10 @@ func (b *Biloba) ensureBiloba() {
 	if installed {
 		return
 	}
+	b.reloadBiloba()
+}
+
+func (b *Biloba) reloadBiloba() {
 	b.Run(bilobaJS)
 	b.lock.Lock()
 	b.bilobaIsInstalled = true

diff --git a/javascript.go b/javascript.go
@@ -50,6 +50,10 @@ func (b *Biloba) RunErr(script string, args ...any) (any, error) {
 	}
 	err := chromedp.Run(b.Context, chromedp.EvaluateAsDevTools(script, &encodedResult, withUserGesture))
 	if err != nil {
+		if strings.Contains(err.Error(), "_biloba is not defined") {
+			b.reloadBiloba()
+			return b.RunErr(script, args...)
+		}
 		return nil, err
 	}