xiaozhi-esp32-server/main/xiaozhi-server/core/xiaozhi_server_facade.py

#!/usr/bin/env python3
"""
小智服务器门面类
统一管理所有协议服务器的启动和停止
"""

import asyncio
from typing import Dict, Any, Optional
from config.logger import setup_logging
from config.config_loader import get_protocol_config, get_mqtt_server_config
from core.servers.multi_protocol_server import MultiProtocolServer

logger = setup_logging()


class XiaozhiServerFacade:
    """
    小智服务器门面类
    提供统一的服务器管理接口，屏蔽内部协议复杂性
    """
    
    def __init__(self, config: Dict[str, Any]):
        """
        初始化服务器门面
        
        Args:
            config: 服务器配置字典
        """
        self.config = config
        self.multi_protocol_server: Optional[MultiProtocolServer] = None
        self.is_initialized = False
        self.is_running = False
        
        # 处理协议配置
        self._setup_protocol_config()
    
    def _setup_protocol_config(self):
        """设置协议配置"""
        try:
            # 获取协议配置
            try:
                protocol_config = get_protocol_config()
                mqtt_config = get_mqtt_server_config()
            except Exception as e:
                logger.warning(f"获取协议配置失败，使用默认配置: {e}")
                # 使用默认配置
                protocol_config = type('ProtocolConfig', (), {
                    'websocket_enabled': True,
                    'mqtt_enabled': False
                })()
                mqtt_config = type('MQTTConfig', (), {
                    'enabled': False,
                    'host': '0.0.0.0',
                    'port': 1883,
                    'udp_port': 1883,
                    'public_ip': 'localhost',
                    'max_connections': 1000,
                    'heartbeat_interval': 30,
                    'max_payload_size': 8192
                })()
            
            # 确定启用的协议
            enabled_protocols = []
            
            # WebSocket协议（默认启用）
            if getattr(protocol_config, 'websocket_enabled', True):
                enabled_protocols.append('websocket')
                logger.info("WebSocket协议已启用")
            
            # MQTT协议
            mqtt_enabled = (
                getattr(protocol_config, 'mqtt_enabled', False) or 
                getattr(mqtt_config, 'enabled', False)
            )
            if mqtt_enabled:
                enabled_protocols.append('mqtt')
                logger.info("MQTT协议已启用")
            
            # 如果没有启用任何协议，默认启用WebSocket
            if not enabled_protocols:
                logger.warning("没有启用任何协议，默认启用WebSocket")
                enabled_protocols = ['websocket']
            
            # 更新配置
            self.config['enabled_protocols'] = enabled_protocols
            
            # 添加MQTT服务器配置
            self.config['mqtt_server'] = {
                'enabled': getattr(mqtt_config, 'enabled', False),
                'host': getattr(mqtt_config, 'host', '0.0.0.0'),
                'port': getattr(mqtt_config, 'port', 1883),
                'udp_port': getattr(mqtt_config, 'udp_port', 1883),
                'public_ip': getattr(mqtt_config, 'public_ip', 'localhost'),
                'max_connections': getattr(mqtt_config, 'max_connections', 1000),
                'heartbeat_interval': getattr(mqtt_config, 'heartbeat_interval', 30),
                'max_payload_size': getattr(mqtt_config, 'max_payload_size', 8192)
            }
            
            logger.info(f"启用的协议: {enabled_protocols}")
            
        except Exception as e:
            logger.error(f"设置协议配置失败: {e}")
            # 使用最基本的配置
            self.config['enabled_protocols'] = ['websocket']
            self.config['mqtt_server'] = {
                'enabled': False,
                'host': '0.0.0.0',
                'port': 1883,
                'udp_port': 1883,
                'public_ip': 'localhost',
                'max_connections': 1000,
                'heartbeat_interval': 30,
                'max_payload_size': 8192
            }
    
    async def initialize(self):
        """初始化服务器"""
        if self.is_initialized:
            logger.warning("服务器已经初始化")
            return
        
        try:
            logger.info("正在初始化小智服务器...")
            
            # 创建多协议服务器
            self.multi_protocol_server = MultiProtocolServer(self.config)
            
            self.is_initialized = True
            logger.info("小智服务器初始化完成")
            
        except Exception as e:
            logger.error(f"初始化服务器失败: {e}")
            raise
    
    async def start(self):
        """启动服务器"""
        if not self.is_initialized:
            await self.initialize()
        
        if self.is_running:
            logger.warning("服务器已经在运行中")
            return
        
        try:
            logger.info("正在启动小智服务器...")
            
            # 启动多协议服务器
            await self.multi_protocol_server.start()
            
            self.is_running = True
            logger.info("小智服务器启动成功")
            
        except Exception as e:
            logger.error(f"启动服务器失败: {e}")
            self.is_running = False
            raise
    
    async def stop(self):
        """停止服务器"""
        if not self.is_running:
            logger.info("服务器未在运行")
            return
        
        try:
            logger.info("正在停止小智服务器...")
            
            if self.multi_protocol_server:
                await self.multi_protocol_server.stop()
            
            self.is_running = False
            logger.info("小智服务器已停止")
            
        except Exception as e:
            logger.error(f"停止服务器失败: {e}")
    
    async def restart(self):
        """重启服务器"""
        logger.info("重启小智服务器...")
        await self.stop()
        await asyncio.sleep(1)  # 等待清理完成
        await self.start()
    
    async def update_config(self, new_config: Dict[str, Any]) -> bool:
        """
        更新服务器配置
        
        Args:
            new_config: 新的配置字典
            
        Returns:
            bool: 更新是否成功
        """
        try:
            logger.info("更新服务器配置...")
            
            # 更新配置
            self.config.update(new_config)
            self._setup_protocol_config()
            
            # 如果服务器正在运行，更新多协议服务器配置
            if self.is_running and self.multi_protocol_server:
                success = await self.multi_protocol_server.update_config(self.config)
                if success:
                    logger.info("服务器配置更新成功")
                else:
                    logger.error("服务器配置更新失败")
                return success
            
            logger.info("配置更新完成（服务器未运行）")
            return True
            
        except Exception as e:
            logger.error(f"更新配置失败: {e}")
            return False
    
    def get_server_status(self) -> Dict[str, Any]:
        """获取服务器状态"""
        base_status = {
            'is_initialized': self.is_initialized,
            'is_running': self.is_running,
            'enabled_protocols': self.config.get('enabled_protocols', [])
        }
        
        if self.multi_protocol_server:
            server_status = self.multi_protocol_server.get_server_status()
            base_status.update(server_status)
        
        return base_status
    
    def get_active_connections_count(self) -> Dict[str, int]:
        """获取各协议的活跃连接数"""
        if self.multi_protocol_server:
            return self.multi_protocol_server.get_active_connections_count()
        return {}
    
    def get_supported_protocols(self) -> list:
        """获取支持的协议列表"""
        if self.multi_protocol_server:
            return self.multi_protocol_server.get_supported_protocols()
        return ['websocket', 'mqtt']
    
    def is_protocol_enabled(self, protocol: str) -> bool:
        """检查协议是否启用"""
        enabled_protocols = self.config.get('enabled_protocols', [])
        return protocol in enabled_protocols
    
    async def broadcast_message(self, message: Dict[str, Any], protocol: Optional[str] = None):
        """
        向所有连接广播消息
        
        Args:
            message: 要广播的消息
            protocol: 指定协议，None表示向所有协议广播
        """
        if self.multi_protocol_server:
            await self.multi_protocol_server.broadcast_message(message, protocol)
    
    def get_websocket_info(self) -> Dict[str, Any]:
        """获取WebSocket连接信息"""
        if not self.is_protocol_enabled('websocket'):
            return {'enabled': False}
        
        server_config = self.config.get('server', {})
        return {
            'enabled': True,
            'host': server_config.get('ip', '0.0.0.0'),
            'port': server_config.get('port', 8000),
            'path': '/xiaozhi/v1/'
        }
    
    def get_mqtt_info(self) -> Dict[str, Any]:
        """获取MQTT连接信息"""
        if not self.is_protocol_enabled('mqtt'):
            return {'enabled': False}
        
        mqtt_config = self.config.get('mqtt_server', {})
        return {
            'enabled': True,
            'host': mqtt_config.get('host', '0.0.0.0'),
            'port': mqtt_config.get('port', 1883),
            'udp_port': mqtt_config.get('udp_port', 1883),
            'public_ip': mqtt_config.get('public_ip', 'localhost')
        }
    
    def get_connection_info(self) -> Dict[str, Any]:
        """获取所有协议的连接信息"""
        return {
            'websocket': self.get_websocket_info(),
            'mqtt': self.get_mqtt_info(),
            'active_connections': self.get_active_connections_count()
        }