lyrebird support CORS (#818)

Co-authored-by: yangfan100 <[email protected]>

lyrebird/mock/mock_server.py

@@ -1,4 +1,5 @@
 from flask import Flask, redirect, url_for, request
+from flask_cors import CORS
 from . import context
 from .blueprints.apis import api
 from .blueprints.ui import ui
@@ -45,8 +46,7 @@ def __init__(self):
         self.debug = False
         self.port = 9090
         self._working_thread = None
-        self.app = Flask('MOCK')
-
+        self.app = self.create_app()
         # async_mode = threading / eventlet / gevent / gevent_uwsgi
         self.socket_io = SocketIO(self.app, async_mode='threading', logger=False, cors_allowed_origins='*')
 
@@ -87,6 +87,16 @@ def on_app_error(error):
 
             return application.make_fail_response(f'Mock server error:\n {trace}')
 
+    def create_app(self):
+        app = Flask('MOCK')
+        cors_resources = application.config.get('mock.server.cors.resources')
+        if cors_resources and isinstance(cors_resources, dict):
+            try:
+                CORS(app, resources=cors_resources)
+            except Exception as e:
+                _logger.warning(f"An error occurred while setting CORS. The default mode does not support CORS. Error msg: {e}")
+        return app
+
     def run(self):
         server_ip = application.config.get('ip')
         _logger.log(60, f'Core start on http://{server_ip}:{self.port}')

requirements.txt

@@ -12,3 +12,4 @@ SQLAlchemy==1.3.22
 aiohttp==3.8.3
 netifaces==0.11.0
 jsonschema==4.17.0
+Flask-Cors==4.0.0

requirements.txt.lock

@@ -14,6 +14,7 @@ colorama==0.4.1
 Flask==2.2.2
 Flask-RESTful==0.3.9
 Flask-SocketIO==5.3.0
+Flask-Cors==4.0.0
 frozenlist==1.3.1
 idna==2.8
 importlib-metadata==5.0.0