Austin/reader
1
0
Fork 0
mirror of https://github.com/intergalacticalvariable/reader.git synced 2024-11-16 11:42:32 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix: Improve logging and validation for cookie setting in web scraping

Browse Source
This commit is contained in:
Harsh Gupta (aider) 2024-08-26 17:40:50 +05:30
parent 6309cbd7a0
commit cc4d316764
1 changed files with 23 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
backend/functions/src/services/puppeteer.ts
View File

@ -13,6 +13,16 @@ import { SecurityCompromiseError, ServiceCrashedError } from '../shared/errors';
import { TimeoutError } from 'puppeteer';
import tldExtract from 'tld-extract';
// Add this new function for cookie validation
const validateCookie = (cookie: CookieParam) => {
const requiredFields = ['name', 'value', 'domain'];
for (const field of requiredFields) {
if (!cookie[field]) {
throw new Error(`Cookie is missing required field: ${field}`);
}
}
};
const READABILITY_JS = fs.readFileSync(require.resolve('@mozilla/readability/Readability.js'), 'utf-8');
@ -472,12 +482,24 @@ document.addEventListener('load', handlePageLoad);
const page = await this.getNextPage();
const sn = this.snMap.get(page);
this.logger.info(`Page ${sn}: Scraping ${url}`, { url });
if (options?.proxyUrl) {
this.logger.info(`Page ${sn}: Using proxy:`, options.proxyUrl);
await page.useProxy(options.proxyUrl);
}
if (options?.cookies) {
this.logger.info(`Page ${sn}: Attempting to set cookies:`, JSON.stringify(options.cookies, null, 2));
try {
options.cookies.forEach(validateCookie);
await page.setCookie(...options.cookies);
} catch (error) {
this.logger.error(`Page ${sn}: Error setting cookies:`, error);
this.logger.info(`Page ${sn}: Problematic cookies:`, JSON.stringify(options.cookies, null, 2));
throw error;
}
}
if (options?.overrideUserAgent) {
await page.setUserAgent(options.overrideUserAgent);
}