add more translators...

src/libs/fetch.js

@@ -1,3 +1,4 @@
+import queryString from "query-string";
 import { isExt, isGm } from "./client";
 import { sendBgMsg } from "./msg";
 import { taskPool } from "./pool";
@@ -6,11 +7,20 @@ 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";
@@ -53,7 +63,8 @@ export const fetchGM = async (input, { method = "GET", headers, body } = {}) =>
  * @param {*} request
  * @returns
  */
-const newCacheReq = async (request) => {
+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);
@@ -64,22 +75,116 @@ const newCacheReq = async (request) => {
   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 = {}, translator, token }) => {
+export const fetchApi = async ({
+  input,
+  init,
+  transOpts,
+  apiSetting,
+  token,
+}) => {
   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
-    }
+    apiSetting.key = token;
   }
+  if (transOpts?.translator) {
+    [input, init] = 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;
@@ -109,8 +214,8 @@ export const fetchApi = async ({ input, init = {}, translator, token }) => {
  */
 export const fetchPool = taskPool(
   fetchApi,
-  async ({ translator }) => {
-    if (translator === OPT_TRANS_MICROSOFT) {
+  async ({ transOpts }) => {
+    if (transOpts?.translator === OPT_TRANS_MICROSOFT) {
       const [token] = await msAuth();
       return { token };
     }
@@ -128,9 +233,9 @@ export const fetchPool = taskPool(
  */
 export const fetchData = async (
   input,
-  { useCache, usePool, translator, token, ...init } = {}
+  { useCache, usePool, transOpts, apiSetting, ...init } = {}
 ) => {
-  const cacheReq = await newCacheReq(new Request(input, init));
+  const cacheReq = await newCacheReq(input, init);
   let res;
 
   // 查询缓存
@@ -146,9 +251,9 @@ export const fetchData = async (
   if (!res) {
     // 发送请求
     if (usePool) {
-      res = await fetchPool.push({ input, init, translator, token });
+      res = await fetchPool.push({ input, init, transOpts, apiSetting });
     } else {
-      res = await fetchApi({ input, init, translator, token });
+      res = await fetchApi({ input, init, transOpts, apiSetting });
     }
 
     if (!res?.ok) {
@@ -180,7 +285,7 @@ export const fetchData = async (
  * @returns
  */
 export const fetchPolyfill = async (input, opts) => {
-  if (!input.trim()) {
+  if (!input?.trim()) {
     throw new Error("URL is empty");
   }