From 49f3b2702235806bcda010973d37d71694127bb7 Mon Sep 17 00:00:00 2001
From: huyunan <huyunan@bilibili.com>
Date: Fri, 22 Mar 2024 08:58:43 +0000
Subject: [PATCH] enable_cors_header

---
 __init__.py | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/__init__.py b/__init__.py
index b2c92ed0..e8980ca6 100644
--- a/__init__.py
+++ b/__init__.py
@@ -2126,6 +2126,15 @@ async def api_get_comfyworkflows_auth(request):
         return web.Response(status=404)
     return web.json_response({"comfyworkflows_sharekey" : comfyworkflows_auth})
 
+# 跨域获得本地数据
+args.enable_cors_header = "*"  # 或者可以是 '*'
+if hasattr(server.PromptServer.instance, "app"):
+    # 获取应用实例
+    app = server.PromptServer.instance.app
+    # 创建并添加 CORS 中间件
+    cors_middleware = server.create_cors_middleware(args.enable_cors_header)
+    app.middlewares.append(cors_middleware)
+
 @server.PromptServer.instance.routes.post("/manager/set_esheep_workflow_and_images")
 async def set_esheep_workflow_and_images(request):
     json_data = await request.json()