import {
  ipcMain
} from "electron";
import { DEFINE_STRING } from '../../define/define_string'
import {
  ImageGenerate
} from '../backPrompt/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
}