fix(playwright): only create adblocker if flag is true

huggingface · Jul 12, 2024 · df67a2c · df67a2c
1 parent d67b280
commit df67a2c
Showing 1 changed file with 14 additions and 11 deletions.
diff --git a/src/lib/server/websearch/scrape/playwright.ts b/src/lib/server/websearch/scrape/playwright.ts
@@ -11,16 +11,19 @@ import { env } from "$env/dynamic/private";
 import { logger } from "$lib/server/logger";
 import { onExit } from "$lib/server/exitHandler";
 
-const blocker = await PlaywrightBlocker.fromPrebuiltAdsAndTracking(fetch)
-	.then((blker) => {
-		const mostBlocked = blker.blockFonts().blockMedias().blockFrames().blockImages();
-		if (env.WEBSEARCH_JAVASCRIPT === "false") return mostBlocked.blockScripts();
-		return mostBlocked;
-	})
-	.catch((err) => {
-		logger.error(err, "Failed to initialize PlaywrightBlocker from prebuilt lists");
-		return PlaywrightBlocker.empty();
-	});
+const blocker =
+	env.PLAYWRIGHT_ADBLOCKER === "true"
+		? await PlaywrightBlocker.fromPrebuiltAdsAndTracking(fetch)
+				.then((blker) => {
+					const mostBlocked = blker.blockFonts().blockMedias().blockFrames().blockImages();
+					if (env.WEBSEARCH_JAVASCRIPT === "false") return mostBlocked.blockScripts();
+					return mostBlocked;
+				})
+				.catch((err) => {
+					logger.error(err, "Failed to initialize PlaywrightBlocker from prebuilt lists");
+					return PlaywrightBlocker.empty();
+				})
+		: PlaywrightBlocker.empty();
 
 let browserSingleton: Promise<Browser> | undefined;
 async function getBrowser() {
@@ -65,7 +68,7 @@ export async function withPage<T>(
 
 	try {
 		const page = await ctx.newPage();
-		process.env.PLAYWRIGHT_ADBLOCKER === "true" && (await blocker.enableBlockingInPage(page));
+		env.PLAYWRIGHT_ADBLOCKER === "true" && (await blocker.enableBlockingInPage(page));
 
 		const res = await page.goto(url, { waitUntil: "load", timeout: 3500 }).catch(() => {
 			console.warn(`Failed to load page within 2s: ${url}`);