mirror of
https://github.com/xinnan-tech/xiaozhi-esp32-server.git
synced 2026-07-21 22:53:56 +08:00
93 lines
3.4 KiB
Python
93 lines
3.4 KiB
Python
import asyncio
|
|
from aiohttp import web
|
|
from config.logger import setup_logging
|
|
from core.api.ota_handler import OTAHandler
|
|
from core.api.vision_handler import VisionHandler
|
|
|
|
TAG = __name__
|
|
|
|
|
|
class SimpleHttpServer:
|
|
def __init__(self, config: dict):
|
|
self.config = config
|
|
self.logger = setup_logging()
|
|
self.ota_handler = OTAHandler(config)
|
|
self.vision_handler = VisionHandler(config)
|
|
|
|
def _get_websocket_url(self, local_ip: str, port: int) -> str:
|
|
"""获取websocket地址
|
|
|
|
Args:
|
|
local_ip: 本地IP地址
|
|
port: 端口号
|
|
|
|
Returns:
|
|
str: websocket地址
|
|
"""
|
|
server_config = self.config["server"]
|
|
websocket_config = server_config.get("websocket")
|
|
|
|
if websocket_config and "你" not in websocket_config:
|
|
return websocket_config
|
|
else:
|
|
return f"ws://{local_ip}:{port}/xiaozhi/v1/"
|
|
|
|
async def start(self):
|
|
try:
|
|
server_config = self.config["server"]
|
|
read_config_from_api = self.config.get("read_config_from_api", False)
|
|
host = server_config.get("ip", "0.0.0.0")
|
|
port = int(server_config.get("http_port", 8003))
|
|
|
|
if port:
|
|
app = web.Application()
|
|
|
|
if not read_config_from_api:
|
|
# 如果没有开启智控台，只是单模块运行，就需要再添加简单OTA接口，用于下发websocket接口
|
|
app.add_routes(
|
|
[
|
|
web.get("/xiaozhi/ota/", self.ota_handler.handle_get),
|
|
web.post("/xiaozhi/ota/", self.ota_handler.handle_post),
|
|
web.options(
|
|
"/xiaozhi/ota/", self.ota_handler.handle_options
|
|
),
|
|
# 下载接口，仅提供 data/bin/*.bin 下载
|
|
web.get(
|
|
"/xiaozhi/ota/download/{filename}",
|
|
self.ota_handler.handle_download,
|
|
),
|
|
web.options(
|
|
"/xiaozhi/ota/download/{filename}",
|
|
self.ota_handler.handle_options,
|
|
),
|
|
]
|
|
)
|
|
# 添加路由
|
|
app.add_routes(
|
|
[
|
|
web.get("/mcp/vision/explain", self.vision_handler.handle_get),
|
|
web.post(
|
|
"/mcp/vision/explain", self.vision_handler.handle_post
|
|
),
|
|
web.options(
|
|
"/mcp/vision/explain", self.vision_handler.handle_options
|
|
),
|
|
]
|
|
)
|
|
|
|
# 运行服务
|
|
runner = web.AppRunner(app)
|
|
await runner.setup()
|
|
site = web.TCPSite(runner, host, port)
|
|
await site.start()
|
|
|
|
# 保持服务运行
|
|
while True:
|
|
await asyncio.sleep(3600) # 每隔 1 小时检查一次
|
|
except Exception as e:
|
|
self.logger.bind(tag=TAG).error(f"HTTP服务器启动失败: {e}")
|
|
import traceback
|
|
|
|
self.logger.bind(tag=TAG).error(f"错误堆栈: {traceback.format_exc()}")
|
|
raise
|