import logging import os import re import aiohttp from yuxi.channel.extensions.twitch.config import TwitchAccountConfig logger = logging.getLogger(__name__) def resolve_twitch_token(account: TwitchAccountConfig, account_id: str = "default") -> tuple[str, str]: token = account.access_token if token: return normalize_twitch_token(token), "config" if account_id == "default": env_token = os.getenv("OPENCLAW_TWITCH_ACCESS_TOKEN") or os.getenv("TWITCH_ACCESS_TOKEN") if env_token: return normalize_twitch_token(env_token), "env" return "", "none" def normalize_twitch_token(token: str) -> str: token = token.strip() if not token.lower().startswith("oauth:"): token = f"oauth:{token}" return token def denormalize_token(token: str) -> str: if token.lower().startswith("oauth:"): return token[6:] return token def is_oauth_token(token: str) -> bool: return bool(token) and token.lower().startswith("oauth:") def extract_access_token(account: TwitchAccountConfig) -> str: token = account.access_token return denormalize_token(token) def detect_mentions(text: str, bot_username: str) -> bool: mentions = re.findall(r"@(\w+)", text) bot_lower = bot_username.lower() return any(m.lower() == bot_lower for m in mentions) async def validate_twitch_token(account: TwitchAccountConfig) -> dict: token, _ = resolve_twitch_token(account) bare = denormalize_token(token) if not bare: return {"valid": False, "reason": "no token configured"} url = "https://id.twitch.tv/oauth2/validate" headers = {"Authorization": f"OAuth {bare}"} try: async with aiohttp.ClientSession() as session: async with session.get(url, headers=headers, timeout=aiohttp.ClientTimeout(total=10)) as resp: if resp.status == 200: data = await resp.json() return {"valid": True, **data} return {"valid": False, "status": resp.status, "message": await resp.text()} except Exception as e: return {"valid": False, "error": str(e)} async def refresh_twitch_token(account: TwitchAccountConfig) -> dict: if not account.refresh_token or not account.client_id or not account.client_secret: return {"ok": False, "error": "missing refresh_token, client_id, or client_secret"} url = "https://id.twitch.tv/oauth2/token" params = { "grant_type": "refresh_token", "refresh_token": account.refresh_token, "client_id": account.client_id, "client_secret": account.client_secret, } try: async with aiohttp.ClientSession() as session: async with session.post(url, json=params, timeout=aiohttp.ClientTimeout(total=10)) as resp: data = await resp.json() if resp.status == 200: return {"ok": True, **data} return {"ok": False, "error": data.get("message", f"HTTP {resp.status}")} except Exception as e: return {"ok": False, "error": str(e)}