48 lines
1.8 KiB
JavaScript
48 lines
1.8 KiB
JavaScript
import {
|
|
ipcMain
|
|
} from "electron";
|
|
import { DEFINE_STRING } from '../../define/define_string'
|
|
import {
|
|
ImageGenerate
|
|
} from '../ReverseManage/imageGenerate'
|
|
let imageGenerate = new ImageGenerate(global);
|
|
|
|
function ImageGenerateIpc() {
|
|
// 监听获取任务列表
|
|
ipcMain.handle(DEFINE_STRING.GET_GENERATE_TASK_LIST, async (event) => await imageGenerate.GetGenerateTaskList());
|
|
|
|
// 监听修改任务信息
|
|
ipcMain.handle(DEFINE_STRING.MODIFY_IMAGE_TASK_LIST, async (event, value) => await imageGenerate.ModifyImageTaskList(value));
|
|
|
|
// 监听生成一次图片任务
|
|
ipcMain.handle(DEFINE_STRING.GENERATE_IMAGWE_IN_SELECT_TASK, async (event, value) => await imageGenerate.GenerateImageInSelectTask(value));
|
|
// 监听生成sd api json 的任务
|
|
|
|
ipcMain.handle(DEFINE_STRING.ADD_WEBUI_JSON, async (event) => await imageGenerate.AddWebuiJson());
|
|
// 下载文件到指定位置
|
|
|
|
ipcMain.handle(DEFINE_STRING.DOWNLOAD_IMAGE_FILE, async (event, value) => await imageGenerate.DownloadImageFile(value));
|
|
// 获取指定名字的prompt的配置文件
|
|
|
|
ipcMain.handle(DEFINE_STRING.GET_PROMPT_JSON, async (event, value) => await imageGenerate.GetPromptJson(value));
|
|
// 修改后台队列任务的状态
|
|
|
|
ipcMain.handle(DEFINE_STRING.MODIFY_GENERATE_TASK_STATUS, async (event, value) => await imageGenerate.ModifyGenerateTaskStatus(value));
|
|
// 监听获取基础数据
|
|
|
|
ipcMain.handle(DEFINE_STRING.GET_IAMGE_PROMPT_LIST, async (event) => {
|
|
// 获取最新的那个窗口
|
|
let w = global.newWindow[global.newWindow.length - 1];
|
|
let res = await imageGenerate.getFolderImageList(w, w.init_folder, true);
|
|
return res;
|
|
})
|
|
|
|
// 监听修改配置json
|
|
ipcMain.handle(DEFINE_STRING.MODIFY_INPUT_CROP_JSON, async (event, value) => await imageGenerate.ModifyInpurCropJson(value));
|
|
|
|
}
|
|
|
|
export {
|
|
ImageGenerateIpc
|
|
}
|