kiss-translator/src/apis/index.js

import queryString from "query-string";
import { fetchPolyfill } from "../libs/fetch";
import {
  OPT_TRANS_GOOGLE,
  OPT_TRANS_MICROSOFT,
  OPT_TRANS_OPENAI,
  URL_MICROSOFT_TRANS,
  OPT_LANGS_SPECIAL,
  PROMPT_PLACE_FROM,
  PROMPT_PLACE_TO,
} from "../config";
import { getSetting, detectLang } from "../libs";

/**
 * 谷歌翻译
 * @param {*} text
 * @param {*} to
 * @param {*} from
 * @returns
 */
const apiGoogleTranslate = async (text, to, from) => {
  const params = {
    client: "gtx",
    dt: "t",
    dj: 1,
    ie: "UTF-8",
    sl: from,
    tl: to,
    q: text,
  };
  const { googleUrl } = await getSetting();
  const input = `${googleUrl}?${queryString.stringify(params)}`;
  return fetchPolyfill(input, {
    useCache: true,
    usePool: true,
    headers: {
      "Content-type": "application/json",
      "X-Translator": OPT_TRANS_GOOGLE,
    },
  });
};

/**
 * 微软翻译
 * @param {*} text
 * @param {*} to
 * @param {*} from
 * @returns
 */
const apiMicrosoftTranslate = (text, to, from) => {
  const params = {
    from,
    to,
    "api-version": "3.0",
  };
  const input = `${URL_MICROSOFT_TRANS}?${queryString.stringify(params)}`;
  return fetchPolyfill(input, {
    useCache: true,
    usePool: true,
    headers: {
      "Content-type": "application/json",
      "X-Translator": OPT_TRANS_MICROSOFT,
    },
    method: "POST",
    body: JSON.stringify([{ Text: text }]),
  });
};

/**
 * OpenAI 翻译
 * @param {*} text 
 * @param {*} to 
 * @param {*} from 
 * @returns 
 */
const apiOpenaiTranslate = async (text, to, from) => {
  const { openaiUrl, openaiModel, openaiPrompt } = await getSetting();
  let prompt = openaiPrompt
    .replaceAll(PROMPT_PLACE_FROM, from)
    .replaceAll(PROMPT_PLACE_TO, to);
  return fetchPolyfill(openaiUrl, {
    useCache: true,
    usePool: true,
    headers: {
      "Content-type": "application/json",
      "X-Translator": OPT_TRANS_OPENAI,
    },
    method: "POST",
    body: JSON.stringify({
      model: openaiModel,
      messages: [
        {
          role: "system",
          content: prompt,
        },
        {
          role: "user",
          content: text,
        },
      ],
      temperature: 0,
      max_tokens: 256,
    }),
  });
};

/**
 * 统一翻译接口
 * @param {*} param0
 * @returns
 */
export const apiTranslate = async ({ translator, q, fromLang, toLang }) => {
  let trText = "";
  let isSame = false;

  let from = OPT_LANGS_SPECIAL?.[translator]?.get(fromLang) ?? fromLang;
  let to = OPT_LANGS_SPECIAL?.[translator]?.get(toLang) ?? toLang;

  if (translator === OPT_TRANS_GOOGLE) {
    const res = await apiGoogleTranslate(q, to, from);
    trText = res.sentences.map((item) => item.trans).join(" ");
    isSame = to === res.src;
  } else if (translator === OPT_TRANS_MICROSOFT) {
    const res = await apiMicrosoftTranslate(q, to, from);
    trText = res[0].translations[0].text;
    isSame = to === res[0].detectedLanguage.language;
  } else if (translator === OPT_TRANS_OPENAI) {
    const res = await apiOpenaiTranslate(q, to, from);
    trText = res?.choices?.[0].message.content;
    isSame = (await detectLang(q)) === (await detectLang(trText));
  }

  return [trText, isSame];
};