nodejs_chromium 1.0.1 → 1.0.2

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "nodejs_chromium",
-  "version": "1.0.1",
+  "version": "1.0.2",
   "description": "",
   "main": "index.js",
   "scripts": {

package/src/chrome.js

@@ -45,8 +45,12 @@ module.exports = class {
 		this.responseCall = call;
 	}
 
+	async setCookie(cookies) {
+		await this.page.setCookie(...cookies);
+	}
+
 	async open(url) {
-		return await this.page.goto(url, { waitUntil: 'load' });
+		await this.page.goto(url, { waitUntil: 'load' });
 	}
 
 	/**
@@ -54,15 +58,14 @@ module.exports = class {
 	 *
 	 * @param url
 	 * @param data
-	 * @returns {Promise<*>}
 	 */
 	async post(url, data) {
 		//JSON.stringify(data)
-		return await this.page.goto(url, { method: 'POST', body: data, waitUntil: 'load' });
+		await this.page.goto(url, { method: 'POST', body: data, waitUntil: 'load' });
 	}
 
 	async input(el, value) {
-		return await this.page.type(el, value);
+		await this.page.type(el, value);
 	}
 
 	async click(el) {
@@ -87,7 +90,7 @@ module.exports = class {
 		if (ele) {
 			return await this.waitSelector(ele, timeout, tryCount);
 		}
-		return await this.page.waitForNavigation();
+		await this.page.waitForNavigation();
 	}
 
 	async waitSelector(ele, timeout = 0, tryCount = 0) {
@@ -104,52 +107,6 @@ module.exports = class {
 		}
 	}
 
-	async parseResponse(response) {
-		// const response = await this.page.waitForResponse(res => res);
-
-		const value = {};
-		const request = await response.request();
-		const headers = await response.headers();
-		value.method = await request.method();
-		if (value.method === 'OPTIONS') return;
-
-		value.type = await request.resourceType();
-		// value.redirect = await response.redirectURL();
-		if (value.type === 'xhr') value.type = 'AJAX';
-		value.url = await response.url();
-		value.domain = parseUrl(value.url)['host'];
-		value.content = headers['content-type'];
-		value.length = headers['content-length'];
-		value.status = await response.status();
-		value.ok = await response.ok();
-		value.datetime = (new Date(headers['date'])).date('yyyy-mm-dd hh:ii:ss');
-		// value.headers = headers;
-		if (headers['server']) value.server = headers['server'];
-		if (headers['set-cookie']) value.cookies = headers['set-cookie'];
-		value.remote = await response.remoteAddress(); //目标服务器
-		if (value.status === 301 || value.status === 302) return value;
-		if (['image', 'font', 'other', 'script', 'stylesheet', 'document', 'ping', 'fetch'].has(value.type)) return value;
-		if (value.content) {
-			if (value.content.startsWith('application/vnd')) return value;
-			if (value.content.startsWith('application/xml')) return value;
-			if (value.content.startsWith('text/css')) return value;
-		}
-
-		value.post = await request.postData();
-		if (value.post) value.post = value.post.toString();
-
-		try {
-			value.response = await response.buffer();
-			value.response = value.response.toString();
-		}
-		catch (e) {
-			value.response = e.parse();
-		}
-
-		return value;
-	}
-
-
 	async saveHtml(file) {
 		try {
 			await this.improveUrls(); //修正js/css的域名
@@ -175,7 +132,7 @@ module.exports = class {
 
 	async watermark(conf) {
 		await this.page.evaluate((conf) => {
-			const wmDiv = document.createElement('div');
+			const wmDiv = document.createElement('div#watermark');
 			wmDiv.style.position = 'fixed';
 			wmDiv.style.top = `50%`;
 			wmDiv.style.left = `50%`;
@@ -188,7 +145,7 @@ module.exports = class {
 			wmDiv.innerText = conf.text;
 			document.body.appendChild(wmDiv);
 		}, conf);
-		await this.page.waitForSelector('div');
+		await this.page.waitForSelector('div#watermark', { timeout: 1000 });
 	}
 
 
@@ -206,7 +163,7 @@ module.exports = class {
 			quality: quality,
 			omitBackground: true, //显示背景
 		});
-		console.log('photograph=', file);
+		// console.log('photograph=', file);
 	}
 
 	/**
@@ -228,4 +185,56 @@ module.exports = class {
 		}, domain);
 	}
 
+
+	async parseResponse(response) {
+		// const response = await this.page.waitForResponse(res => res);
+
+		const value = {};
+		const request = await response.request();
+		const headers = await response.headers();
+		value.method = await request.method();
+		if (value.method === 'OPTIONS') return;
+
+		value.type = await request.resourceType();
+		// value.redirect = await response.redirectURL();
+		if (value.type === 'xhr') value.type = 'AJAX';
+		value.url = await response.url();
+		value.domain = parseUrl(value.url)['host'];
+		value.content = headers['content-type'];
+		value.length = headers['content-length'];
+		value.status = await response.status();
+		value.ok = await response.ok();
+		value.datetime = (new Date(headers['date'])).date('yyyy-mm-dd hh:ii:ss');
+		// value.headers = headers;
+		if (headers['server']) value.server = headers['server'];
+		if (headers['set-cookie']) value.cookies = headers['set-cookie'];
+		value.remote = await response.remoteAddress(); //目标服务器
+		if (value.status === 301 || value.status === 302) return value;
+		if (['image', 'font', 'other', 'script', 'stylesheet', 'document', 'ping', 'fetch'].has(value.type)) return value;
+		if (value.content) {
+			if (value.content.startsWith('application/vnd')) return value;
+			if (value.content.startsWith('application/xml')) return value;
+			if (value.content.startsWith('text/css')) return value;
+		}
+
+		value.post = await request.postData();
+		if (value.post) value.post = value.post.toString();
+
+		try {
+			value.response = await response.buffer();
+			value.response = value.response.toString();
+		}
+		catch (e) {
+			value.response = e.parse();
+		}
+
+		return value;
+	}
+
+
+
+
+
+
+
 }