add more translators: baidu/tencent/deeplfree

This commit is contained in:
Gabe Yuan
2023-10-21 11:54:04 +08:00
parent 507d54dba0
commit 160fc218fc
9 changed files with 340 additions and 550 deletions

View File

@@ -1,4 +1,3 @@
import queryString from "query-string";
import { isExt, isGm } from "./client";
import { sendBgMsg } from "./msg";
import { taskPool } from "./pool";
@@ -7,23 +6,11 @@ import {
MSG_FETCH_LIMIT,
MSG_FETCH_CLEAR,
CACHE_NAME,
OPT_TRANS_GOOGLE,
OPT_TRANS_MICROSOFT,
OPT_TRANS_DEEPL,
OPT_TRANS_DEEPLFREE,
OPT_TRANS_DEEPLX,
OPT_TRANS_BAIDU,
OPT_TRANS_TENCENT,
OPT_TRANS_OPENAI,
OPT_TRANS_CUSTOMIZE,
DEFAULT_FETCH_INTERVAL,
DEFAULT_FETCH_LIMIT,
OPT_LANGS_SPECIAL,
URL_MICROSOFT_TRAN,
URL_TENCENT_TRANSMART,
} from "../config";
import { msAuth } from "./auth";
import { isBg } from "./browser";
import { newCacheReq, newTransReq } from "./req";
/**
* 油猴脚本的请求封装
@@ -38,153 +25,35 @@ export const fetchGM = async (input, { method = "GET", headers, body } = {}) =>
url: input,
headers,
data: body,
onload: ({ response, responseHeaders, status, statusText }) => {
const headers = new Headers();
// withCredentials: true,
onload: ({ response, responseHeaders, status, statusText, ...opts }) => {
const headers = {};
responseHeaders.split("\n").forEach((line) => {
const [name, value] = line.split(":").map((item) => item.trim());
if (name && value) {
headers.append(name, value);
headers[name] = value;
}
});
resolve(
new Response(response, {
headers,
status,
statusText,
})
);
resolve({
body: response,
headers,
status,
statusText,
});
},
onerror: reject,
});
});
/**
* 构造缓存 request
* @param {*} request
* @returns
*/
const newCacheReq = async (input, init) => {
let request = new Request(input, init);
if (request.method !== "GET") {
const body = await request.text();
const cacheUrl = new URL(request.url);
cacheUrl.pathname += body;
request = new Request(cacheUrl.toString(), { method: "GET" });
}
return request;
};
const newTransReq = (
{ translator, text, fromLang, toLang },
{ url, key } = {}
) => {
// console.log({ translator, text, fromLang, toLang }, { url, key });
let params;
let data;
let input;
let init;
const from = OPT_LANGS_SPECIAL[translator].get(fromLang) ?? "";
const to = OPT_LANGS_SPECIAL[translator].get(toLang);
if (!to) {
throw new Error(`[trans] target lang: ${toLang} not support`);
}
switch (translator) {
case OPT_TRANS_GOOGLE:
break;
case OPT_TRANS_MICROSOFT:
params = {
from,
to,
"api-version": "3.0",
};
input = `${URL_MICROSOFT_TRAN}?${queryString.stringify(params)}`;
init = {
headers: {
"Content-type": "application/json",
Authorization: `Bearer ${key}`,
},
method: "POST",
body: JSON.stringify([{ Text: text }]),
};
break;
case OPT_TRANS_DEEPL:
break;
case OPT_TRANS_DEEPLFREE:
break;
case OPT_TRANS_DEEPLX:
break;
case OPT_TRANS_BAIDU:
break;
case OPT_TRANS_TENCENT:
data = {
header: {
fn: "auto_translation_block",
},
source: {
text_block: text,
},
target: {
lang: to,
},
};
if (from) {
data.source.lang = from;
}
input = URL_TENCENT_TRANSMART;
init = {
headers: {
"Content-Type": "application/json",
},
method: "POST",
body: JSON.stringify(data),
};
break;
case OPT_TRANS_OPENAI:
break;
case OPT_TRANS_CUSTOMIZE:
break;
default:
break;
}
if (!input) {
throw new Error(`[trans] translator: ${translator} not support`);
}
return [input, init];
};
/**
* 发起请求
* @param {*} param0
* @returns
*/
export const fetchApi = async ({
input,
init,
transOpts,
apiSetting,
token,
}) => {
if (token) {
apiSetting.key = token;
}
export const fetchApi = async ({ input, init, transOpts, apiSetting }) => {
if (transOpts?.translator) {
[input, init] = newTransReq(transOpts, apiSetting);
[input, init] = await newTransReq(transOpts, apiSetting);
}
// if (token) {
// if (translator === OPT_TRANS_DEEPL) {
// init.headers["Authorization"] = `DeepL-Auth-Key ${token}`; // DeepL
// } else if (translator === OPT_TRANS_OPENAI) {
// init.headers["Authorization"] = `Bearer ${token}`; // OpenAI
// init.headers["api-key"] = token; // Azure OpenAI
// } else {
// init.headers["Authorization"] = `Bearer ${token}`; // Microsoft & others
// }
// }
if (isGm) {
let info;
@@ -193,19 +62,26 @@ export const fetchApi = async ({
} else {
info = GM.info;
}
// Tampermonkey --> .connects
// Violentmonkey --> .connect
const connects = info?.script?.connects || info?.script?.connect || [];
const url = new URL(input);
const isSafe = connects.find((item) => url.hostname.endsWith(item));
if (isSafe) {
if (window.KISS_GM) {
return window.KISS_GM.fetch(input, init);
} else {
return fetchGM(input, init);
}
const { body, headers, status, statusText } = window.KISS_GM
? await window.KISS_GM.fetch(input, init)
: await fetchGM(input, init);
return new Response(body, {
headers: new Headers(headers),
status,
statusText,
});
}
}
return fetch(input, init);
};
@@ -214,13 +90,7 @@ export const fetchApi = async ({
*/
export const fetchPool = taskPool(
fetchApi,
async ({ transOpts }) => {
if (transOpts?.translator === OPT_TRANS_MICROSOFT) {
const [token] = await msAuth();
return { token };
}
return {};
},
null,
DEFAULT_FETCH_INTERVAL,
DEFAULT_FETCH_LIMIT
);