456 lines
19 KiB
JavaScript
456 lines
19 KiB
JavaScript
// ==UserScript==
|
|
// @name ajaxHooker
|
|
// @author cxxjackie
|
|
// @version 1.4.3
|
|
// @supportURL https://bbs.tampermonkey.net.cn/thread-3284-1-1.html
|
|
// ==/UserScript==
|
|
|
|
var ajaxHooker = function() {
|
|
'use strict';
|
|
const version = '1.4.3';
|
|
const hookInst = {
|
|
hookFns: [],
|
|
filters: []
|
|
};
|
|
const win = window.unsafeWindow || document.defaultView || window;
|
|
let winAh = win.__ajaxHooker;
|
|
const resProto = win.Response.prototype;
|
|
const xhrResponses = ['response', 'responseText', 'responseXML'];
|
|
const fetchResponses = ['arrayBuffer', 'blob', 'formData', 'json', 'text'];
|
|
const fetchInitProps = ['method', 'headers', 'body', 'mode', 'credentials', 'cache', 'redirect',
|
|
'referrer', 'referrerPolicy', 'integrity', 'keepalive', 'signal', 'priority'];
|
|
const xhrAsyncEvents = ['readystatechange', 'load', 'loadend'];
|
|
const getType = ({}).toString.call.bind(({}).toString);
|
|
const getDescriptor = Object.getOwnPropertyDescriptor.bind(Object);
|
|
const emptyFn = () => {};
|
|
const errorFn = e => console.error(e);
|
|
function isThenable(obj) {
|
|
return obj && ['object', 'function'].includes(typeof obj) && typeof obj.then === 'function';
|
|
}
|
|
function catchError(fn, ...args) {
|
|
try {
|
|
const result = fn(...args);
|
|
if (isThenable(result)) return result.then(null, errorFn);
|
|
return result;
|
|
} catch (err) {
|
|
console.error(err);
|
|
}
|
|
}
|
|
function defineProp(obj, prop, getter, setter) {
|
|
Object.defineProperty(obj, prop, {
|
|
configurable: true,
|
|
enumerable: true,
|
|
get: getter,
|
|
set: setter
|
|
});
|
|
}
|
|
function readonly(obj, prop, value = obj[prop]) {
|
|
defineProp(obj, prop, () => value, emptyFn);
|
|
}
|
|
function writable(obj, prop, value = obj[prop]) {
|
|
Object.defineProperty(obj, prop, {
|
|
configurable: true,
|
|
enumerable: true,
|
|
writable: true,
|
|
value: value
|
|
});
|
|
}
|
|
function parseHeaders(obj) {
|
|
const headers = {};
|
|
switch (getType(obj)) {
|
|
case '[object String]':
|
|
for (const line of obj.trim().split(/[\r\n]+/)) {
|
|
const [header, value] = line.split(/\s*:\s*/);
|
|
if (!header) break;
|
|
const lheader = header.toLowerCase();
|
|
headers[lheader] = lheader in headers ? `${headers[lheader]}, ${value}` : value;
|
|
}
|
|
break;
|
|
case '[object Headers]':
|
|
for (const [key, val] of obj) {
|
|
headers[key] = val;
|
|
}
|
|
break;
|
|
case '[object Object]':
|
|
return {...obj};
|
|
}
|
|
return headers;
|
|
}
|
|
function stopImmediatePropagation() {
|
|
this.ajaxHooker_isStopped = true;
|
|
}
|
|
class SyncThenable {
|
|
then(fn) {
|
|
fn && fn();
|
|
return new SyncThenable();
|
|
}
|
|
}
|
|
class AHRequest {
|
|
constructor(request) {
|
|
this.request = request;
|
|
this.requestClone = {...this.request};
|
|
}
|
|
shouldFilter(filters) {
|
|
const {type, url, method, async} = this.request;
|
|
return filters.length && !filters.find(obj => {
|
|
switch (true) {
|
|
case obj.type && obj.type !== type:
|
|
case getType(obj.url) === '[object String]' && !url.includes(obj.url):
|
|
case getType(obj.url) === '[object RegExp]' && !obj.url.test(url):
|
|
case obj.method && obj.method.toUpperCase() !== method.toUpperCase():
|
|
case 'async' in obj && obj.async !== async:
|
|
return false;
|
|
}
|
|
return true;
|
|
});
|
|
}
|
|
waitForRequestKeys() {
|
|
const requestKeys = ['url', 'method', 'abort', 'headers', 'data'];
|
|
if (!this.request.async) {
|
|
win.__ajaxHooker.hookInsts.forEach(({hookFns, filters}) => {
|
|
if (this.shouldFilter(filters)) return;
|
|
hookFns.forEach(fn => {
|
|
if (getType(fn) === '[object Function]') catchError(fn, this.request);
|
|
});
|
|
requestKeys.forEach(key => {
|
|
if (isThenable(this.request[key])) this.request[key] = this.requestClone[key];
|
|
});
|
|
});
|
|
return new SyncThenable();
|
|
}
|
|
const promises = [];
|
|
win.__ajaxHooker.hookInsts.forEach(({hookFns, filters}) => {
|
|
if (this.shouldFilter(filters)) return;
|
|
promises.push(Promise.all(hookFns.map(fn => catchError(fn, this.request))).then(() =>
|
|
Promise.all(requestKeys.map(key => Promise.resolve(this.request[key]).then(
|
|
val => this.request[key] = val,
|
|
() => this.request[key] = this.requestClone[key]
|
|
)))
|
|
));
|
|
});
|
|
return Promise.all(promises);
|
|
}
|
|
waitForResponseKeys(response) {
|
|
const responseKeys = this.request.type === 'xhr' ? xhrResponses : fetchResponses;
|
|
if (!this.request.async) {
|
|
if (getType(this.request.response) === '[object Function]') {
|
|
catchError(this.request.response, response);
|
|
responseKeys.forEach(key => {
|
|
if ('get' in getDescriptor(response, key) || isThenable(response[key])) {
|
|
delete response[key];
|
|
}
|
|
});
|
|
}
|
|
return new SyncThenable();
|
|
}
|
|
return Promise.resolve(catchError(this.request.response, response)).then(() =>
|
|
Promise.all(responseKeys.map(key => {
|
|
const descriptor = getDescriptor(response, key);
|
|
if (descriptor && 'value' in descriptor) {
|
|
return Promise.resolve(descriptor.value).then(
|
|
val => response[key] = val,
|
|
() => delete response[key]
|
|
);
|
|
} else {
|
|
delete response[key];
|
|
}
|
|
}))
|
|
);
|
|
}
|
|
}
|
|
const proxyHandler = {
|
|
get(target, prop) {
|
|
const descriptor = getDescriptor(target, prop);
|
|
if (descriptor && !descriptor.configurable && !descriptor.writable && !descriptor.get) return target[prop];
|
|
const ah = target.__ajaxHooker;
|
|
if (ah && ah.proxyProps) {
|
|
if (prop in ah.proxyProps) {
|
|
const pDescriptor = ah.proxyProps[prop];
|
|
if ('get' in pDescriptor) return pDescriptor.get();
|
|
if (typeof pDescriptor.value === 'function') return pDescriptor.value.bind(ah);
|
|
return pDescriptor.value;
|
|
}
|
|
if (typeof target[prop] === 'function') return target[prop].bind(target);
|
|
}
|
|
return target[prop];
|
|
},
|
|
set(target, prop, value) {
|
|
const descriptor = getDescriptor(target, prop);
|
|
if (descriptor && !descriptor.configurable && !descriptor.writable && !descriptor.set) return true;
|
|
const ah = target.__ajaxHooker;
|
|
if (ah && ah.proxyProps && prop in ah.proxyProps) {
|
|
const pDescriptor = ah.proxyProps[prop];
|
|
pDescriptor.set ? pDescriptor.set(value) : (pDescriptor.value = value);
|
|
} else {
|
|
target[prop] = value;
|
|
}
|
|
return true;
|
|
}
|
|
};
|
|
class XhrHooker {
|
|
constructor(xhr) {
|
|
const ah = this;
|
|
Object.assign(ah, {
|
|
originalXhr: xhr,
|
|
proxyXhr: new Proxy(xhr, proxyHandler),
|
|
resThenable: new SyncThenable(),
|
|
proxyProps: {},
|
|
proxyEvents: {}
|
|
});
|
|
xhr.addEventListener('readystatechange', e => {
|
|
if (ah.proxyXhr.readyState === 4 && ah.request && typeof ah.request.response === 'function') {
|
|
const response = {
|
|
finalUrl: ah.proxyXhr.responseURL,
|
|
status: ah.proxyXhr.status,
|
|
responseHeaders: parseHeaders(ah.proxyXhr.getAllResponseHeaders())
|
|
};
|
|
const tempValues = {};
|
|
for (const key of xhrResponses) {
|
|
try {
|
|
tempValues[key] = ah.originalXhr[key];
|
|
} catch (err) {}
|
|
defineProp(response, key, () => {
|
|
return response[key] = tempValues[key];
|
|
}, val => {
|
|
delete response[key];
|
|
response[key] = val;
|
|
});
|
|
}
|
|
ah.resThenable = new AHRequest(ah.request).waitForResponseKeys(response).then(() => {
|
|
for (const key of xhrResponses) {
|
|
ah.proxyProps[key] = {get: () => {
|
|
if (!(key in response)) response[key] = tempValues[key];
|
|
return response[key];
|
|
}};
|
|
}
|
|
});
|
|
}
|
|
ah.dispatchEvent(e);
|
|
});
|
|
xhr.addEventListener('load', e => ah.dispatchEvent(e));
|
|
xhr.addEventListener('loadend', e => ah.dispatchEvent(e));
|
|
for (const evt of xhrAsyncEvents) {
|
|
const onEvt = 'on' + evt;
|
|
ah.proxyProps[onEvt] = {
|
|
get: () => ah.proxyEvents[onEvt] || null,
|
|
set: val => ah.addEvent(onEvt, val)
|
|
};
|
|
}
|
|
for (const method of ['setRequestHeader', 'addEventListener', 'removeEventListener', 'open', 'send']) {
|
|
ah.proxyProps[method] = {value: ah[method]};
|
|
}
|
|
}
|
|
toJSON() {} // Converting circular structure to JSON
|
|
addEvent(type, event) {
|
|
if (type.startsWith('on')) {
|
|
this.proxyEvents[type] = typeof event === 'function' ? event : null;
|
|
} else {
|
|
if (typeof event === 'object' && event !== null) event = event.handleEvent;
|
|
if (typeof event !== 'function') return;
|
|
this.proxyEvents[type] = this.proxyEvents[type] || new Set();
|
|
this.proxyEvents[type].add(event);
|
|
}
|
|
}
|
|
removeEvent(type, event) {
|
|
if (type.startsWith('on')) {
|
|
this.proxyEvents[type] = null;
|
|
} else {
|
|
if (typeof event === 'object' && event !== null) event = event.handleEvent;
|
|
this.proxyEvents[type] && this.proxyEvents[type].delete(event);
|
|
}
|
|
}
|
|
dispatchEvent(e) {
|
|
e.stopImmediatePropagation = stopImmediatePropagation;
|
|
defineProp(e, 'target', () => this.proxyXhr);
|
|
defineProp(e, 'currentTarget', () => this.proxyXhr);
|
|
this.proxyEvents[e.type] && this.proxyEvents[e.type].forEach(fn => {
|
|
this.resThenable.then(() => !e.ajaxHooker_isStopped && fn.call(this.proxyXhr, e));
|
|
});
|
|
if (e.ajaxHooker_isStopped) return;
|
|
const onEvent = this.proxyEvents['on' + e.type];
|
|
onEvent && this.resThenable.then(onEvent.bind(this.proxyXhr, e));
|
|
}
|
|
setRequestHeader(header, value) {
|
|
this.originalXhr.setRequestHeader(header, value);
|
|
if (!this.request) return;
|
|
const headers = this.request.headers;
|
|
headers[header] = header in headers ? `${headers[header]}, ${value}` : value;
|
|
}
|
|
addEventListener(...args) {
|
|
if (xhrAsyncEvents.includes(args[0])) {
|
|
this.addEvent(args[0], args[1]);
|
|
} else {
|
|
this.originalXhr.addEventListener(...args);
|
|
}
|
|
}
|
|
removeEventListener(...args) {
|
|
if (xhrAsyncEvents.includes(args[0])) {
|
|
this.removeEvent(args[0], args[1]);
|
|
} else {
|
|
this.originalXhr.removeEventListener(...args);
|
|
}
|
|
}
|
|
open(method, url, async = true, ...args) {
|
|
this.request = {
|
|
type: 'xhr',
|
|
url: url.toString(),
|
|
method: method.toUpperCase(),
|
|
abort: false,
|
|
headers: {},
|
|
data: null,
|
|
response: null,
|
|
async: !!async
|
|
};
|
|
this.openArgs = args;
|
|
this.resThenable = new SyncThenable();
|
|
['responseURL', 'readyState', 'status', 'statusText', ...xhrResponses].forEach(key => {
|
|
delete this.proxyProps[key];
|
|
});
|
|
return this.originalXhr.open(method, url, async, ...args);
|
|
}
|
|
send(data) {
|
|
const ah = this;
|
|
const xhr = ah.originalXhr;
|
|
const request = ah.request;
|
|
if (!request) return xhr.send(data);
|
|
request.data = data;
|
|
new AHRequest(request).waitForRequestKeys().then(() => {
|
|
if (request.abort) {
|
|
if (typeof request.response === 'function') {
|
|
Object.assign(ah.proxyProps, {
|
|
responseURL: {value: request.url},
|
|
readyState: {value: 4},
|
|
status: {value: 200},
|
|
statusText: {value: 'OK'}
|
|
});
|
|
xhrAsyncEvents.forEach(evt => xhr.dispatchEvent(new Event(evt)));
|
|
}
|
|
} else {
|
|
xhr.open(request.method, request.url, request.async, ...ah.openArgs);
|
|
for (const header in request.headers) {
|
|
xhr.setRequestHeader(header, request.headers[header]);
|
|
}
|
|
xhr.send(request.data);
|
|
}
|
|
});
|
|
}
|
|
}
|
|
function fakeXHR() {
|
|
const xhr = new winAh.realXHR();
|
|
if ('__ajaxHooker' in xhr) console.warn('检测到不同版本的ajaxHooker，可能发生冲突！');
|
|
xhr.__ajaxHooker = new XhrHooker(xhr);
|
|
return xhr.__ajaxHooker.proxyXhr;
|
|
}
|
|
fakeXHR.prototype = win.XMLHttpRequest.prototype;
|
|
Object.keys(win.XMLHttpRequest).forEach(key => fakeXHR[key] = win.XMLHttpRequest[key]);
|
|
function fakeFetch(url, options = {}) {
|
|
if (!url) return winAh.realFetch.call(win, url, options);
|
|
return new Promise(async (resolve, reject) => {
|
|
const init = {};
|
|
if (getType(url) === '[object Request]') {
|
|
for (const prop of fetchInitProps) init[prop] = url[prop];
|
|
if (url.body) init.body = await url.arrayBuffer();
|
|
url = url.url;
|
|
}
|
|
url = url.toString();
|
|
Object.assign(init, options);
|
|
init.method = init.method || 'GET';
|
|
init.headers = init.headers || {};
|
|
const request = {
|
|
type: 'fetch',
|
|
url: url,
|
|
method: init.method.toUpperCase(),
|
|
abort: false,
|
|
headers: parseHeaders(init.headers),
|
|
data: init.body,
|
|
response: null,
|
|
async: true
|
|
};
|
|
const req = new AHRequest(request);
|
|
await req.waitForRequestKeys();
|
|
if (request.abort) {
|
|
if (typeof request.response === 'function') {
|
|
const response = {
|
|
finalUrl: request.url,
|
|
status: 200,
|
|
responseHeaders: {}
|
|
};
|
|
await req.waitForResponseKeys(response);
|
|
const key = fetchResponses.find(k => k in response);
|
|
let val = response[key];
|
|
if (key === 'json' && typeof val === 'object') {
|
|
val = catchError(JSON.stringify.bind(JSON), val);
|
|
}
|
|
const res = new Response(val, {
|
|
status: 200,
|
|
statusText: 'OK'
|
|
});
|
|
defineProp(res, 'type', () => 'basic');
|
|
defineProp(res, 'url', () => request.url);
|
|
resolve(res);
|
|
} else {
|
|
reject(new DOMException('aborted', 'AbortError'));
|
|
}
|
|
return;
|
|
}
|
|
init.method = request.method;
|
|
init.headers = request.headers;
|
|
init.body = request.data;
|
|
winAh.realFetch.call(win, request.url, init).then(res => {
|
|
if (typeof request.response === 'function') {
|
|
const response = {
|
|
finalUrl: res.url,
|
|
status: res.status,
|
|
responseHeaders: parseHeaders(res.headers)
|
|
};
|
|
fetchResponses.forEach(key => res[key] = function() {
|
|
if (key in response) return Promise.resolve(response[key]);
|
|
return resProto[key].call(this).then(val => {
|
|
response[key] = val;
|
|
return req.waitForResponseKeys(response).then(() => key in response ? response[key] : val);
|
|
});
|
|
});
|
|
}
|
|
resolve(res);
|
|
}, reject);
|
|
});
|
|
}
|
|
function fakeFetchClone() {
|
|
const descriptors = Object.getOwnPropertyDescriptors(this);
|
|
const res = winAh.realFetchClone.call(this);
|
|
Object.defineProperties(res, descriptors);
|
|
return res;
|
|
}
|
|
winAh = win.__ajaxHooker = winAh || {
|
|
version, fakeXHR, fakeFetch, fakeFetchClone,
|
|
realXHR: win.XMLHttpRequest,
|
|
realFetch: win.fetch,
|
|
realFetchClone: resProto.clone,
|
|
hookInsts: new Set()
|
|
};
|
|
if (winAh.version !== version) console.warn('检测到不同版本的ajaxHooker，可能发生冲突！');
|
|
win.XMLHttpRequest = winAh.fakeXHR;
|
|
win.fetch = winAh.fakeFetch;
|
|
resProto.clone = winAh.fakeFetchClone;
|
|
winAh.hookInsts.add(hookInst);
|
|
return {
|
|
hook: fn => hookInst.hookFns.push(fn),
|
|
filter: arr => {
|
|
if (Array.isArray(arr)) hookInst.filters = arr;
|
|
},
|
|
protect: () => {
|
|
readonly(win, 'XMLHttpRequest', winAh.fakeXHR);
|
|
readonly(win, 'fetch', winAh.fakeFetch);
|
|
readonly(resProto, 'clone', winAh.fakeFetchClone);
|
|
},
|
|
unhook: () => {
|
|
winAh.hookInsts.delete(hookInst);
|
|
if (!winAh.hookInsts.size) {
|
|
writable(win, 'XMLHttpRequest', winAh.realXHR);
|
|
writable(win, 'fetch', winAh.realFetch);
|
|
writable(resProto, 'clone', winAh.realFetchClone);
|
|
delete win.__ajaxHooker;
|
|
}
|
|
}
|
|
};
|
|
}(); |