256 lines
9.2 KiB
TypeScript
256 lines
9.2 KiB
TypeScript
import { GeneralResponse } from "../../../model/generalResponse";
|
||
import { TranslateModel } from "../../../model/translate";
|
||
import { errorMessage, successMessage } from "../../Public/generalTools";
|
||
import { Translate } from "./Translate";
|
||
import { DEFINE_STRING } from "../../../define/define_string"
|
||
import { TranslateAPIType, TranslateType } from "../../../define/enum/translate";
|
||
import { BookTaskDetailService } from "../../../define/db/service/Book/bookTaskDetailService";
|
||
import { Book } from "../../../model/book";
|
||
import { ResponseMessageType } from "../../../define/enum/softwareEnum";
|
||
import { SoftwareService } from '../../../define/db/service/SoftWare/softwareService'
|
||
import { isEmpty } from "lodash";
|
||
import { ValidateJson } from "../../../define/Tools/validate";
|
||
|
||
/**
|
||
* 翻译实现服务
|
||
*/
|
||
export class TranslateService {
|
||
translate: Translate
|
||
bookTaskDetail: BookTaskDetailService;
|
||
softwareService: SoftwareService
|
||
|
||
constructor() {
|
||
}
|
||
|
||
async InitService() {
|
||
if (!this.bookTaskDetail) {
|
||
this.bookTaskDetail = await BookTaskDetailService.getInstance()
|
||
}
|
||
if (!this.softwareService) {
|
||
this.softwareService = await SoftwareService.getInstance()
|
||
}
|
||
if (!this.translate) {
|
||
this.translate = new Translate()
|
||
}
|
||
}
|
||
|
||
// 返回翻译结果。用于前端修改
|
||
private sendTranslateReturn(windowId: number, data: GeneralResponse.MessageResponse): void {
|
||
let win = global.newWindow[0]
|
||
if (windowId) {
|
||
win = global.newWindow.filter(item => item.id == windowId)[0];
|
||
}
|
||
win.win.webContents.send(DEFINE_STRING.BOOK.MAIN_DATA_RETURN, data)
|
||
}
|
||
|
||
|
||
//#region 翻译设置
|
||
|
||
/**
|
||
* 初始化翻译设置
|
||
* @returns
|
||
*/
|
||
InitialTranslateSetting(): TranslateModel.TranslateModel {
|
||
return {
|
||
selectModel: TranslateAPIType.BAIDU,
|
||
translation_auto: true,
|
||
translates: [
|
||
{
|
||
name: TranslateAPIType.LAITOOL,
|
||
translation_business: "https://api.laitool.cc/v1/chat/completions",
|
||
translation_app_id: "gpt-4o-mini",
|
||
translation_secret: "LAI API 令牌"
|
||
},
|
||
{
|
||
name: TranslateAPIType.BAIDU,
|
||
translation_business: "https://fanyi-api.baidu.com/api/trans/vip/translate",
|
||
translation_app_id: "百度翻译的APP ID",
|
||
translation_secret: "百度翻译的密钥"
|
||
},
|
||
{
|
||
name: TranslateAPIType.TENCENT,
|
||
translation_business: "https://tmt.tencentcloudapi.com",
|
||
translation_app_id: '腾讯云的APP ID',
|
||
translation_secret: "腾讯云的密钥"
|
||
},
|
||
{
|
||
name: TranslateAPIType.VOLCENGINE,
|
||
translation_business: "https://translate.volcengineapi.com?",
|
||
translation_app_id: '火山引擎的APP ID',
|
||
translation_secret: '火山引擎的密钥'
|
||
},
|
||
{
|
||
name: TranslateAPIType.ALI,
|
||
translation_business: "https://mt.cn-hangzhou.aliyuncs.com",
|
||
translation_app_id: '阿里云的APP ID',
|
||
translation_secret: '阿里云的密钥'
|
||
}
|
||
]
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 获取翻译设置
|
||
*/
|
||
async GetTranslateSetting(): Promise<GeneralResponse.SuccessItem | GeneralResponse.ErrorItem> {
|
||
try {
|
||
await this.InitService()
|
||
let translateSetting = undefined as TranslateModel.TranslateModel
|
||
let translateSettingString = this.softwareService.GetSoftWarePropertyData('translationSetting');
|
||
if (isEmpty(translateSettingString)) {
|
||
// 初始化
|
||
translateSetting = this.InitialTranslateSetting();
|
||
await this.ResetTranslateSetting();
|
||
translateSettingString = this.softwareService.GetSoftWarePropertyData('translationSetting');
|
||
translateSetting = JSON.parse(translateSettingString);
|
||
} else {
|
||
// 解析
|
||
let tryParse = ValidateJson(translateSettingString)
|
||
if (!tryParse) {
|
||
throw new Error("翻译设置数据解析失败")
|
||
}
|
||
translateSetting = JSON.parse(translateSettingString);
|
||
}
|
||
|
||
// v3.0.1 preview 4 版本新增laitool gpt翻译，之前版本没有，这边要检测下是不是有配置，没有初始化
|
||
let index = translateSetting.translates.findIndex(item => item.name == TranslateAPIType.LAITOOL);
|
||
if (index < 0) {
|
||
// 初始化，将数据加在最前面
|
||
let laitool = {
|
||
name: TranslateAPIType.LAITOOL,
|
||
translation_business: "https://api.laitool.cc/v1/chat/completions",
|
||
translation_app_id: "gpt-4o-mini",
|
||
translation_secret: "LAI API 令牌"
|
||
}
|
||
translateSetting.translates.unshift(laitool)
|
||
}
|
||
|
||
return successMessage(translateSetting, "获取翻译设置成功", "TranslateService_GetTranslateSetting")
|
||
} catch (error) {
|
||
return errorMessage("获取翻译设置失败，失败信息如下：" + error.toString(), "TranslateService_GetTranslateSetting")
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 重置翻译设置
|
||
*/
|
||
async ResetTranslateSetting(): Promise<GeneralResponse.SuccessItem | GeneralResponse.ErrorItem> {
|
||
try {
|
||
let translateSetting = this.InitialTranslateSetting()
|
||
let res = this.softwareService.SaveSoftwarePropertyData('translationSetting', JSON.stringify(translateSetting))
|
||
return successMessage(translateSetting, "重置翻译设置成功", "TranslateService_ResetTranslateSetting")
|
||
|
||
} catch (error) {
|
||
return errorMessage("重置翻译设置失败，失败信息如下：" + error.toString(), "TranslateService_ResetTranslateSetting")
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 保存翻译设置
|
||
* @param value 保存的数据
|
||
* @returns
|
||
*/
|
||
async SaveTranslateSetting(value: TranslateModel.TranslateModel): Promise<GeneralResponse.SuccessItem | GeneralResponse.ErrorItem> {
|
||
try {
|
||
let res = this.softwareService.SaveSoftwarePropertyData('translationSetting', JSON.stringify(value))
|
||
// 这边要判断是不是用的laitool
|
||
let laitool = value.translates.filter(item => item.name == TranslateAPIType.LAITOOL)
|
||
if (laitool.length > 0) {
|
||
let laitoolData = laitool[0]
|
||
if (!laitoolData.translation_business.includes('laitool')) {
|
||
throw new Error("不允许修改laitool的地址为其他外站地址")
|
||
}
|
||
}
|
||
return successMessage(value, "保存翻译设置成功", "TranslateService_SaveTranslateSetting")
|
||
} catch (error) {
|
||
return errorMessage("保存翻译设置失败，失败信息如下：" + error.toString(), "TranslateService_SaveTranslateSetting")
|
||
|
||
}
|
||
}
|
||
|
||
|
||
//#endregion
|
||
|
||
|
||
|
||
|
||
/**
|
||
* 反推翻译提示词处理
|
||
* @param bookTaskDetailId 对应的分镜ID
|
||
* @param reversePromptId 对应的反推提示词数据ID
|
||
* @param to 目标语言
|
||
* @param dstString 翻译后的字符串
|
||
*/
|
||
private async TranslateProcessReversePrompt(bookTaskDetailId: string, reversePromptId: string, to: string, dstString: string): Promise<void> {
|
||
// 开始修改翻译后的数据
|
||
let updateData = {} as Book.ReversePrompt
|
||
if (to == "zh") {
|
||
updateData.promptCN = dstString
|
||
} else if (to == "en") {
|
||
updateData.prompt = dstString
|
||
updateData.promptCN = dstString
|
||
}
|
||
// 修改数据
|
||
this.bookTaskDetail.UpdateBookTaskDetailReversePrompt(bookTaskDetailId, reversePromptId, updateData)
|
||
}
|
||
|
||
// 处理返回的数据
|
||
async TranslateReturnProcess(value: TranslateModel.TranslateNowIPCParams, to: string, dstString: string) {
|
||
switch (value.type) {
|
||
case TranslateType.REVERSE_PROMPT_TRANSLATE:
|
||
await this.TranslateProcessReversePrompt(value.bookTaskDetailId, value.reversePromptId, to, dstString)
|
||
break;
|
||
default:
|
||
throw new Error("未知的翻译类型");
|
||
}
|
||
}
|
||
|
||
// 翻译
|
||
async TranslateNowReturn(value: TranslateModel.TranslateNowIPCParams[]): Promise<GeneralResponse.SuccessItem | GeneralResponse.ErrorItem> {
|
||
try {
|
||
await this.InitService()
|
||
// 循环所有的数据，返回翻译结果
|
||
for (let i = 0; i < value.length; i++) {
|
||
const element = value[i];
|
||
let res = await this.translate.TranslateReturnNow(element)
|
||
// 单个翻译，将返回的数据写入到原数据中
|
||
|
||
// 添加一个对返回信息进行处理的函数
|
||
let data = res.data
|
||
// 先将数据进行拼接
|
||
let srcString = ""
|
||
if (element.isSplit) {
|
||
let dstStrs = []
|
||
} else {
|
||
// 没有拆分的，只有一句
|
||
srcString = data.data[0].dst
|
||
}
|
||
// 写回数据库
|
||
await this.TranslateReturnProcess(element, data.to, srcString);
|
||
|
||
// 做个返回数据
|
||
this.sendTranslateReturn(element.windowId, {
|
||
code: 1,
|
||
id: element.bookTaskDetailId,
|
||
type: ResponseMessageType.PROMPT_TRANSLATE,
|
||
data: {
|
||
progress: i + 1,
|
||
total: value.length,
|
||
from: element.from,
|
||
to: data.to,
|
||
bookTaskDetailId: element.bookTaskDetailId,
|
||
reversePromptId: element.reversePromptId,
|
||
prompt: srcString,
|
||
promptCN: srcString
|
||
}
|
||
})
|
||
}
|
||
|
||
return successMessage(null, "全部翻译完成", "TranslateService_TranslateNowReturn")
|
||
|
||
} catch (error) {
|
||
return errorMessage("翻译失败，失败信息如下：" + error.toString(), "TranslateService_TranslateNowReturn")
|
||
}
|
||
}
|
||
|
||
} |