39 lines
1.3 KiB
JavaScript
39 lines
1.3 KiB
JavaScript
import fs from "fs";
|
|
|
|
/**
|
|
* 设置页面的LocalStorage
|
|
* @param {import('puppeteer').Page} page
|
|
* @param {String} file 用户信息
|
|
*/
|
|
export async function setLocalStorage(page, file) {
|
|
const outputLocalStorge = fs.readFileSync(file, "utf-8");
|
|
let localStorageTest = JSON.parse(outputLocalStorge);
|
|
|
|
// 设置页面的localStorage
|
|
await page.evaluateOnNewDocument(localStorageTest => {
|
|
localStorage.clear();
|
|
localStorage.setItem("merchant_access_token", localStorageTest.merchant_access_token);
|
|
localStorage.setItem("merchant_refresh_token", localStorageTest.merchant_refresh_token);
|
|
localStorage.setItem("merchant_login_data", localStorageTest.merchant_login_data);
|
|
localStorage.setItem("merchant_metadata_data", localStorageTest.merchant_metadata_data);
|
|
}, localStorageTest);
|
|
|
|
console.log("set success");
|
|
}
|
|
|
|
/**
|
|
* @param {Page} page
|
|
* @param {String} webUrl 网站
|
|
* @param {String} cookieFile cookie地址Url
|
|
*/
|
|
export async function injectionCookie(page, webUrl, cookieFile) {
|
|
try {
|
|
await page.goto(webUrl);
|
|
const cookies = JSON.parse(fs.readFileSync(cookieFile, "utf-8"));
|
|
await page.setCookie(...cookies);
|
|
await page.goto(webUrl);
|
|
} catch (error) {
|
|
console.error(error);
|
|
}
|
|
}
|