82 lines
2.4 KiB
JavaScript
82 lines
2.4 KiB
JavaScript
/**
|
|
* 请求封装模块（VM沙箱版本）
|
|
* 支持两阶段请求：先获取动态JS生成Cookie，再携带Cookie请求数据
|
|
*/
|
|
|
|
const axios = require('axios');
|
|
|
|
const DEFAULT_HEADERS = {
|
|
'User-Agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36',
|
|
'Accept': 'application/json, text/javascript, */*; q=0.01',
|
|
'Accept-Language': 'zh-CN,zh;q=0.9,en;q=0.8',
|
|
'X-Requested-With': 'XMLHttpRequest',
|
|
};
|
|
|
|
class TwoPhaseClient {
|
|
constructor(options = {}) {
|
|
this.baseURL = options.baseURL || '';
|
|
this.cookies = options.cookies || {};
|
|
this.headers = { ...DEFAULT_HEADERS, ...options.headers };
|
|
this.delay = options.delay || 1500;
|
|
|
|
this.client = axios.create({
|
|
baseURL: this.baseURL,
|
|
timeout: options.timeout || 30000,
|
|
headers: this.headers,
|
|
});
|
|
}
|
|
|
|
getCookieString() {
|
|
return Object.entries(this.cookies).map(([k, v]) => `${k}=${v}`).join('; ');
|
|
}
|
|
|
|
setCookies(cookieObj) {
|
|
Object.assign(this.cookies, cookieObj);
|
|
}
|
|
|
|
/**
|
|
* 第一阶段：获取动态JS代码
|
|
* @param {string} url - 返回JS代码的接口
|
|
* @returns {string} JS 代码字符串
|
|
*/
|
|
async fetchDynamicJS(url, params = {}) {
|
|
const response = await this.client.get(url, {
|
|
params,
|
|
headers: { Cookie: this.getCookieString() },
|
|
transformResponse: [(data) => data],
|
|
});
|
|
return response.data;
|
|
}
|
|
|
|
/**
|
|
* 第二阶段：携带Cookie请求数据
|
|
*/
|
|
async fetchData(url, params = {}) {
|
|
const response = await this.client.get(url, {
|
|
params,
|
|
headers: { Cookie: this.getCookieString() },
|
|
});
|
|
return response.data;
|
|
}
|
|
|
|
async post(url, data = {}) {
|
|
const response = await this.client.post(url, data, {
|
|
headers: { Cookie: this.getCookieString() },
|
|
});
|
|
return response.data;
|
|
}
|
|
|
|
sleep(ms) {
|
|
return new Promise(resolve => setTimeout(resolve, ms));
|
|
}
|
|
|
|
async fetchWithDelay(url, params = {}) {
|
|
const data = await this.fetchData(url, params);
|
|
const jitter = Math.random() * this.delay * 0.3;
|
|
await this.sleep(this.delay + jitter);
|
|
return data;
|
|
}
|
|
}
|
|
|
|
module.exports = { TwoPhaseClient, DEFAULT_HEADERS };
|