org.openea.eap.module.system.service.language.TranslateUtil Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of eap-module-system-biz Show documentation
system 模块下，我们放通用业务，支撑上层的核心业务。例如说：用户、部门、权限、数据字典等等
The newest version!
package org.openea.eap.module.system.service.language;

import cn.hutool.core.util.ObjectUtil;
import cn.hutool.extra.spring.SpringUtil;
import cn.hutool.json.JSONObject;
import lombok.extern.slf4j.Slf4j;
import org.openea.eap.module.infra.api.translate.TranslateApi;

import java.util.*;

@Slf4j
public class TranslateUtil {

    public static TranslateApi getTranslateApi() {
        return SpringUtil.getBean(TranslateApi.class);
    }

    /**
     * 翻译文本
     * @param originText 原始文本
     * @param targetLang 目标语言
     * @param params 参数，可支持type、sourceLang
     * @return 翻译结果
     */
    public static String translateText(String originText, String targetLang, Map params) {
        if(ObjectUtil.isEmpty(originText) || ObjectUtil.isEmpty(targetLang)){
            return originText;
        }
        return getTranslateApi().translateText(originText, targetLang, params);
    }

    public static String translateText(String originText, String targetLang) {
        return translateText(originText, targetLang, Collections.EMPTY_MAP);
    }

    public static String translateText(String originText, String sourceLang, String targetLang){
        if(ObjectUtil.isNotEmpty(sourceLang) && !"auto".equals(sourceLang)){
            Map params = new HashMap<>();
            params.put("sourceLang", sourceLang);
            return translateText(originText, targetLang, params);
        }
        return translateText(originText, targetLang);
    }


    /**
     * 国际化词条
     * @param type 类型
     *             menu/label 翻译保持词性，默认名词，保持大致长度（长度要求更严格）
     *             button/action 翻译保持词性，默认动词，保持大致长度
     *             message/tip/placeholder 一般翻译，支持变量替换
     * @param key  i18nKey 可选，用于辅助翻译
     * @param originText 原始词条
     * @param params 参数，可支持sourceLang 原始语言, targetLangs 多目标语言, length
     * @return 翻译结果json {lang1:target1Text,lang2:target2Text,key:i18nKey}
     */
    public static JSONObject translateI18nJson(String type, String key, String originText, Map params){
        JSONObject targetJson = new JSONObject();
        // check type
        if(ObjectUtil.isEmpty(type)){
            if(params != null && params.containsKey("type")){
                type = params.get("type");
            }
        }
        if(ObjectUtil.isEmpty(type)){
            type = "text";
        }else{
            type = type.toLowerCase();
        }
        // check source lang
        String sourceLang = "auto";
        if(params != null && params.containsKey("sourceLang")){
            sourceLang = params.get("sourceLang");
        }
        if(!"auto".equals(sourceLang)){
            targetJson.set(sourceLang, originText);
        }

        // chatgpt
        if("menu".equals(type) || "button".equals(type)){
            // todo len
            JSONObject resultJson = queryMenuI18n(type, key, originText, 0);
            if(ObjectUtil.isNotEmpty(resultJson)){
                targetJson.putAll(resultJson);
            }
        }
        if(!"auto".equals(sourceLang)){
            targetJson.set(sourceLang, originText);
        }

        // google translate
        if(ObjectUtil.isNotEmpty(targetJson)){
            Set langs = new HashSet<>();
            langs.add("zh-CN");
            langs.add("en-US");
            langs.add("ja-JP");
            for(String lang : langs){
                if(targetJson.containsKey(lang)){
                    continue;
                }
                String result = translateText(originText, lang);
                targetJson.set(lang, result);
            };
        }
        return targetJson;
    }

    public static JSONObject queryMenuI18n(String type, String key, String name, int len) {
        JSONObject json = null;
        String strJson = null;
        try{
            Map params = new HashMap<>();
            params.put("type", type);
            if(len>0){
                params.put("len", ""+len);
            }
            json = getTranslateApi().queryMenuI18n(key, name, params);
            //strJson = ChatGPT.getInstance().queryMenuI18n(type, key, name, len);
            //json = JSONUtil.parseObj(strJson);
        }catch (Exception e){
            log.warn(e.getMessage()+"\r\n"+strJson);
            log.debug(String.format("queryMenuI18n key=%s name=%s",key, name),e);
        }
        return json;
    }
}