tg-spam-guard/handlers/bind.py

"""账号绑定 handler"""
import asyncio
import logging
import re

from telegram import InlineKeyboardButton, InlineKeyboardMarkup, Update
from telegram.ext import ContextTypes, CommandHandler, MessageHandler, filters

from db.models import Action, GameScore
from services.bind_service import bind_with_email_token, unbind, get_binding
from services.message_utils import record_bot_message, safe_reply
from services.time_utils import format_dt
from services.game_service import get_scores_summary

logger = logging.getLogger("spam_guard")

EMAIL_RE = re.compile(r"^[^@\s]+@[^@\s]+\.[^@\s]+$")


async def cmd_bind(update: Update, context: ContextTypes.DEFAULT_TYPE):
    """绑定授权账号。群内引导到私聊；私聊进入等待邮箱+Token 状态。"""
    user = update.effective_user
    logger.info(f"🔗 收到绑定指令: user={user.id}, chat_type={update.effective_chat.type}")
    existing = await get_binding(user.id)
    if existing:
        text = (
            "🔗 你已绑定授权账号\n"
            f"📧 邮箱: {existing.get('email') or existing.get('harvest_uid') or '-'}\n"
            f"⏳ 到期: {existing.get('time_expire') or '-'}\n"
            f"📅 绑定时间: {existing.get('bound_at')}\n\n"
            "如需重新绑定，先 /unbind"
        )
        reply = await safe_reply(update, context, text)
        asyncio.create_task(auto_delete(update.message, seconds=60))
        await record_bot_message(reply)
        asyncio.create_task(auto_delete(reply, seconds=60))
        return

    if update.effective_chat.type != "private":
        me = await context.bot.get_me()
        bot_username = me.username
        url = f"https://t.me/{bot_username}?start=bind"
        keyboard = InlineKeyboardMarkup([[InlineKeyboardButton("去私聊绑定", url=url)]])
        text = (
            "🔐 为了保护 Token，请到私聊完成绑定。\n\n"
            f"👉 点击进入：{url}"
        )
        try:
            reply = await safe_reply(update, context, text, reply_markup=keyboard, disable_web_page_preview=True)
        except Exception as e:
            logger.warning(f"🔗 绑定跳转按钮发送失败，改发普通文本: {type(e).__name__}: {e}")
            reply = await update.effective_chat.send_message(text, disable_web_page_preview=True)
        asyncio.create_task(auto_delete(update.message, seconds=60))
        await record_bot_message(reply)
        asyncio.create_task(auto_delete(reply, seconds=60))
        return

    context.user_data["awaiting_bind"] = True
    await safe_reply(update, context, 
        "🔐 授权绑定\n"
        "请发送邮箱和 Token，支持空格或换行分隔：\n\n"
        "邮箱 Token\n\n"
        "例如：user@example.com xxxxx"
    )


async def cmd_start(update: Update, context: ContextTypes.DEFAULT_TYPE):
    if update.effective_chat.type == "private" and context.args and context.args[0] == "bind":
        await cmd_bind(update, context)


async def handle_private_bind_text(update: Update, context: ContextTypes.DEFAULT_TYPE):
    if update.effective_chat.type != "private" or not context.user_data.get("awaiting_bind"):
        return

    text = (update.message.text or "").strip()
    parts = text.replace("\n", " ").split()
    if len(parts) < 2:
        await safe_reply(update, context, "❌ 格式不对，请发送：邮箱 Token")
        return

    email, token = parts[0].strip(), parts[1].strip()
    if not EMAIL_RE.match(email):
        await safe_reply(update, context, "❌ 邮箱格式不正确，请重新发送：邮箱 Token")
        return

    processing = await safe_reply(update, context, "⏳ 正在验证授权信息...")
    ok, result = await bind_with_email_token(
        tg_user_id=update.effective_user.id,
        tg_username=update.effective_user.username or update.effective_user.first_name,
        email=email,
        token=token,
    )
    context.user_data["awaiting_bind"] = False

    try:
        await processing.delete()
    except Exception:
        pass

    if ok:
        data = result.get("data") or {}
        await Action.create(action="BIND", user_id=update.effective_user.id, username=update.effective_user.username, details={"email": data.get("email")})
        await safe_reply(update, context, 
            "✅ 绑定成功\n"
            f"📧 邮箱: {data.get('email') or email}\n"
            f"⏳ 到期: {data.get('time_expire') or '-'}"
        )
    else:
        context.user_data["awaiting_bind"] = True
        await safe_reply(update, context, f"❌ 验证失败：{result.get('msg') or result.get('reason') or '授权信息无效'}\n请重新发送：邮箱 Token")


async def cmd_me(update: Update, context: ContextTypes.DEFAULT_TYPE):
    """查看当前 TG 用户绑定的授权信息和积分。"""
    user = update.effective_user
    binding = await get_binding(user.id)
    summary = await get_scores_summary(user.id)
    scores = summary["scores"]
    if scores:
        score_lines = []
        for item in scores:
            last = format_dt(item.last_played) if item.last_played else "未参与"
            score_lines.append(f"🎮 {item.game_type}: {item.score} 分｜次数 {item.total}｜连续 {item.streak}｜最近 {last}")
        checkin = summary["by_type"].get("checkin")
        trivia = summary["by_type"].get("trivia")
        checkin_text = "✅ 今日签到：已签到" if checkin and checkin.last_played and checkin.last_played.strftime("%Y-%m-%d") == __import__("datetime").datetime.now().strftime("%Y-%m-%d") else "✅ 今日签到：未签到"
        trivia_text = f"🎮 答题积分：{trivia.score} 分｜正确 {trivia.wins}/{trivia.total}｜连击 {trivia.streak}" if trivia else "🎮 答题积分：暂无"
        score_text = f"📊 总积分：{summary['total']} 分\n{checkin_text}\n{trivia_text}\n" + "\n".join(score_lines)
    else:
        score_text = "📊 积分：暂无记录\n✅ 今日签到：未签到\n🎮 答题积分：暂无\n💡 发送 /checkin 签到，或 /trivia 答题获取积分"

    if not binding:
        text = (
            "👤 我的账号\n"
            "━━━━━━━━━━━━━━━━━━━━\n"
            "🔗 授权状态：未绑定\n"
            f"{score_text}\n\n"
            "绑定授权后，我可以帮你查看授权邮箱和到期时间。\n"
            "请发送 /bind，然后按提示到私聊里提交邮箱和 Token。"
        )
    else:
        text = (
            "👤 我的账号\n"
            "━━━━━━━━━━━━━━━━━━━━\n"
            "🔗 授权状态：已绑定\n"
            f"📧 授权邮箱：{binding.get('email') or '-'}\n"
            f"⏳ 到期时间：{binding.get('time_expire') or '未返回'}\n"
            f"🕒 绑定时间：{format_dt(binding.get('bound_at')) or '-'}\n\n"
            f"{score_text}\n\n"
            "如需更换授权，请先 /unbind，再 /bind 重新绑定。"
        )
    reply = await safe_reply(update, context, text)
    if update.effective_chat.type != "private":
        asyncio.create_task(auto_delete(update.message, seconds=60))
        await record_bot_message(reply)
        asyncio.create_task(auto_delete(reply, seconds=60))


async def cmd_unbind(update: Update, context: ContextTypes.DEFAULT_TYPE):
    """解绑 harvest 账号"""
    user = update.effective_user
    deleted = await unbind(user.id)
    if deleted:
        reply = await safe_reply(update, context, "🔓 已解除账号绑定")
        await Action.create(action="UNBIND", user_id=user.id, username=user.username)
    else:
        reply = await safe_reply(update, context, "❌ 你没有绑定的账号")
    asyncio.create_task(auto_delete(update.message, seconds=60))
    await record_bot_message(reply)
    asyncio.create_task(auto_delete(reply, seconds=60))


async def auto_delete(message, seconds=60):
    try:
        await asyncio.sleep(seconds)
        await message.delete()
    except Exception:
        pass


def register_bind(app):
    app.add_handler(CommandHandler("start", cmd_start, filters.ChatType.PRIVATE))
    app.add_handler(CommandHandler("bind", cmd_bind))
    app.add_handler(CommandHandler("me", cmd_me))
    app.add_handler(CommandHandler("unbind", cmd_unbind))
    app.add_handler(MessageHandler(filters.TEXT & filters.ChatType.PRIVATE & ~filters.COMMAND, handle_private_bind_text), group=0)