mirror of
https://github.com/BTMuli/TeyvatGuide.git
synced 2025-12-12 09:18:14 +08:00
761 lines
23 KiB
TypeScript
761 lines
23 KiB
TypeScript
/**
|
||
* @file utils/TGClient.ts
|
||
* @desc 负责米游社客户端的 callback 处理
|
||
* @since Beta v0.3.8
|
||
*/
|
||
|
||
import { event, invoke } from "@tauri-apps/api";
|
||
import type { Event } from "@tauri-apps/api/event";
|
||
import { appWindow, WebviewWindow } from "@tauri-apps/api/window";
|
||
|
||
import { parseLink } from "./linkParser";
|
||
import { createPost } from "./TGWindow";
|
||
import { getDeviceInfo } from "./toolFunc";
|
||
import showSnackbar from "../components/func/snackbar";
|
||
import { useAppStore } from "../store/modules/app";
|
||
import { useUserStore } from "../store/modules/user";
|
||
import TGConstant from "../web/constant/TGConstant";
|
||
import TGRequest from "../web/request/TGRequest";
|
||
import { getDS4JS } from "../web/utils/getRequestHeader";
|
||
|
||
// 正常 arg 参数
|
||
interface NormalArg {
|
||
method: string;
|
||
payload: any;
|
||
callback: string;
|
||
}
|
||
|
||
// invoke 参数
|
||
interface InvokeArg {
|
||
func: string;
|
||
}
|
||
|
||
/**
|
||
* @class TGClient
|
||
* @since Beta v0.3.8
|
||
* @description 米游社客户端
|
||
*/
|
||
class TGClient {
|
||
/**
|
||
* @private 监听实例
|
||
* @since Beta v0.3.3
|
||
* @type {EventEmitter}
|
||
* @memberof TGClient
|
||
*/
|
||
private listener: any;
|
||
|
||
/**
|
||
* @private 窗口实例
|
||
* @since Beta v0.3.4
|
||
* @type {WebviewWindow}
|
||
* @memberof TGClient
|
||
*/
|
||
private window: WebviewWindow | null;
|
||
|
||
/**
|
||
* @private 模拟路由
|
||
* @since Beta v0.3.4
|
||
* @type {string[]}
|
||
* @memberof TGClient
|
||
*/
|
||
private route: string[] = [];
|
||
|
||
/**
|
||
* @constructor
|
||
* @since Beta v0.3.4
|
||
* @description 构造函数
|
||
* @memberof TGClient
|
||
*/
|
||
constructor() {
|
||
try {
|
||
this.window = WebviewWindow.getByLabel("mhy_client");
|
||
} catch (error) {
|
||
this.window = null;
|
||
}
|
||
this.route = [];
|
||
this.listener = undefined;
|
||
}
|
||
|
||
/**
|
||
* @func run
|
||
* @since Beta v0.3.4
|
||
* @desc 运行米游社客户端
|
||
* @returns {void} - 无返回值
|
||
*/
|
||
async run(): Promise<void> {
|
||
if (this.listener === undefined) {
|
||
this.listener = await event.listen("post_mhy_client", async (arg: Event<string>) => {
|
||
await this.handleCallback(arg);
|
||
});
|
||
}
|
||
}
|
||
|
||
/**
|
||
* @func loadJSBridge
|
||
* @since Beta v0.3.8
|
||
* @desc 加载 JSBridge
|
||
* @returns {void} - 无返回值
|
||
*/
|
||
async loadJSBridge(): Promise<void> {
|
||
const executeJS = `javascript:(function() {
|
||
if(window.MiHoYoJSInterface) return;
|
||
window.MiHoYoJSInterface = {
|
||
postMessage: function(arg) { window.__TAURI__.event.emit('post_mhy_client', arg) },
|
||
closePage: function() { this.postMessage('{"method":"closePage"}') },
|
||
};
|
||
})();`;
|
||
await invoke("execute_js", { label: "mhy_client", js: executeJS });
|
||
}
|
||
|
||
/**
|
||
* @func hideSideBar
|
||
* @since Beta v0.3.5
|
||
* @desc 隐藏侧边栏
|
||
* @returns {void} - 无返回值
|
||
*/
|
||
async hideSideBar(): Promise<void> {
|
||
const executeJS = `javascript:(function(){
|
||
if (document.getElementById('teyvat_style')) {
|
||
return;
|
||
} else {
|
||
let style = document.createElement('style');
|
||
style.innerHTML = '::-webkit-scrollbar{display:none}';
|
||
style.id = 'teyvat_style';
|
||
document.querySelector('body').appendChild(style);
|
||
}
|
||
})();`;
|
||
await invoke("execute_js", { label: "mhy_client", js: executeJS });
|
||
}
|
||
|
||
/**
|
||
* @func hideOverlay
|
||
* @since Beta v0.3.7
|
||
* @desc 隐藏遮罩
|
||
* @returns {void}
|
||
*/
|
||
async hideOverlay(): Promise<void> {
|
||
const executeJS = `javascript:(function(){
|
||
if (document.getElementById('mihoyo_landscape') !== null) {
|
||
let box = document.getElementById('mihoyo_landscape');
|
||
box.remove();
|
||
}
|
||
})();`;
|
||
await invoke("execute_js", { label: "mhy_client", js: executeJS });
|
||
}
|
||
|
||
/**
|
||
* @func getUrl
|
||
* @since Beta v0.3.8
|
||
* @desc 获取 url
|
||
* @param {string} func - 方法名
|
||
* @returns {string} - url
|
||
*/
|
||
getUrl(func: string): string {
|
||
switch (func) {
|
||
case "sign_in":
|
||
return "https://act.mihoyo.com/bbs/event/signin/hk4e/index.html?act_id=e202311201442471&bbs_auth_required=true&bbs_presentation_style=fullscreen&mhy_presentation_style=fullscreen&utm_source=bbs&utm_medium=ys&utm_campaign=icon";
|
||
case "game_record":
|
||
return "https://webstatic.mihoyo.com/app/community-game-records/index.html?bbs_presentation_style=fullscreen";
|
||
case "daily_note":
|
||
return "https://webstatic.mihoyo.com/app/community-game-records/index.html?bbs_presentation_style=fullscreen#/ys/daily/";
|
||
case "tavern":
|
||
return "https://m.miyoushe.com/ys/#/home/26";
|
||
case "birthday":
|
||
return "https://webstatic.mihoyo.com/ys/event/e20220303-birthday/index.html?activity_id=20220301153521";
|
||
case "toolbox":
|
||
return "https://webstatic.mihoyo.com/bbs/event/e20200511toolbox/index.html?game_biz=ys_cn";
|
||
default:
|
||
return "https://api-static.mihoyo.com/";
|
||
}
|
||
}
|
||
|
||
/**
|
||
* @func open
|
||
* @since Beta v0.3.7
|
||
* @desc 打开米游社客户端
|
||
* @param {string} func - 方法名
|
||
* @param {string} url - url
|
||
* @returns {void} - 无返回值
|
||
*/
|
||
async open(func: string, url?: string): Promise<void> {
|
||
if (this.window !== null) {
|
||
try {
|
||
await this.window.close();
|
||
} catch (e) {
|
||
console.error(e);
|
||
await invoke<InvokeArg>("create_mhy_client", {
|
||
func: "default",
|
||
url: "https://api-static.mihoyo.com/",
|
||
});
|
||
await this.open(func, url);
|
||
}
|
||
}
|
||
if (url === undefined) url = this.getUrl(func);
|
||
this.route = [url];
|
||
console.log(`[open] ${url}`);
|
||
await invoke<InvokeArg>("create_mhy_client", { func, url });
|
||
this.window = WebviewWindow.getByLabel("mhy_client");
|
||
await this.window?.show();
|
||
await this.window?.setFocus();
|
||
await this.loadJSBridge();
|
||
}
|
||
|
||
/**
|
||
* @func handleCustomCallback
|
||
* @since Beta v0.3.8
|
||
* @desc 处理自定义的 callback
|
||
* @param {Event<string>} arg - 事件参数
|
||
* @returns {any} - 返回值
|
||
*/
|
||
async handleCustomCallback(arg: Event<string>): Promise<any> {
|
||
const { method, payload } = <NormalArg>JSON.parse(arg.payload);
|
||
switch (method) {
|
||
case "teyvat_open":
|
||
createPost(payload);
|
||
break;
|
||
default:
|
||
console.warn(`[${arg.windowLabel}] ${arg.payload}`);
|
||
}
|
||
}
|
||
|
||
/**
|
||
* @func handleCallback
|
||
* @since Beta v0.3.8
|
||
* @desc 处理米游社客户端的 callback
|
||
* @param {Event<string>} arg - 事件参数
|
||
* @returns {any} - 返回值
|
||
*/
|
||
async handleCallback(arg: Event<string>): Promise<any> {
|
||
const { method, payload, callback } = <NormalArg>JSON.parse(arg.payload);
|
||
if (method.startsWith("teyvat")) {
|
||
await this.handleCustomCallback(arg);
|
||
return;
|
||
}
|
||
console.log(`[${arg.windowLabel}] ${arg.payload}`);
|
||
await this.hideSideBar();
|
||
await this.hideOverlay();
|
||
switch (method) {
|
||
case "closePage":
|
||
await this.closePage();
|
||
break;
|
||
case "configure_share":
|
||
break;
|
||
case "eventTrack":
|
||
await this.eventTrack(payload);
|
||
break;
|
||
case "getStatusBarHeight":
|
||
await this.getStatusBarHeight(callback);
|
||
break;
|
||
case "genAuthKey":
|
||
await this.genAuthKey(payload, callback);
|
||
break;
|
||
case "getCookieInfo":
|
||
await this.getCookieInfo(payload, callback);
|
||
break;
|
||
case "getCookieToken":
|
||
await this.getCookieToken(callback);
|
||
break;
|
||
case "getActionTicket":
|
||
await this.getActionTicket(payload, callback);
|
||
break;
|
||
case "getHTTPRequestHeaders":
|
||
await this.getHTTPRequestHeaders(callback);
|
||
break;
|
||
case "getDS":
|
||
await this.getDS(1, callback, payload);
|
||
break;
|
||
case "getDS2":
|
||
await this.getDS(2, callback, payload);
|
||
break;
|
||
case "getUserInfo":
|
||
await this.getUserInfo(callback);
|
||
break;
|
||
case "login":
|
||
await this.nullCallback(arg);
|
||
break;
|
||
case "onBeginDragging":
|
||
await this.nullCallback(arg);
|
||
break;
|
||
case "onClickImg":
|
||
await this.onClickImg(payload);
|
||
break;
|
||
case "openApplication":
|
||
await this.openApplication(payload);
|
||
break;
|
||
case "pushPage":
|
||
await this.pushPage(payload);
|
||
break;
|
||
case "share":
|
||
await this.share(payload, callback);
|
||
break;
|
||
case "share2":
|
||
await this.nullCallback(arg);
|
||
break;
|
||
default:
|
||
console.warn(`[${arg.windowLabel}] ${arg.payload}`);
|
||
}
|
||
}
|
||
|
||
/**
|
||
* @func callback
|
||
* @since Beta v0.3.4
|
||
* @desc 回调函数
|
||
* @param {string} callback - 回调函数名
|
||
* @param {object} data - 回调数据
|
||
* @returns {void} - 无返回值
|
||
*/
|
||
async callback(callback: string, data: object): Promise<void> {
|
||
const response = {
|
||
retcode: 0,
|
||
message: "success",
|
||
data: data ?? {},
|
||
};
|
||
const js = `javascript:mhyWebBridge("${callback}", ${JSON.stringify(response)});`;
|
||
console.info(`[callback] ${js}`);
|
||
await invoke("create_mhy_client", { func: "execute_js", url: "" });
|
||
await invoke("execute_js", { label: "mhy_client", js });
|
||
}
|
||
|
||
/**
|
||
* @func getStatusBarHeight
|
||
* @since Beta v0.3.4
|
||
* @desc 获取状态栏高度
|
||
* @param {string} callback - 回调函数名
|
||
* @returns {void} - 无返回值
|
||
*/
|
||
async getStatusBarHeight(callback: string): Promise<void> {
|
||
const data = {
|
||
statusBarHeight: 0,
|
||
};
|
||
await this.callback(callback, data);
|
||
}
|
||
|
||
/**
|
||
* @func genAuthKey
|
||
* @since Beta v0.3.7
|
||
* @desc 获取米游社客户端的 authkey
|
||
* @param {Record<string, string>} payload - 请求参数
|
||
* @param {string} callback - 回调函数名
|
||
* @returns {void} - 无返回值
|
||
*/
|
||
async genAuthKey(payload: Record<string, string>, callback: string): Promise<void> {
|
||
const userStore = useUserStore();
|
||
const cookie = {
|
||
mid: userStore.cookie.mid,
|
||
stoken: userStore.cookie.stoken,
|
||
};
|
||
const res = await TGRequest.User.getAuthkey2(cookie, payload);
|
||
await this.callback(callback, res.data);
|
||
}
|
||
|
||
/**
|
||
* @func getCookieInfo
|
||
* @since Beta v0.3.4
|
||
* @desc 获取米游社客户端的 cookie
|
||
* @param {unknown} payload - 请求参数
|
||
* @param {string} callback - 回调函数名
|
||
* @returns {void} - 无返回值
|
||
*/
|
||
async getCookieInfo(payload: unknown, callback: string): Promise<void> {
|
||
const user = useUserStore();
|
||
const data = {
|
||
ltoken: user.cookie.ltoken,
|
||
ltuid: user.cookie.ltuid,
|
||
login_ticket: "",
|
||
};
|
||
await this.callback(callback, data);
|
||
}
|
||
|
||
/**
|
||
* @func getCookieToken
|
||
* @since Beta v0.3.4
|
||
* @todo 待完善
|
||
* @desc 获取米游社客户端的 cookie_token
|
||
* @param {string} callback - 回调函数名
|
||
* @returns {void} - 无返回值
|
||
*/
|
||
async getCookieToken(callback: string): Promise<void> {
|
||
const user = useUserStore();
|
||
const executeJS =
|
||
"javascript:(function(){" +
|
||
`document.cookie = "account_id=${user.cookie.account_id};domain=.mihoyo.com;path=/;expires=Fri, 31 Dec 9999 23:59:59 GMT;";` +
|
||
`document.cookie = "cookie_token=${user.cookie.cookie_token};domain=.mihoyo.com;path=/;expires=Fri, 31 Dec 9999 23:59:59 GMT;";` +
|
||
`document.cookie = "ltoken=${user.cookie.ltoken};domain=.mihoyo.com;path=/;expires=Fri, 31 Dec 9999 23:59:59 GMT;";` +
|
||
`document.cookie = "ltuid=${user.cookie.ltuid};domain=.mihoyo.com;path=/;expires=Fri, 31 Dec 9999 23:59:59 GMT;";` +
|
||
`document.cookie = "stuid=${user.cookie.stuid};domain=.mihoyo.com;path=/;expires=Fri, 31 Dec 9999 23:59:59 GMT;";` +
|
||
`document.cookie = "stoken=${user.cookie.stoken};domain=.mihoyo.com;path=/;expires=Fri, 31 Dec 9999 23:59:59 GMT;";` +
|
||
`document.cookie = "mid=${user.cookie.mid};domain=.mihoyo.com;path=/;expires=Fri, 31 Dec 9999 23:59:59 GMT;";` +
|
||
"})();";
|
||
console.info(`[getCookieToken] ${executeJS}`);
|
||
await invoke("execute_js", { label: "mhy_client", js: executeJS });
|
||
// callback
|
||
const data = {
|
||
cookie_token: user.cookie.cookie_token,
|
||
};
|
||
await this.callback(callback, data);
|
||
}
|
||
|
||
/**
|
||
* @func getActionTicket
|
||
* @since Beta v0.3.4
|
||
* @desc 获取米游社客户端的 action_ticket\
|
||
* @param {unknown} payload - 请求参数
|
||
* @param {string} callback - 回调函数名
|
||
* @returns {void} - 无返回值
|
||
*/
|
||
async getActionTicket(payload: any, callback: string): Promise<void> {
|
||
const actionType = payload.action_type;
|
||
const user = useUserStore();
|
||
const uid = user.getCurAccount().gameUid;
|
||
const mid = user.cookie.mid;
|
||
const stoken = user.cookie.stoken;
|
||
const ActionTicket = await TGRequest.User.bySToken.getActionTicket(
|
||
actionType,
|
||
stoken,
|
||
mid,
|
||
uid,
|
||
);
|
||
await this.callback(callback, ActionTicket.data);
|
||
}
|
||
|
||
/**
|
||
* @func getHTTPRequestHeaders
|
||
* @since Beta v0.3.6
|
||
* @desc 获取米游社客户端的 HTTP 请求头
|
||
* @param {string} callback - 回调函数名
|
||
* @returns {void} - 无返回值
|
||
*/
|
||
async getHTTPRequestHeaders(callback: string): Promise<void> {
|
||
const localFp = getDeviceInfo("device_fp");
|
||
let deviceInfo = useAppStore().deviceInfo;
|
||
if (localFp === "0000000000000") {
|
||
deviceInfo = await TGRequest.Device.getFp(deviceInfo);
|
||
}
|
||
const data = {
|
||
"user-agent": TGConstant.BBS.UA_MOBILE,
|
||
"x-rpc-client_type": "5",
|
||
"x-rpc-device_id": deviceInfo.device_id,
|
||
"x-rpc-app_version": TGConstant.BBS.VERSION,
|
||
"x-rpc-device_fp": deviceInfo.device_fp,
|
||
};
|
||
await this.callback(callback, data);
|
||
}
|
||
|
||
/**
|
||
* @func getDS
|
||
* @since Beta v0.3.4
|
||
* @desc 获取米游社客户端的 DS 参数
|
||
* @param {number} dsType - DS 类型
|
||
* @param {string} callback - 回调函数名
|
||
* @param {unknown} payload - 请求参数
|
||
* @returns {void} - 无返回值
|
||
*/
|
||
async getDS(dsType: 1, callback: string, payload: undefined): Promise<void>;
|
||
async getDS(dsType: 2, callback: string, payload: any): Promise<void>;
|
||
async getDS(dsType: 1 | 2, callback: string, payload?: any): Promise<void> {
|
||
const saltType = dsType === 1 ? "lk2" : "common";
|
||
let ds: string;
|
||
if (dsType === 2) {
|
||
const { body, query } = payload;
|
||
ds = getDS4JS(saltType, dsType, body, query);
|
||
} else {
|
||
ds = getDS4JS(saltType, dsType, undefined, undefined);
|
||
}
|
||
const data = {
|
||
DS: ds,
|
||
};
|
||
await this.callback(callback, data);
|
||
}
|
||
|
||
/**
|
||
* @func getUserInfo
|
||
* @since Beta v0.3.4
|
||
* @desc 获取米游社客户端的用户信息
|
||
* @param {string} callback - 回调函数名
|
||
* @returns {void} - 无返回值
|
||
*/
|
||
async getUserInfo(callback: string): Promise<void> {
|
||
const user = useUserStore();
|
||
const cookieToken = user.cookie.cookie_token;
|
||
const accountId = user.cookie.account_id;
|
||
const userInfo = await TGRequest.User.byCookie.getUserInfo(cookieToken, accountId);
|
||
if ("retcode" in userInfo) {
|
||
console.error(`[${callback}] ${userInfo.message}`);
|
||
return;
|
||
}
|
||
await this.callback(callback, userInfo);
|
||
}
|
||
|
||
/**
|
||
* @func pushPage
|
||
* @since Beta v0.3.8
|
||
* @desc 打开米游社客户端的页面
|
||
* @param {unknown} payload - 请求参数
|
||
* @returns {void} - 无返回值
|
||
*/
|
||
async pushPage(payload: any): Promise<void> {
|
||
const url = payload.page;
|
||
const res = await parseLink(url, true);
|
||
if (!res) {
|
||
await appWindow.setFocus();
|
||
showSnackbar({
|
||
text: `未知链接:${url}`,
|
||
color: "error",
|
||
timeout: 3000,
|
||
});
|
||
await new Promise<void>((resolve) => {
|
||
setTimeout(() => {
|
||
resolve();
|
||
}, 3000);
|
||
});
|
||
await this.window?.setFocus();
|
||
return;
|
||
}
|
||
if (typeof res !== "string") return;
|
||
this.route.push(res);
|
||
console.log(`[pushPage] ${res}`);
|
||
const executeJS = `javascript:(function(){
|
||
window.location.href = '${res}';
|
||
})();`;
|
||
await invoke("execute_js", { label: "mhy_client", js: executeJS });
|
||
await this.loadJSBridge();
|
||
await this.hideSideBar();
|
||
await this.hideOverlay();
|
||
await this.window?.setFocus();
|
||
}
|
||
|
||
/**
|
||
* @func closePage
|
||
* @since Beta v0.3.7
|
||
* @desc 关闭米游社客户端的页面
|
||
* @returns {void} - 无返回值
|
||
*/
|
||
async closePage(): Promise<void> {
|
||
this.route.pop();
|
||
if (this.route.length === 0) {
|
||
await this.window?.close();
|
||
return;
|
||
}
|
||
const url = this.route[this.route.length - 1];
|
||
const executeJS = `javascript:(function(){
|
||
window.location.href = '${url}';
|
||
})();`;
|
||
await invoke("execute_js", { label: "mhy_client", js: executeJS });
|
||
await this.loadJSBridge();
|
||
}
|
||
|
||
/**
|
||
* @func nullCallback
|
||
* @since Beta v0.3.4
|
||
* @desc 空回调函数
|
||
* @param {Event<string>} arg - 回调参数
|
||
* @returns {void} - 无返回值
|
||
*/
|
||
async nullCallback(arg: Event<string>): Promise<void> {
|
||
const { callback } = <NormalArg>JSON.parse(arg.payload);
|
||
await this.callback(callback, {});
|
||
}
|
||
|
||
/**
|
||
* @func onClickImg
|
||
* @since Beta v0.3.7
|
||
* @desc 点击图片，下载到本地
|
||
* @param {unknown} payload - 请求参数
|
||
* @returns {void} - 无返回值
|
||
*/
|
||
async onClickImg(payload: any): Promise<void> {
|
||
const image = payload.image_list[0];
|
||
const executeJS = this.getSaveImgJS(image.url, image.format);
|
||
await invoke("execute_js", { label: "mhy_client", js: executeJS });
|
||
}
|
||
|
||
/**
|
||
* @func getSaveImgJS
|
||
* @since Beta v0.3.7
|
||
* @desc 获取保存图片的 JS
|
||
* @param {string} url - 图片链接
|
||
* @param {string} format - 图片格式
|
||
* @returns {string} - JS
|
||
*/
|
||
getSaveImgJS(url: string, format: string): string {
|
||
return `javascript:(async function() {
|
||
const _t = window.__TAURI__;
|
||
const defaultPath = await _t.path.downloadDir() + Date.now() + '.${format}';
|
||
const savePath = await _t.dialog.save({
|
||
title: '保存图片',
|
||
filters: [{ name: '图片', extensions: ['png'] }],
|
||
defaultPath: defaultPath,
|
||
});
|
||
if (savePath) {
|
||
const resBlob = await _t.http.fetch('${url}',{
|
||
method: 'GET',
|
||
responseType: _t.http.ResponseType.Binary
|
||
});
|
||
const buffer = new Uint8Array(resBlob.data);
|
||
await _t.fs.writeBinaryFile({
|
||
contents: buffer,
|
||
path: savePath,
|
||
});
|
||
alert('保存成功');
|
||
}
|
||
})();`;
|
||
}
|
||
|
||
/**
|
||
* @func share
|
||
* @since Beta v0.3.8
|
||
* @desc 分享
|
||
* @param {unknown} payload - 请求参数
|
||
* @param {string} callback - 回调函数名
|
||
* @returns {void} - 无返回值
|
||
*/
|
||
async share(payload: any, callback: string): Promise<void> {
|
||
// 如果有数据
|
||
if (payload?.content.image_url !== undefined) {
|
||
const image = payload.content.image_url;
|
||
const format = image.split(".").pop();
|
||
const executeJS = this.getSaveImgJS(image, format);
|
||
await invoke("execute_js", { label: "mhy_client", js: executeJS });
|
||
await this.callback(callback, {});
|
||
return;
|
||
}
|
||
if (payload?.type === "screenshot") {
|
||
const executeJS = `javascript:(async function() {
|
||
// 查找 id 为 dom2img 的 script
|
||
var hasDom2img = false;
|
||
var scripts = document.querySelectorAll('script');
|
||
for (var i = 0; i < scripts.length; i++) {
|
||
if (scripts[i].src === 'https://cdn.bootcdn.net/ajax/libs/dom-to-image/2.6.0/dom-to-image.min.js') {
|
||
hasDom2img = true;
|
||
break;
|
||
}
|
||
}
|
||
// 如果没有 dom2img 的 js
|
||
if (!hasDom2img) {
|
||
// 添加 dom2img 的 js
|
||
var script = document.createElement('script');
|
||
script.src = 'https://cdn.bootcdn.net/ajax/libs/dom-to-image/2.6.0/dom-to-image.min.js';
|
||
document.body.appendChild(script);
|
||
// 等待 dom2img 加载完成
|
||
await new Promise((resolve) => {
|
||
script.onload = function() {
|
||
resolve();
|
||
}
|
||
});
|
||
}
|
||
var shareDom;
|
||
if(${JSON.stringify(payload?.content)} === '{}') {
|
||
shareDom = document.querySelector('.share');
|
||
} else if (${payload?.content?.preview} === true) {
|
||
shareDom = document.querySelector('#root');
|
||
}
|
||
if(shareDom === undefined || shareDom === null) {
|
||
shareDom = document.body;
|
||
}
|
||
var scale = 1.5;
|
||
var img = await domtoimage.toPng(shareDom, {
|
||
height: shareDom.scrollHeight * scale,
|
||
width: shareDom.scrollWidth * scale,
|
||
style: {
|
||
transform: 'scale(' + scale + ')',
|
||
transformOrigin: 'top left'
|
||
}
|
||
});
|
||
// 转换成 blob
|
||
var buffer = new Uint8Array(atob(img.split(',')[1]).split('').map(function(item) {
|
||
return item.charCodeAt(0);
|
||
}));
|
||
var _t = window.__TAURI__;
|
||
var savePath = await _t.path.downloadDir() + Date.now() + '.png';
|
||
var save = await _t.dialog.save({
|
||
title: '保存图片',
|
||
filters: [{ name: '图片', extensions: ['png'] }],
|
||
defaultPath: savePath,
|
||
});
|
||
if (save) {
|
||
await _t.fs.writeBinaryFile({
|
||
contents: buffer,
|
||
path: save,
|
||
});
|
||
alert('保存成功');
|
||
}
|
||
mhyWebBridge('${callback}', {});
|
||
})();`;
|
||
await invoke("execute_js", { label: "mhy_client", js: executeJS });
|
||
return;
|
||
}
|
||
if (payload?.type === "image") {
|
||
if (payload?.content.image_base64 !== undefined) {
|
||
let image = payload.content.image_base64;
|
||
image = `data:image/png;base64,${image}`;
|
||
const executeJS = `javascript:(async function() {
|
||
// 转换成 blob
|
||
var buffer = new Uint8Array(atob('${image}'.split(',')[1]).split('').map(function(item) {
|
||
return item.charCodeAt(0);
|
||
}));
|
||
var _t = window.__TAURI__;
|
||
var savePath = await _t.path.downloadDir() + Date.now() + '.png';
|
||
var save = await _t.dialog.save({
|
||
title: '保存图片',
|
||
filters: [{ name: '图片', extensions: ['png'] }],
|
||
defaultPath: savePath,
|
||
});
|
||
if (save) {
|
||
await _t.fs.writeBinaryFile({
|
||
contents: buffer,
|
||
path: save,
|
||
});
|
||
alert('保存成功');
|
||
}
|
||
mhyWebBridge('${callback}', {});
|
||
})();`;
|
||
await invoke("execute_js", { label: "mhy_client", js: executeJS });
|
||
return;
|
||
}
|
||
}
|
||
console.warn("[share]", payload);
|
||
// 延时 3s
|
||
setTimeout(async () => {
|
||
await this.callback(callback, {});
|
||
}, 10000);
|
||
}
|
||
|
||
/**
|
||
* @func eventTrack
|
||
* @since Beta v0.3.8
|
||
* @desc 事件跟踪
|
||
* @param {unknown} payload - 请求参数
|
||
* @returns {void} - 无返回值
|
||
*/
|
||
async eventTrack(payload: unknown): Promise<void> {
|
||
console.log(`[eventTrack] ${JSON.stringify(payload)}`);
|
||
await this.loadJSBridge();
|
||
}
|
||
|
||
/**
|
||
* @func openApplication
|
||
* @since Beta v0.3.8
|
||
* @desc 打开应用
|
||
* @param {unknown} payload - 请求参数
|
||
* @returns {void} - 无返回值
|
||
*/
|
||
async openApplication(payload: unknown): Promise<void> {
|
||
console.log(`[openApplication] ${JSON.stringify(payload)}`);
|
||
await appWindow.setFocus();
|
||
showSnackbar({
|
||
text: `不支持的操作：OpenApplication(${JSON.stringify(payload)})`,
|
||
color: "error",
|
||
});
|
||
await new Promise<void>((resolve) => {
|
||
setTimeout(() => {
|
||
resolve();
|
||
}, 1500);
|
||
});
|
||
await this.window?.setFocus();
|
||
}
|
||
}
|
||
|
||
const mhyClient = new TGClient();
|
||
|
||
export default mhyClient;
|