from __future__ import annotations import logging import secrets import time logger = logging.getLogger(__name__) class QQBotPairing: def __init__(self): self._codes: dict[str, tuple[str, float]] = {} self._code_ttl = 300 self._id_label = "QQ OpenID" @property def id_label(self) -> str: return self._id_label async def generate_code(self, peer_id: str) -> str: code = secrets.token_hex(3).upper() self._codes[peer_id] = (code, time.time()) logger.info("Pairing code generated for peer=%s: %s", peer_id, code) return code async def verify_code(self, peer_id: str, code: str) -> bool: entry = self._codes.get(peer_id) if entry is None: return False stored_code, created_at = entry if time.time() - created_at > self._code_ttl: del self._codes[peer_id] return False if stored_code.upper() == code.upper(): del self._codes[peer_id] return True return False async def notify_approval(self, config: dict, peer_id: str, account_id: str | None = None) -> None: logger.info("Pairing approval notification for peer=%s", peer_id) def normalize_allow_entry(self, entry: str) -> str: if entry.startswith("qqbot:"): return entry.split(":", 1)[1] if len(entry.split(":", 1)) >= 2 else entry return entry