Kris
552aef767c
新增QQ Bot适配器完整代码栈，包含： 1. 基础适配器入口与工具类封装 2. 会话管理、重试队列与流量控制 3. 命令系统与内置指令（ping/help/status等） 4. 富媒体消息处理与格式转换 5. 引用存储与审批管理 6. 凭证备份与会话持久化 7. 健康检查与交互回调系统
118 lines
4.2 KiB
Python
118 lines
4.2 KiB
Python
from __future__ import annotations
|
|
|
|
import asyncio
|
|
import time
|
|
|
|
import aiohttp
|
|
|
|
from yuxi.channels.exceptions import ChannelAuthenticationError
|
|
from yuxi.utils.logging_config import logger
|
|
|
|
|
|
class QQBotTokenManager:
|
|
def __init__(
|
|
self,
|
|
app_id: str,
|
|
app_secret: str,
|
|
sandbox: bool = False,
|
|
http_client: aiohttp.ClientSession | None = None,
|
|
):
|
|
self.app_id = app_id
|
|
self.app_secret = app_secret
|
|
self.sandbox = sandbox
|
|
self._http_client = http_client
|
|
self._access_token: str | None = None
|
|
self._expires_at: float | None = None
|
|
self._token_lock = asyncio.Lock()
|
|
self._refresh_in_progress: asyncio.Event | None = None
|
|
self._refresh_task: asyncio.Task | None = None
|
|
self._refresh_interval: float = 60.0
|
|
|
|
@property
|
|
def api_base(self) -> str:
|
|
if self.sandbox:
|
|
return "https://sandbox.api.sgroup.qq.com"
|
|
return "https://api.sgroup.qq.com"
|
|
|
|
async def get_token(self) -> str:
|
|
async with self._token_lock:
|
|
if self._is_expired():
|
|
await self._do_refresh()
|
|
return self._access_token
|
|
|
|
async def force_refresh(self) -> str:
|
|
async with self._token_lock:
|
|
await self._do_refresh()
|
|
return self._access_token
|
|
|
|
async def _do_refresh(self) -> None:
|
|
if self._refresh_in_progress is not None:
|
|
await self._refresh_in_progress.wait()
|
|
return
|
|
|
|
self._refresh_in_progress = asyncio.Event()
|
|
try:
|
|
await self._refresh()
|
|
self._refresh_in_progress.set()
|
|
except Exception:
|
|
self._refresh_in_progress.set()
|
|
raise
|
|
finally:
|
|
self._refresh_in_progress = None
|
|
|
|
async def _refresh(self) -> None:
|
|
client = self._http_client or aiohttp.ClientSession()
|
|
try:
|
|
async with client.post(
|
|
f"{self.api_base}/oauth2/token",
|
|
json={
|
|
"app_id": self.app_id,
|
|
"app_secret": self.app_secret,
|
|
},
|
|
) as resp:
|
|
if resp.status != 200:
|
|
raise ChannelAuthenticationError(f"Token refresh failed: HTTP {resp.status}")
|
|
data = await resp.json()
|
|
self._access_token = data["access_token"]
|
|
expires_in = data.get("expires_in", 7200)
|
|
self._expires_at = time.monotonic() + expires_in
|
|
logger.info(f"[QQBot] Token refreshed, expires in {expires_in}s (app_id={self.app_id[:6]}...)")
|
|
finally:
|
|
if not self._http_client:
|
|
await client.close()
|
|
|
|
def start_background_refresh(self) -> None:
|
|
if self._refresh_task is not None and not self._refresh_task.done():
|
|
return
|
|
self._refresh_task = asyncio.create_task(self._background_refresh_loop())
|
|
logger.debug(f"[QQBot] Background token refresh started (interval={self._refresh_interval}s)")
|
|
|
|
def stop_background_refresh(self) -> None:
|
|
if self._refresh_task and not self._refresh_task.done():
|
|
self._refresh_task.cancel()
|
|
self._refresh_task = None
|
|
logger.debug("[QQBot] Background token refresh stopped")
|
|
|
|
async def _background_refresh_loop(self) -> None:
|
|
while True:
|
|
try:
|
|
await asyncio.sleep(self._refresh_interval)
|
|
async with self._token_lock:
|
|
if not self._is_expired():
|
|
continue
|
|
try:
|
|
await self._do_refresh()
|
|
except Exception as e:
|
|
logger.warning(f"[QQBot] Background token refresh failed (will retry): {e}")
|
|
except asyncio.CancelledError:
|
|
logger.debug("[QQBot] Background token refresh cancelled")
|
|
break
|
|
except Exception as e:
|
|
logger.error(f"[QQBot] Background token refresh loop error: {e}")
|
|
await asyncio.sleep(5)
|
|
|
|
def _is_expired(self) -> bool:
|
|
if self._access_token is None or self._expires_at is None:
|
|
return True
|
|
return time.monotonic() > self._expires_at - 300
|