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: false, translates: [ { name: TranslateAPIType.BAIDU, translation_business: "https://fanyi-api.baidu.com/api/trans/vip/translate", translation_app_id: undefined, translation_secret: undefined }, { name: TranslateAPIType.TENCENT, translation_business: "https://tmt.tencentcloudapi.com", translation_app_id: undefined, translation_secret: undefined }, { name: TranslateAPIType.VOLCENGINE, translation_business: "https://translate.volcengineapi.com?", translation_app_id: undefined, translation_secret: undefined }, { name: TranslateAPIType.ALI, translation_business: "https://mt.cn-hangzhou.aliyuncs.com", translation_app_id: undefined, translation_secret: undefined } ] } } /** * 获取翻译设置 */ async GetTranslateSetting(): Promise { try { await this.InitService() let translateSetting = undefined as TranslateModel.TranslateModel let translateSettingString = this.softwareService.GetSoftWarePropertyData('translationSetting'); if (isEmpty(translateSettingString)) { // 初始化 translateSetting = this.InitialTranslateSetting() } else { // 解析 let tryParse = ValidateJson(translateSettingString) if (!tryParse) { throw new Error("翻译设置数据解析失败") } translateSetting = JSON.parse(translateSettingString); } return successMessage(translateSetting, "获取翻译设置成功", "TranslateService_GetTranslateSetting") } catch (error) { return errorMessage("获取翻译设置失败，失败信息如下：" + error.toString(), "TranslateService_GetTranslateSetting") } } /** * 重置翻译设置 */ async ResetTranslateSetting(): Promise { 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 { try { let res = this.softwareService.SaveSoftwarePropertyData('translationSetting', JSON.stringify(value)) 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 { // 开始修改翻译后的数据 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 { 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") } } }