AskaEth
e6875f0b4b
- 13-service async plugin framework - Textual TUI with CLI fallback - Plugin hot-reload + permission system - Web management panel (aiohttp) - Bridge-based inter-module communication - 10 regression tests Fixes applied: - PBKDF2-SHA256 auth (was plain SHA256) - Auth bypass removed (was allow-all on fail) - Bare excepts replaced with logged errors - CatFramework/DreamSu -> SenSu naming unified - ServiceManager: health checks + startup_order - Env var credentials (SENSU_ADMIN_PASSWORD etc)
128 lines
4.8 KiB
Python
128 lines
4.8 KiB
Python
#!/usr/bin/env python3
|
|
# -*- coding: utf-8 -*-
|
|
|
|
import logging
|
|
import asyncio
|
|
from typing import Dict, List, Callable, Any
|
|
import json
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
class PluginNetworkBridge:
|
|
"""插件网络桥接 - 简化插件的网络交互"""
|
|
|
|
def __init__(self, plugin_name: str, internet_service, plugin_bridge):
|
|
self.plugin_name = plugin_name
|
|
self.internet_service = internet_service
|
|
self.plugin_bridge = plugin_bridge
|
|
self.registered_routes: List[Dict] = []
|
|
self.websocket_handlers: List[Dict] = []
|
|
|
|
logger.debug(f"插件网络桥接初始化: {plugin_name}")
|
|
|
|
def is_network_available(self):
|
|
"""检查网络服务是否可用"""
|
|
return self.internet_service is not None and hasattr(self.internet_service, 'register_plugin_route')
|
|
|
|
async def register_http_route(self, route_path: str, handler: Callable,
|
|
methods: List[str] = ["GET"], require_auth: bool = True):
|
|
"""注册HTTP路由"""
|
|
try:
|
|
if not self.is_network_available():
|
|
logger.warning("网络服务不可用，跳过HTTP路由注册")
|
|
return
|
|
|
|
await self.internet_service.register_plugin_route(
|
|
self.plugin_name, route_path, handler, methods, require_auth
|
|
)
|
|
|
|
self.registered_routes.append({
|
|
'type': 'http',
|
|
'path': route_path,
|
|
'methods': methods,
|
|
'require_auth': require_auth
|
|
})
|
|
|
|
logger.debug(f"插件 {self.plugin_name} 注册HTTP路由: {route_path}")
|
|
|
|
except Exception as e:
|
|
logger.error(f"注册HTTP路由时出错: {str(e)}")
|
|
# 不抛出异常，让插件继续运行
|
|
|
|
async def register_websocket(self, ws_path: str, handler: Callable, require_auth: bool = True):
|
|
"""注册WebSocket处理器"""
|
|
try:
|
|
if not self.is_network_available():
|
|
logger.warning("网络服务不可用，跳过WebSocket注册")
|
|
return
|
|
|
|
await self.internet_service.register_plugin_websocket(
|
|
self.plugin_name, ws_path, handler, require_auth
|
|
)
|
|
|
|
self.websocket_handlers.append({
|
|
'path': ws_path,
|
|
'require_auth': require_auth
|
|
})
|
|
|
|
logger.debug(f"插件 {self.plugin_name} 注册WebSocket: {ws_path}")
|
|
|
|
except Exception as e:
|
|
logger.error(f"注册WebSocket时出错: {str(e)}")
|
|
# 不抛出异常，让插件继续运行
|
|
|
|
async def broadcast_websocket(self, message: Dict):
|
|
"""向插件的所有WebSocket连接广播消息"""
|
|
try:
|
|
if not self.internet_service:
|
|
logger.warning("网络服务不可用，无法广播消息")
|
|
return
|
|
|
|
await self.internet_service.broadcast_to_websockets(self.plugin_name, message)
|
|
logger.debug(f"插件 {self.plugin_name} WebSocket广播: {len(message)} 字节")
|
|
except Exception as e:
|
|
logger.error(f"WebSocket广播时出错: {str(e)}")
|
|
|
|
async def send_data_to_client(self, client_id: str, message: Dict):
|
|
"""向特定客户端发送数据"""
|
|
try:
|
|
# 这里可以实现更精确的客户端消息发送
|
|
# 目前先使用广播
|
|
message['target_client'] = client_id
|
|
await self.broadcast_websocket(message)
|
|
|
|
except Exception as e:
|
|
logger.error(f"发送数据到客户端时出错: {str(e)}")
|
|
|
|
def get_network_info(self) -> Dict[str, Any]:
|
|
"""获取网络配置信息"""
|
|
if not self.internet_service:
|
|
return {
|
|
'plugin_name': self.plugin_name,
|
|
'registered_routes': [],
|
|
'websocket_handlers': [],
|
|
'base_url': '网络服务不可用'
|
|
}
|
|
|
|
return {
|
|
'plugin_name': self.plugin_name,
|
|
'registered_routes': self.registered_routes,
|
|
'websocket_handlers': self.websocket_handlers,
|
|
'base_url': f"http://{self.internet_service.http_host}:{self.internet_service.http_port}/{self.plugin_name}"
|
|
}
|
|
|
|
async def setup_data_transfer(self, data_handler: Callable):
|
|
"""设置跨端数据传输"""
|
|
try:
|
|
# 订阅网络数据接收事件
|
|
self.plugin_bridge.subscribe_plugin(
|
|
self.plugin_name,
|
|
"network.data.receive",
|
|
data_handler
|
|
)
|
|
|
|
logger.debug(f"插件 {self.plugin_name} 设置跨端数据传输")
|
|
|
|
except Exception as e:
|
|
logger.error(f"设置数据传输时出错: {str(e)}")
|