270 lines
11 KiB
JavaScript
270 lines
11 KiB
JavaScript
"use strict";
|
||
const common_vendor = require("../common/vendor.js");
|
||
const types_index = require("../types/index.js");
|
||
const utils_request = require("../utils/request.js");
|
||
const composables_useCommon = require("./useCommon.js");
|
||
const stores_appStore = require("../stores/appStore.js");
|
||
const utils_common = require("../utils/common.js");
|
||
const submitCustomWorkflow = (data) => utils_request.request("/draw/customWorkflow", {
|
||
method: "POST",
|
||
data
|
||
});
|
||
const getModelListByWorkflowId = (workflow_id) => utils_request.request(`/draw/getModelListById/${workflow_id}`);
|
||
const creatDrawHistoryTask = (data) => utils_request.request("/draw/history", { method: "POST", data });
|
||
function useWorkFlow() {
|
||
const workflow = common_vendor.ref({});
|
||
const params_component_list = [
|
||
{ param: "seed", component: "Seed", title: "随机种子" },
|
||
{ param: "seed_2", component: "Seed", title: "随机种子" },
|
||
{ param: "seed_3", component: "Seed", title: "随机种子" },
|
||
{ param: "seed_4", component: "Seed", title: "随机种子" },
|
||
{ param: "seed_5", component: "Seed", title: "随机种子" },
|
||
{ param: "ckpt_name", component: "ModeSelect", title: "大模型选择" },
|
||
{ param: "positive", component: "Positive", title: "正向提示词" },
|
||
{ param: "width", component: "Width", title: "图片宽度" },
|
||
{ param: "height", component: "Height", title: "图片高度" },
|
||
{ param: "batch_size", component: "CustomNumberBox", title: "生成批次" },
|
||
{ param: "image_path_origin", component: "ImageUpload", title: "原图上传" },
|
||
{ param: "image_path_mask", component: "ImageUpload", title: "遮罩上传" },
|
||
{ param: "image_path_face", component: "ImageUpload", title: "参考上传" },
|
||
{ param: "image_path_style", component: "ImageUpload", title: "参考上传" },
|
||
{ param: "advance_select_image_preview", component: "ImageSelectPreview", title: "高级-图像预览选择" },
|
||
{ param: "advance_select_image_preview_2", component: "ImageSelectPreview", title: "高级-图像预览选择" },
|
||
{ param: "advance_select_image_preview_3", component: "ImageSelectPreview", title: "高级-图像预览选择" },
|
||
{ param: "advance_select_image_preview_4", component: "ImageSelectPreview", title: "高级-图像预览选择" }
|
||
];
|
||
const bindParam = common_vendor.ref({});
|
||
const handleGetWorkFlwById = async (id) => {
|
||
workflow.value = await utils_request.request(`/workflow/${id}`);
|
||
handleWorkFlowParamsToBindParam();
|
||
};
|
||
const { localTasks } = common_vendor.storeToRefs(stores_appStore.useAppStore());
|
||
const workFlowParamLists = common_vendor.computed(() => {
|
||
if (!workflow.value || !workflow.value.params) {
|
||
return [];
|
||
}
|
||
return workflow.value.params.filter((item) => item.name !== "output");
|
||
});
|
||
const outputType = common_vendor.computed(() => {
|
||
var _a;
|
||
if (!workflow.value || !workflow.value.params) {
|
||
return "image";
|
||
}
|
||
return ((_a = workflow.value.params.find((item) => item.name === "output")) == null ? void 0 : _a.outputType) || "image";
|
||
});
|
||
const handleFindComponentName = (param) => {
|
||
const component = params_component_list.find((item) => item.param === param);
|
||
return component == null ? void 0 : component.component;
|
||
};
|
||
const handleWorkFlowParamsToBindParam = () => {
|
||
if (!workflow.value || !workflow.value.params) {
|
||
return;
|
||
}
|
||
workflow.value.params.forEach((item) => {
|
||
bindParam.value[item.name] = item.param;
|
||
});
|
||
};
|
||
const socketState = common_vendor.inject("socketState");
|
||
if (!socketState) {
|
||
throw new Error("socketState is not provided");
|
||
}
|
||
const closeSocketAsync = () => {
|
||
return new Promise((resolve) => {
|
||
common_vendor.index.closeSocket({
|
||
code: 1e3,
|
||
reason: "Initializing new WebSocket",
|
||
success(result) {
|
||
resolve(true);
|
||
},
|
||
fail(result) {
|
||
resolve(false);
|
||
}
|
||
});
|
||
});
|
||
};
|
||
const socketInit = async (options = { params: { type: types_index.IWebsocketSceneType.drawProcessPush, data: { scene_str: "" } } }) => {
|
||
var _a, _b, _c, _d, _e, _f;
|
||
if (!composables_useCommon.isLogin.value) {
|
||
throw new Error("未登录状态，不允许初始化Websocket");
|
||
}
|
||
console.log("socket init execution，status", (_a = socketState.socket) == null ? void 0 : _a.readyState);
|
||
if ((socketState == null ? void 0 : socketState.isInitialized) && ((_c = (_b = socketState.options) == null ? void 0 : _b.params) == null ? void 0 : _c.type) === ((_d = options.params) == null ? void 0 : _d.type)) {
|
||
console.log("WebSocket is already initialized");
|
||
return;
|
||
}
|
||
if ((socketState == null ? void 0 : socketState.isInitialized) && ((_e = socketState == null ? void 0 : socketState.options.params) == null ? void 0 : _e.type) !== ((_f = options.params) == null ? void 0 : _f.type)) {
|
||
console.log("WebSocket is already initialized,but scene is different,reinitialize");
|
||
await closeSocketAsync();
|
||
socketState.isInitialized = false;
|
||
}
|
||
const { params } = options;
|
||
const { type, data } = params;
|
||
const { uniPlatform } = common_vendor.index.getSystemInfoSync();
|
||
socketState.socket = common_vendor.index.connectSocket({
|
||
url: `${utils_request.getBaseWsURL()}?user_id=${composables_useCommon.getLoginInfo()._id}&type=${type}&platform=${uniPlatform}&data=${data}`,
|
||
complete: () => {
|
||
console.log("WebSocket connect complete");
|
||
}
|
||
});
|
||
socketState.isInitialized = true;
|
||
common_vendor.index.onSocketOpen((result) => {
|
||
console.log("WebSocket opened", result);
|
||
socketState.options = options;
|
||
if (options.onConnect) {
|
||
options.onConnect();
|
||
}
|
||
});
|
||
common_vendor.index.onSocketMessage((msg) => {
|
||
if (options.onMessage) {
|
||
options.onMessage(msg.data);
|
||
}
|
||
handleSocketMessage(msg.data);
|
||
});
|
||
common_vendor.index.onSocketError((err) => {
|
||
console.error("WebSocket onError", err);
|
||
socketState.isInitialized = false;
|
||
if (options.onConnectError) {
|
||
options.onConnectError(err);
|
||
}
|
||
});
|
||
common_vendor.index.onSocketClose(() => {
|
||
socketState.isInitialized = false;
|
||
console.log("WebSocket onClose");
|
||
if (options.onDisconnect) {
|
||
options.onDisconnect();
|
||
}
|
||
});
|
||
};
|
||
const handleSocketHandShake = (timeout = 200) => {
|
||
return new Promise((resolve) => {
|
||
var _a;
|
||
const timeoutID = setTimeout(() => {
|
||
resolve(false);
|
||
}, timeout);
|
||
if (!socketState || !socketState.socket || !socketState.isInitialized) {
|
||
resolve(false);
|
||
return;
|
||
}
|
||
(_a = socketState.socket) == null ? void 0 : _a.send({
|
||
data: "ping",
|
||
success(result) {
|
||
console.log(result);
|
||
clearTimeout(timeoutID);
|
||
resolve(true);
|
||
}
|
||
});
|
||
});
|
||
};
|
||
const handleSocketMessage = (msg, callback) => {
|
||
console.log("原始消息", msg);
|
||
const msgObj = utils_common.parseJSONToObject(msg);
|
||
console.log("handleSocketMessage", msgObj);
|
||
const { type, data, queue_status } = msgObj;
|
||
if (queue_status) {
|
||
const index = localTasks.value.findIndex((item) => item._id === queue_status.task_id);
|
||
if (index !== -1) {
|
||
localTasks.value[index].progress = queue_status.progress;
|
||
localTasks.value[index].queue = queue_status.queue;
|
||
queue_status.time_remained ? localTasks.value[index].time_remained = queue_status.time_remained : null;
|
||
queue_status.message ? localTasks.value[index].message = queue_status.message : null;
|
||
}
|
||
if (queue_status.status === "started" && index !== -1) {
|
||
localTasks.value[index].status = 4;
|
||
localTasks.value[index].message = "任务开始";
|
||
}
|
||
if (queue_status.status !== "failed" && index !== -1 && localTasks.value[index].status === 3) {
|
||
localTasks.value[index].status = 0;
|
||
localTasks.value[index].message = queue_status.message;
|
||
}
|
||
if (queue_status.status === "success" && index !== -1 && localTasks.value[index].status !== 1) {
|
||
localTasks.value[index].output = queue_status.data.output;
|
||
localTasks.value[index].type = queue_status.data.type;
|
||
localTasks.value[index].status = 1;
|
||
console.log("localTasks", localTasks);
|
||
composables_useCommon.refreshUserInfo().then();
|
||
}
|
||
if (queue_status.status === "failed") {
|
||
if (index !== -1 && localTasks.value[index].status !== 2) {
|
||
localTasks.value[index].status = 2;
|
||
localTasks.value[index].message = queue_status.message;
|
||
}
|
||
}
|
||
}
|
||
};
|
||
const handleCreateTask = async () => {
|
||
const item = {
|
||
params: { ...bindParam.value },
|
||
workflow_id: workflow.value._id,
|
||
user_id: composables_useCommon.getLoginInfo()._id,
|
||
options: {
|
||
workflow_id: workflow.value._id,
|
||
workflow_title: workflow.value.title,
|
||
workflow_name: workflow.value.name
|
||
},
|
||
type: outputType.value
|
||
};
|
||
return await creatDrawHistoryTask(item);
|
||
};
|
||
const handleUpdateTaskStatus = async (taskItem, output) => {
|
||
const index = localTasks.value.findIndex((item) => item._id === taskItem._id);
|
||
if (index !== -1) {
|
||
localTasks.value[index].output = output.output;
|
||
localTasks.value[index].status = 1;
|
||
localTasks.value = [...localTasks.value];
|
||
}
|
||
};
|
||
const handleSubmitTaskTask = async () => {
|
||
if (!composables_useCommon.isLogin.value) {
|
||
common_vendor.index.showToast({
|
||
title: "请先登录",
|
||
icon: "none",
|
||
duration: 2e3
|
||
});
|
||
return;
|
||
}
|
||
if (!await handleSocketHandShake()) {
|
||
await socketInit();
|
||
}
|
||
const newTask = await handleCreateTask();
|
||
console.log("newTask", newTask);
|
||
if (!newTask) {
|
||
return;
|
||
}
|
||
localTasks.value.unshift(newTask);
|
||
const requestParams = {
|
||
params: { ...bindParam.value },
|
||
options: {
|
||
workflow_id: workflow.value._id,
|
||
task_id: newTask._id
|
||
//重要，已经创建任务的时候传入任务ID，否则系统会新建
|
||
}
|
||
};
|
||
const result = await submitCustomWorkflow(requestParams);
|
||
if (result.status === "success" && result.output && result.output.length > 0) {
|
||
await handleUpdateTaskStatus(newTask, result);
|
||
composables_useCommon.refreshUserInfo().then();
|
||
}
|
||
if (result.status === "failed" || result.status === "rejected") {
|
||
common_vendor.index.showToast({
|
||
title: result.message,
|
||
icon: "none",
|
||
duration: 2e3
|
||
});
|
||
}
|
||
};
|
||
return {
|
||
workflow,
|
||
workFlowParamLists,
|
||
bindParam,
|
||
params_component_list,
|
||
socketInit,
|
||
// socketState,
|
||
// closeSocketAsync,
|
||
handleFindComponentName,
|
||
handleGetWorkFlwById,
|
||
handleSubmitTaskTask
|
||
};
|
||
}
|
||
exports.getModelListByWorkflowId = getModelListByWorkflowId;
|
||
exports.useWorkFlow = useWorkFlow;
|