Function ApiRouteInstallExtension

apps/node/src/routes/api/extension.ts:16–80

(req: Request, res: Response)

 * @param res 
 */
export async function ApiRouteInstallExtension(req: Request, res: Response) {
    try {
        const {data} = req.body;
        const taskParams = data as TaskProps; 
        const task: TaskProps = {
            taskId: taskParams.taskId,
            name: taskParams.name,
            params: taskParams.params,
            executor: async (dispatcher: TaskEventDispatcher) => {
                const newDispatcher = (event: PartialTaskEvent) => {
                    dispatcher(event);
                    comfyuiService.comfyuiProgressEvent.emit({
                        type: "OUTPUT",
                        message: event.message || event.error
                    })
                }
                const extensions = taskParams.params as Extension[];
                let updated = false;

                newDispatcher({
                    message: `\n Start to install ${extensions.map(e => e.title).join(", ")}\n`
                });

                for (const extension of extensions) {
                    await checkAExtensionInstalled(extension)
                    if (extension.installed) {
                        newDispatcher({
                            message: `\nExtension ${extension.title} installed`
                        });
                        continue;
                    }
                    const ret = await installExtension(newDispatcher, extension);
                    if (ret) {
                        updated = true
                    } else {
                        newDispatcher({
                            error: `\nExtension ${extension.title} install failed`
                        });
                    }
                }

                if (updated) {
                    await comfyuiService.restartComfyUI();
                    return true;
                }

                return false;
            }
        };
        taskQueue.addTask(task);
        res.send({
            success: true,
            data: {
                taskId: task.taskId
            }
        });
    } catch (err: any) {
        logger.error("error", err);