103 lines
2.7 KiB
JavaScript
103 lines
2.7 KiB
JavaScript
import { fetchGM } from "./fetch";
|
||
import { genEventName } from "./utils";
|
||
|
||
const MSG_GM_xmlHttpRequest = "xmlHttpRequest";
|
||
const MSG_GM_setValue = "setValue";
|
||
const MSG_GM_getValue = "getValue";
|
||
const MSG_GM_deleteValue = "deleteValue";
|
||
const MSG_GM_info = "info";
|
||
|
||
/**
|
||
* 注入页面的脚本，请求并接受GM接口信息
|
||
* @param {*} param0
|
||
*/
|
||
export const injectScript = (ping) => {
|
||
window.APP_INFO = {
|
||
name: process.env.REACT_APP_NAME,
|
||
version: process.env.REACT_APP_VERSION,
|
||
eventName: ping,
|
||
};
|
||
};
|
||
|
||
/**
|
||
* 适配GM脚本
|
||
*/
|
||
export const adaptScript = (ping) => {
|
||
const promiseGM = (action, args, timeout = 5000) =>
|
||
new Promise((resolve, reject) => {
|
||
const pong = genEventName();
|
||
const handleEvent = (e) => {
|
||
window.removeEventListener(pong, handleEvent);
|
||
const { data, error } = e.detail;
|
||
if (error) {
|
||
reject(new Error(error));
|
||
} else {
|
||
resolve(data);
|
||
}
|
||
};
|
||
|
||
window.addEventListener(pong, handleEvent);
|
||
window.dispatchEvent(
|
||
new CustomEvent(ping, { detail: { action, args, pong } })
|
||
);
|
||
|
||
setTimeout(() => {
|
||
window.removeEventListener(pong, handleEvent);
|
||
reject(new Error("timeout"));
|
||
}, timeout);
|
||
});
|
||
|
||
window.KISS_GM = {
|
||
fetch: (input, init) => promiseGM(MSG_GM_xmlHttpRequest, { input, init }),
|
||
setValue: (key, val) => promiseGM(MSG_GM_setValue, { key, val }),
|
||
getValue: (key) => promiseGM(MSG_GM_getValue, { key }),
|
||
deleteValue: (key) => promiseGM(MSG_GM_deleteValue, { key }),
|
||
getInfo: async () => {
|
||
if (!window.GM_info) {
|
||
window.GM_info = await promiseGM(MSG_GM_info);
|
||
}
|
||
return window.GM_info;
|
||
},
|
||
};
|
||
};
|
||
|
||
/**
|
||
* 监听并回应页面对GM接口的请求
|
||
* @param {*} param0
|
||
*/
|
||
export const handlePing = async (e) => {
|
||
const { action, args, pong } = e.detail;
|
||
let res;
|
||
try {
|
||
switch (action) {
|
||
case MSG_GM_xmlHttpRequest:
|
||
const { input, init } = args;
|
||
res = await fetchGM(input, init);
|
||
break;
|
||
case MSG_GM_setValue:
|
||
const { key, val } = args;
|
||
await GM.setValue(key, val);
|
||
res = val;
|
||
break;
|
||
case MSG_GM_getValue:
|
||
res = await GM.getValue(args.key);
|
||
break;
|
||
case MSG_GM_deleteValue:
|
||
await GM.deleteValue(args.key);
|
||
res = "ok";
|
||
break;
|
||
case MSG_GM_info:
|
||
res = GM.info;
|
||
break;
|
||
default:
|
||
throw new Error(`message action is unavailable: ${action}`);
|
||
}
|
||
|
||
window.dispatchEvent(new CustomEvent(pong, { detail: { data: res } }));
|
||
} catch (err) {
|
||
window.dispatchEvent(
|
||
new CustomEvent(pong, { detail: { error: err.message } })
|
||
);
|
||
}
|
||
};
|