From 91475c2c4d6198a78f63cb9be785401b118d761b Mon Sep 17 00:00:00 2001
From: Egor Aristov <ef@efprojects.com>
Date: Sun, 23 Mar 2025 09:11:17 +0300
Subject: [PATCH] refactoring

---
 .../extractors/pwextractor/pwextractor.go     | 79 ++++++++++---------
 1 file changed, 43 insertions(+), 36 deletions(-)

diff --git a/internal/extractors/pwextractor/pwextractor.go b/internal/extractors/pwextractor/pwextractor.go
index b451161..ac3bc8e 100644
--- a/internal/extractors/pwextractor/pwextractor.go
+++ b/internal/extractors/pwextractor/pwextractor.go
@@ -145,42 +145,8 @@ func (e *PwExtractor) visitPage(task models.Task, cb func(page playwright.Page)
 		}
 	}()
 
-	if err := bCtx.Route("**", func(route playwright.Route) {
-		log.Debugf("Route: %s", route.Request().URL())
-		allowHost, err := e.allowHost(route.Request().URL())
-		if err != nil {
-			log.Errorf("Allow host: %v", err)
-			allowHost = false
-		}
-		if allowHost {
-			if err := route.Continue(); err != nil {
-				log.Warnf("Route continue error: %v", err)
-			}
-		} else {
-			if err := route.Abort(); err != nil {
-				log.Warnf("Route abort error: %v", err)
-			}
-		}
-	}); err != nil {
-		return fmt.Errorf("set route: %w", err)
-	}
-
-	if err := bCtx.RouteWebSocket("**", func(route playwright.WebSocketRoute) {
-		log.Debugf("Websocket route: %s", route.URL())
-		allowHost, err := e.allowHost(route.URL())
-		if err != nil {
-			log.Errorf("Allow host: %v", err)
-			allowHost = false
-		}
-		if allowHost {
-			if _, err := route.ConnectToServer(); err != nil {
-				log.Warnf("Websocket connect error: %v", err)
-			}
-		} else {
-			route.Close()
-		}
-	}); err != nil {
-		return fmt.Errorf("websocket set route: %w", err)
+	if err := e.setupInterceptors(bCtx); err != nil {
+		return fmt.Errorf("setup interceptors: %w", err)
 	}
 
 	if len(cookies) > 0 {
@@ -252,6 +218,47 @@ func (e *PwExtractor) visitPage(task models.Task, cb func(page playwright.Page)
 	return err
 }
 
+func (e *PwExtractor) setupInterceptors(bCtx playwright.BrowserContext) error {
+	if err := bCtx.Route("**", func(route playwright.Route) {
+		log.Debugf("Route: %s", route.Request().URL())
+		allowHost, err := e.allowHost(route.Request().URL())
+		if err != nil {
+			log.Errorf("Allow host: %v", err)
+			allowHost = false
+		}
+		if allowHost {
+			if err := route.Continue(); err != nil {
+				log.Warnf("Route continue error: %v", err)
+			}
+		} else {
+			if err := route.Abort(); err != nil {
+				log.Warnf("Route abort error: %v", err)
+			}
+		}
+	}); err != nil {
+		return fmt.Errorf("set route: %w", err)
+	}
+
+	if err := bCtx.RouteWebSocket("**", func(route playwright.WebSocketRoute) {
+		log.Debugf("Websocket route: %s", route.URL())
+		allowHost, err := e.allowHost(route.URL())
+		if err != nil {
+			log.Errorf("Allow host: %v", err)
+			allowHost = false
+		}
+		if allowHost {
+			if _, err := route.ConnectToServer(); err != nil {
+				log.Warnf("Websocket connect error: %v", err)
+			}
+		} else {
+			route.Close()
+		}
+	}); err != nil {
+		return fmt.Errorf("websocket set route: %w", err)
+	}
+	return nil
+}
+
 func (e *PwExtractor) allowHost(rawUrl string) (bool, error) {
 	ips, err := getIPs(rawUrl)
 	if err != nil {