Fix plugin router inject

Author: wu-clan

backend/plugin/tools.py

@@ -17,6 +17,10 @@
 from backend.utils.import_parse import import_module_cached
 
 
+class PluginInjectError(Exception):
+    pass
+
+
 def get_plugins() -> list[str]:
     """获取插件"""
     plugin_packages = []
@@ -54,7 +58,7 @@ def plugin_router_inject() -> None:
     for plugin in plugins:
         toml_path = os.path.join(PLUGIN_DIR, plugin, 'plugin.toml')
         if not os.path.exists(toml_path):
-            raise FileNotFoundError(f'插件 {plugin} 缺少 plugin.toml 配置文件，请检查插件是否合法')
+            raise PluginInjectError(f'插件 {plugin} 缺少 plugin.toml 配置文件，请检查插件是否合法')
 
         # 获取 plugin.toml 配置
         with open(toml_path, 'r', encoding='utf-8') as f:
@@ -65,12 +69,12 @@ def plugin_router_inject() -> None:
         if api:
             app_include = data.get('app', {}).get('include', '')
             if not app_include:
-                raise ValueError(f'插件 {plugin} 配置文件存在错误，请检查')
+                raise PluginInjectError(f'非独立 app 插件 {plugin} 配置文件存在错误，请检查')
 
             # 插件中 API 路由文件的路径
             plugin_api_path = os.path.join(PLUGIN_DIR, plugin, 'api')
             if not os.path.exists(plugin_api_path):
-                raise FileNotFoundError(f'插件 {plugin} 缺少 api 目录，请检查插件文件是否完整')
+                raise PluginInjectError(f'插件 {plugin} 缺少 api 目录，请检查插件文件是否完整')
 
             # 将插件路由注入到对应模块的路由中
             for root, _, api_files in os.walk(plugin_api_path):
@@ -86,8 +90,8 @@ def plugin_router_inject() -> None:
                         module_path = f'backend.plugin.{path_to_module_str}'
                         try:
                             module = import_module_cached(module_path)
-                        except ImportError as e:
-                            raise ImportError(f'导入非独立 app 插件 {plugin} 模块 {module_path} 失败：{e}') from e
+                        except PluginInjectError as e:
+                            raise PluginInjectError(f'导入非独立 app 插件 {plugin} 模块 {module_path} 失败：{e}') from e
                         plugin_router = getattr(module, 'router', None)
                         if not plugin_router:
                             warnings.warn(
@@ -102,11 +106,11 @@ def plugin_router_inject() -> None:
                         target_module_path = f'backend.app.{app_include}.api.{relative_path.replace(os.sep, ".")}'
                         try:
                             target_module = import_module_cached(target_module_path)
-                        except ImportError as e:
-                            raise ImportError(f'导入源程序模块 {target_module_path} 失败：{e}') from e
+                        except PluginInjectError as e:
+                            raise PluginInjectError(f'导入源程序模块 {target_module_path} 失败：{e}') from e
                         target_router = getattr(target_module, 'router', None)
                         if not target_router or not isinstance(target_router, APIRouter):
-                            raise AttributeError(
+                            raise PluginInjectError(
                                 f'非独立 app 插件 {plugin} 模块 {module_path} 中没有有效的 router，'
                                 '请检查插件文件是否完整'
                             )
@@ -123,19 +127,23 @@ def plugin_router_inject() -> None:
             module_path = f'backend.plugin.{plugin}.api.router'
             try:
                 module = import_module_cached(module_path)
-            except ImportError as e:
-                raise ImportError(f'导入独立 app 插件 {plugin} 模块 {module_path} 失败：{e}') from e
-            plugin_router = getattr(module, 'router', None)
-            if not plugin_router or not isinstance(plugin_router, APIRouter):
-                raise AttributeError(
-                    f'独立 app 插件 {plugin} 模块 {module_path} 中没有有效的 router，请检查插件文件是否完整'
-                )
-            target_module_path = 'backend.app.router'
-            target_module = import_module_cached(target_module_path)
-            target_router = getattr(target_module, 'router')
-
-            # 将插件路由注入到目标 router 中
-            target_router.include_router(plugin_router)
+            except PluginInjectError as e:
+                raise PluginInjectError(f'导入独立 app 插件 {plugin} 模块 {module_path} 失败：{e}') from e
+            routers = data.get('app', {}).get('router', [])
+            if not routers or not isinstance(routers, list):
+                raise PluginInjectError(f'独立 app 插件 {plugin} 配置文件存在错误，请检查')
+            for router in routers:
+                plugin_router = getattr(module, router, None)
+                if not plugin_router or not isinstance(plugin_router, APIRouter):
+                    raise PluginInjectError(
+                        f'独立 app 插件 {plugin} 模块 {module_path} 中没有有效的 router，请检查插件文件是否完整'
+                    )
+                target_module_path = 'backend.app.router'
+                target_module = import_module_cached(target_module_path)
+                target_router = getattr(target_module, 'router')
+
+                # 将插件路由注入到目标 router 中
+                target_router.include_router(plugin_router)
 
 
 def install_requirements() -> None:
@@ -150,7 +158,7 @@ def install_requirements() -> None:
                 subprocess.run([sys.executable, '-m', 'ensurepip', '--upgrade'])
                 subprocess.check_call([sys.executable, '-m', 'pip', 'install', '-r', requirements_file])
             except subprocess.CalledProcessError as e:
-                raise EnvironmentError(f'插件 {plugin} 依赖安装失败：{e}') from e
+                raise PluginInjectError(f'插件 {plugin} 依赖安装失败：{e}') from e
 
 
 async def install_requirements_async() -> None:
@@ -174,4 +182,4 @@ async def install_requirements_async() -> None:
             )
             _, stderr = await res.communicate()
             if res.returncode != 0:
-                raise EnvironmentError(f'插件 {plugin} 依赖包安装失败：{stderr}')
+                raise PluginInjectError(f'插件 {plugin} 依赖包安装失败：{stderr}')