allow setting panic func for browsers

browser.go

@@ -34,6 +34,8 @@ type Browser struct {
 	// BrowserContextID is the id for incognito window
 	BrowserContextID proto.BrowserBrowserContextID
 
+	e func(args ...interface{})
+
 	ctx context.Context
 
 	sleeper func() utils.Sleeper
@@ -60,7 +62,7 @@ type Browser struct {
 // DefaultDevice is set to devices.LaptopWithMDPIScreen.Landescape() . You can use
 // NoDefaultDevice to disable it.
 func New() *Browser {
-	return &Browser{
+	return (&Browser{
 		ctx:           context.Background(),
 		sleeper:       DefaultSleeper,
 		slowMotion:    defaults.Slow,
@@ -70,7 +72,7 @@ func New() *Browser {
 		defaultDevice: devices.LaptopWithMDPIScreen.Landescape(),
 		targetsLock:   &sync.Mutex{},
 		states:        &sync.Map{},
-	}
+	}).WithPanic(utils.Panic)
 }
 
 // Incognito creates a new incognito browser
@@ -248,6 +250,7 @@ func (b *Browser) Call(ctx context.Context, sessionID, methodName string, params
 // PageFromSession is used for low-level debugging
 func (b *Browser) PageFromSession(sessionID proto.TargetSessionID) *Page {
 	return &Page{
+		e:         b.e,
 		ctx:       b.ctx,
 		sleeper:   b.sleeper,
 		browser:   b,
@@ -274,6 +277,7 @@ func (b *Browser) PageFromTarget(targetID proto.TargetTargetID) (*Page, error) {
 	}
 
 	page = &Page{
+		e:         b.e,
 		ctx:       b.ctx,
 		sleeper:   b.sleeper,
 		browser:   b,

element.go

@@ -26,6 +26,8 @@ var _ proto.Sessionable = &Element{}
 type Element struct {
 	Object *proto.RuntimeRemoteObject
 
+	e func(args ...interface{})
+
 	ctx context.Context
 
 	sleeper func() utils.Sleeper

hijack.go

@@ -174,6 +174,8 @@ func (r *HijackRouter) new(ctx context.Context, e *proto.FetchRequestPaused) *Hi
 			},
 		},
 		OnError: func(err error) {},
+
+		browser: r.browser,
 	}
 }
 
@@ -208,6 +210,8 @@ type Hijack struct {
 
 	// CustomState is used to store things for this context
 	CustomState interface{}
+
+	browser *Browser
 }
 
 // ContinueRequest without hijacking. The RequestID will be set by the router, you don't have to set it.

lib/utils/utils.go

@@ -60,11 +60,14 @@ func MultiLogger(list ...Logger) Log {
 	})
 }
 
+// Panic is the same as the built-in panic
+var Panic = func(v interface{}) { panic(v) }
+
 // E if the last arg is error, panic it
 func E(args ...interface{}) []interface{} {
 	err, ok := args[len(args)-1].(error)
 	if ok {
-		panic(err)
+		Panic(err)
 	}
 	return args
 }