diff --git a/src/routes/reports.ts b/src/routes/reports.ts index dd8dc01..6d7f6a9 100755 --- a/src/routes/reports.ts +++ b/src/routes/reports.ts @@ -32,9 +32,10 @@ export async function handleReportsRequest(req: Request): Promise { // Override headers if provided if (body.headers) { + const targetOrigin = new URL(body.url).origin; await page.route("**/*", async (route, request) => { const url = request.url(); - if (url.startsWith("http://appwrite/")) { + if (url === targetOrigin || url.startsWith(`${targetOrigin}/`)) { return await route.continue({ headers: { ...request.headers(), diff --git a/src/routes/screenshots.ts b/src/routes/screenshots.ts index 6752612..0ebf6c4 100755 --- a/src/routes/screenshots.ts +++ b/src/routes/screenshots.ts @@ -39,11 +39,12 @@ export async function handleScreenshotsRequest( } const page = await context.newPage(); + const targetOrigin = new URL(body.url).origin; - // Override headers + // Override headers only for the target origin await page.route("**/*", async (route, request) => { const url = request.url(); - if (url.startsWith("http://appwrite/")) { + if (url === targetOrigin || url.startsWith(`${targetOrigin}/`)) { return await route.continue({ headers: { ...request.headers(),