import asyncio
import datetime
import sqlite3
import random
import requests
from zoneinfo import ZoneInfo
from typing import Optional, Tuple, List
from telethon import TelegramClient, events, Button, types

# ========== 配置区 ==========
API_ID = 2040
API_HASH = "b18441a1ff607e10a989891a5462e627"
BOT_TOKEN = "8451724418:AAGTGqCmc1JiUr88IABhMiQTHeVLcAcnT5Y"
DB_PATH = "sign.db"

# 积分配置
SIGN_POINTS = 10
INVITE_POINTS = 10
DAILY_SPEAK_TOP_N = 10
DAILY_SPEAK_REWARD = 10

# 允许机器人运行的群组（只处理这些群的消息）
ALLOWED_GROUPS = [-1003238845008]

# 时区（结算规则用）
LOCAL_TZ = ZoneInfo("America/New_York")

# 代理（可选）
PROXY = {
    'proxy_type': "socks5",
    'addr': "202.155.144.102",
    'port': 31102,
    'username': "SyNuejCtrQ",
    'password': "MH8ioL7EXf"
}

INVITE_LINK = "https://www.websea.my/en/signup?key=77346588"

crypto_currencies = {
    "BTC": {"type": "BTC-USDT", "url": "https://www.websea.com/zh-CN/trade/BTC-USDT"},
    "ETH": {"type": "ETH-USDT", "url": "https://www.websea.com/zh-CN/trade/ETH-USDT"},
    "SOL": {"type": "SOL-USDT", "url": "https://www.websea.com/zh-CN/trade/SOL-USDT"},
    "BNB": {"type": "BNB-USDT", "url": "https://www.websea.com/zh-CN/trade/BNB-USDT"},
    "WBS": {"type": "WBS-USDT", "url": "https://www.websea.com/zh-CN/trade/WBS-USDT"}
}

main_buttons = [
    [Button.inline("Sign in", b"/sign"), Button.inline("Today's Top Speakers", b"/daily_rank")],
    [Button.inline("My invitation", b"/my_invites"), Button.inline("coin price", b"/btc")],
    [Button.inline("Assistance", b"/help")]
]

COMMANDS_TEXT = {
    '/sign': 'Daily check-in, available once per day to earn points',
    '/daily_rank': 'View today\'s message activity ranking',
    '/my_invites': 'View the number of people you have invited',
    '/btc': 'Get the Bitcoin price menu',
    '/help': 'Show all available commands and descriptions'
}


# ============================

# ---------- 全局并发控制 ----------
# 使用 asyncio.Lock 在同一时间避免大量数据库并发写入（与 to_thread 配合）
DB_LOCK = asyncio.Lock()


# ---------- SQLite 同步底层实现（运行在线程池） ----------
def _get_conn():
    # 每次操作创建连接，避免 sqlite 的线程/并发问题
    # 不设置 check_same_thread：因为连接会在同一线程（to_thread）中被使用
    return sqlite3.connect(DB_PATH, detect_types=sqlite3.PARSE_DECLTYPES | sqlite3.PARSE_COLNAMES)


def _init_db_sync():
    conn = _get_conn()
    cursor = conn.cursor()
    cursor.execute('''
                   CREATE TABLE IF NOT EXISTS users
                   (
                       user_id
                       INTEGER
                       PRIMARY
                       KEY,
                       username
                       TEXT,
                       points
                       INTEGER
                       DEFAULT
                       0,
                       last_sign_date
                       TEXT,
                       total_invites
                       INTEGER
                       DEFAULT
                       0
                   )
                   ''')
    cursor.execute('''
                   CREATE TABLE IF NOT EXISTS daily_messages
                   (
                       user_id
                       INTEGER,
                       date
                       TEXT,
                       count
                       INTEGER
                       DEFAULT
                       0,
                       PRIMARY
                       KEY
                   (
                       user_id,
                       date
                   )
                       )
                   ''')
    cursor.execute('''
                   CREATE TABLE IF NOT EXISTS invites
                   (
                       inviter_id
                       INTEGER,
                       invitee_id
                       INTEGER,
                       created_at
                       TEXT,
                       PRIMARY
                       KEY
                   (
                       inviter_id,
                       invitee_id
                   )
                       )
                   ''')
    cursor.execute('''
                   CREATE TABLE IF NOT EXISTS points_log
                   (
                       id
                       INTEGER
                       PRIMARY
                       KEY
                       AUTOINCREMENT,
                       user_id
                       INTEGER,
                       points
                       INTEGER,
                       reason
                       TEXT,
                       created_at
                       TEXT
                   )
                   ''')
    conn.commit()
    conn.close()


def _add_or_update_user_sync(user_id: int, username: str):
    conn = _get_conn()
    cursor = conn.cursor()
    cursor.execute("INSERT OR IGNORE INTO users (user_id, username) VALUES (?, ?)", (user_id, username))
    cursor.execute("UPDATE users SET username = ? WHERE user_id = ?", (username, user_id))
    conn.commit()
    conn.close()


def _get_last_sign_date_sync(user_id: int) -> Optional[str]:
    conn = _get_conn()
    cursor = conn.cursor()
    cursor.execute("SELECT last_sign_date FROM users WHERE user_id = ?", (user_id,))
    row = cursor.fetchone()
    conn.close()
    return row[0] if row else None


def _add_points_immediate_sync(user_id: int, points: int, reason: str = ""):
    now = datetime.datetime.now(LOCAL_TZ).isoformat()
    conn = _get_conn()
    cursor = conn.cursor()
    # 如果用户可能不存在，先插入默认记录（避免 update 失败）
    cursor.execute("INSERT OR IGNORE INTO users (user_id, username, points) VALUES (?, ?, 0)", (user_id, None))
    cursor.execute("UPDATE users SET points = COALESCE(points,0) + ? WHERE user_id = ?", (points, user_id))
    cursor.execute("INSERT INTO points_log (user_id, points, reason, created_at) VALUES (?, ?, ?, ?)",
                   (user_id, points, reason, now))
    conn.commit()
    conn.close()


def _get_points_sync(user_id: int) -> int:
    conn = _get_conn()
    cursor = conn.cursor()
    cursor.execute("SELECT points FROM users WHERE user_id = ?", (user_id,))
    row = cursor.fetchone()
    conn.close()
    return row[0] if row else 0


def _set_last_sign_sync(user_id: int):
    today = datetime.datetime.now(LOCAL_TZ).date().isoformat()
    conn = _get_conn()
    cursor = conn.cursor()
    cursor.execute("INSERT OR IGNORE INTO users (user_id, username) VALUES (?, ?)", (user_id, None))
    cursor.execute("UPDATE users SET last_sign_date = ? WHERE user_id = ?", (today, user_id))
    conn.commit()
    conn.close()


def _insert_invite_sync(inviter_id: int, invitee_id: int) -> bool:
    """
    插入邀请记录，返回 True 表示插入成功（即第一次邀请）
    如果已存在则抛出 sqlite3.IntegrityError 或返回 False
    """
    now = datetime.datetime.now(LOCAL_TZ).isoformat()
    conn = _get_conn()
    cursor = conn.cursor()
    try:
        cursor.execute("INSERT INTO invites (inviter_id, invitee_id, created_at) VALUES (?, ?, ?)",
                       (inviter_id, invitee_id, now))
        # 更新邀请计数
        cursor.execute("INSERT OR IGNORE INTO users (user_id, username) VALUES (?, ?)", (inviter_id, None))
        cursor.execute("UPDATE users SET total_invites = COALESCE(total_invites,0) + 1 WHERE user_id = ?",
                       (inviter_id,))
        conn.commit()
        return True
    except sqlite3.IntegrityError:
        # 已存在
        conn.rollback()
        return False
    finally:
        conn.close()


def _get_invite_count_sync(user_id: int) -> int:
    conn = _get_conn()
    cursor = conn.cursor()
    cursor.execute("SELECT COUNT(*) FROM invites WHERE inviter_id = ?", (user_id,))
    count = cursor.fetchone()[0]
    conn.close()
    return count


def _add_message_count_sync(user_id: int):
    today = datetime.datetime.now(LOCAL_TZ).date().isoformat()
    conn = _get_conn()
    cursor = conn.cursor()
    cursor.execute('''
                   INSERT INTO daily_messages (user_id, date, count)
                   VALUES (?, ?, 1) ON CONFLICT(user_id, date) DO
                   UPDATE SET count = count + 1
                   ''', (user_id, today))
    conn.commit()
    conn.close()


def _get_daily_message_ranking_sync(date_iso: Optional[str] = None, limit: Optional[int] = None) -> List[
    Tuple[int, Optional[str], int]]:
    date_iso = date_iso or datetime.datetime.now(LOCAL_TZ).date().isoformat()
    conn = _get_conn()
    cursor = conn.cursor()
    query = '''
            SELECT u.user_id, u.username, d.count
            FROM daily_messages d
                     JOIN users u ON d.user_id = u.user_id
            WHERE d.date = ?
            ORDER BY d.count DESC \
            '''
    if limit:
        query += f" LIMIT {limit}"
    cursor.execute(query, (date_iso,))
    rows = cursor.fetchall()
    conn.close()
    return rows


def _reset_daily_messages_sync(date_iso: Optional[str] = None):
    date_iso = date_iso or datetime.datetime.now(LOCAL_TZ).date().isoformat()
    conn = _get_conn()
    cursor = conn.cursor()
    cursor.execute("DELETE FROM daily_messages WHERE date = ?", (date_iso,))
    conn.commit()
    conn.close()


# ---------- 异步包装（供 bot 使用） ----------
async def init_db():
    # 初始化 DB（放在 to_thread 中）
    await asyncio.to_thread(_init_db_sync)


async def add_or_update_user(user_id: int, username: Optional[str]):
    username = username or None
    # use lock to avoid many threads trying to create the file at once
    async with DB_LOCK:
        await asyncio.to_thread(_add_or_update_user_sync, user_id, username)


async def can_sign(user_id: int) -> bool:
    async with DB_LOCK:
        last = await asyncio.to_thread(_get_last_sign_date_sync, user_id)
    today = datetime.datetime.now(LOCAL_TZ).date().isoformat()
    return False if last and last == today else True


async def add_points_immediate(user_id: int, points: int, reason: str = ""):
    async with DB_LOCK:
        await asyncio.to_thread(_add_points_immediate_sync, user_id, points, reason)


async def get_points(user_id: int) -> int:
    async with DB_LOCK:
        return await asyncio.to_thread(_get_points_sync, user_id)


async def set_last_sign(user_id: int):
    async with DB_LOCK:
        await asyncio.to_thread(_set_last_sign_sync, user_id)


async def inc_invite(inviter_id: int, invitee_id: int):
    """记录邀请关系并奖励 inviter（只奖励第一次记录）"""
    if not inviter_id or not invitee_id:
        print(f"[invite] 无效的邀请者或受邀者ID: inviter={inviter_id}, invitee={invitee_id}")
        return False

    if inviter_id == invitee_id:
        print(f"[invite] 不能邀请自己: user_id={inviter_id}")
        return False

    # 使用锁，保证并发情况下不会重复奖励
    async with DB_LOCK:
        inserted = await asyncio.to_thread(_insert_invite_sync, inviter_id, invitee_id)

    if inserted:
        # 插入成功后再给积分（不在 DB_LOCK 内执行，将奖励记录写进 DB）
        try:
            await add_points_immediate(inviter_id, INVITE_POINTS, reason="invite_reward")
            print(
                f"[invite] 成功记录邀请并奖励积分: inviter={inviter_id}, invitee={invitee_id}, points={INVITE_POINTS}")
            return True
        except Exception as e:
            print(f"[invite] 奖励积分失败: inviter={inviter_id}, error={e}")
            return False
    else:
        print(f"[invite] 邀请记录已存在（重复邀请）: inviter={inviter_id}, invitee={invitee_id}")
        return False


async def get_invite_count(user_id: int) -> int:
    async with DB_LOCK:
        return await asyncio.to_thread(_get_invite_count_sync, user_id)


async def add_message_count(user_id: int):
    async with DB_LOCK:
        await asyncio.to_thread(_add_message_count_sync, user_id)


async def get_daily_message_ranking(date_iso: Optional[str] = None, limit: Optional[int] = None):
    async with DB_LOCK:
        return await asyncio.to_thread(_get_daily_message_ranking_sync, date_iso, limit)


async def reset_daily_messages(date_iso: Optional[str] = None):
    async with DB_LOCK:
        await asyncio.to_thread(_reset_daily_messages_sync, date_iso)


# ---------- 币价获取（阻塞 requests -> 放到线程池） ----------
def _get_price_sync(symbol: str) -> Tuple[float, float, float, float]:
    headers = {'user-agent': 'Mozilla/5.0'}
    tz = LOCAL_TZ
    today = datetime.datetime.now(tz).date()
    start_of_day = datetime.datetime.combine(today, datetime.time.min, tzinfo=tz)
    end_of_day = datetime.datetime.combine(today, datetime.time.max, tzinfo=tz)
    params = {
        'symbol': symbol,
        'period': '30min',
        'start': int(start_of_day.timestamp()),
        'end': int(end_of_day.timestamp()),
    }
    try:
        response = requests.get('https://eapi.websea.com/webApi/market/getKline', params=params, headers=headers,
                                timeout=10)
        data = response.json().get("result", {}).get("data", [])
        if not data:
            return 0.0, 0.0, 0.0, 0.0
        current_price = float(data[0]['close'])
        today_high = max(float(i['high']) for i in data)
        today_low = min(float(i['low']) for i in data)
        price_24h_ago = float(data[-1]['open'])
        change_24h = (current_price - price_24h_ago) / price_24h_ago * 100 if price_24h_ago != 0 else 0.0
        return current_price, change_24h, today_high, today_low
    except Exception as e:
        print("获取币价失败:", e)
        return 0.0, 0.0, 0.0, 0.0


async def get_price(symbol: str):
    return await asyncio.to_thread(_get_price_sync, symbol)


def get_price_msg_from_values(name: str, current_price: float, change_24h: float, today_high: float, today_low: float,
                              url: str):
    msg = (
        f"📊 {name}/USDT Real-time Quotes\n\n"
        f"💰 Current Price：{current_price:.4f} USDT\n"
        f"📈 24-hour price change：{change_24h:.2f}%\n"
        f"⬆️ Highest Price：{today_high:.4f} USDT\n"
        f"⬇️ Lowest Price：{today_low:.4f} USDT\n\n"
        f"🔗 Transaction Link：{url}\n"
        f"🎁 Registration Invitation：{INVITE_LINK}\n\n"
        f"（This message is automatically updated hourly.）"
    )
    return msg


async def get_price_msg(symbol: str):
    current_price, change_24h, today_high, today_low = await get_price(symbol)
    name = symbol.split('-')[0]
    url = crypto_currencies.get(name, {}).get("url", "#")
    return get_price_msg_from_values(name, current_price, change_24h, today_high, today_low, url)


def get_crypto_buttons():
    buttons = []
    row = []
    for name in crypto_currencies.keys():
        row.append(Button.inline(name, f"price_{name}".encode()))
        if len(row) == 3:
            buttons.append(row)
            row = []
    if row:
        buttons.append(row)
    buttons.append([Button.inline("Return to Menu", b"/help")])
    return buttons


# ---------- 命令帮助文本 ----------
def get_command_list_text():
    msg = "🤖 Robot Command List：\n"
    for cmd, desc in COMMANDS_TEXT.items():
        msg += f"{cmd} → {desc}\n"
    msg += f"\nSign-in Reward: {SIGN_POINTS} points each time, and 100 points can be redeemed for a 5 USDT futures fee voucher.\nChat Reward: The top {DAILY_SPEAK_TOP_N} users in daily chat activity each receive {DAILY_SPEAK_REWARD} points.\nInvitation Reward: Each invited user grants {INVITE_POINTS} points.\nDaily Settlement Time: Every day at 12:00 PM (America/New_York timezone)."
    return msg


# ---------- 事件处理（异步） ----------
async def handle_command(event, cmd: str):
    user = await event.get_sender()
    if user is None:
        return
    user_id = user.id
    username = user.username or user.first_name or "未知用户"
    await add_or_update_user(user_id, username)

    reply = ""
    # 统一处理命令名（可能传进来已经小写）
    if cmd in ["/sign", "sign in","/tanda"]:
        if not await can_sign(user_id):
            reply = f"🌞 @{username}，You've already signed in today.！"
        else:
            await add_points_immediate(user_id, SIGN_POINTS, reason="sign")
            await set_last_sign(user_id)
            total = await get_points(user_id)
            reply = f"✅ @{username} Check-in successful！obtain {SIGN_POINTS} Points\nCurrent total points：{total}\n\n100 points can be redeemed for a 5 USDT contract fee voucher.。"
    elif cmd in ["/daily_rank", "rankings","/kedudukan_harian"]:
        ranking = await get_daily_message_ranking()
        if not ranking:
            reply = "📊 No one has posted yet today.～"
        else:
            reply_lines = []
            for i, (uid, uname, cnt) in enumerate(ranking):
                display = f"@{uname}" if uname else str(uid)
                reply_lines.append(f"{i + 1}. {display}: {cnt} 条")
            reply = "📊 Today's Top Speakers：\n" + "\n".join(reply_lines)
    elif cmd in ["/my_invites", "invitation","/daily_position"]:
        count = await get_invite_count(user_id)
        total_pts = await get_points(user_id)
        reply = f"👥 @{username}，You have invited {count} person。\nCurrent points：{total_pts}"
    elif cmd in ["/btc"]:
        await event.respond("Please select the currency you wish to view：", buttons=get_crypto_buttons())
        return
    else:
        await event.reply(get_command_list_text(), buttons=main_buttons)
        return

    await event.reply(reply)


async def handle_price_command(event, symbol_name: str):
    symbol = crypto_currencies[symbol_name]["type"]
    msg = await get_price_msg(symbol)
    try:
        # CallbackQuery 用 edit 更友好
        await event.edit(msg, buttons=get_crypto_buttons(), link_preview=False)
    except Exception:
        await event.respond(msg, buttons=get_crypto_buttons(), link_preview=False)


# ---------- 定时任务：每小时推送随机币价 ----------
async def send_price_periodically(bot, chat_id):
    await asyncio.sleep(5)  # 启动缓冲
    while True:
        try:
            random_key = random.choice(list(crypto_currencies.keys()))
            msg = await get_price_msg(crypto_currencies[random_key]["type"])
            await bot.send_message(chat_id, msg, buttons=get_crypto_buttons(), link_preview=False)
        except Exception as e:
            print(f"发送价格失败: {e}")
        # 等待下一小时整点（保持每小时推送一次）
        now = datetime.datetime.now(LOCAL_TZ)
        next_hour = (now.replace(minute=0, second=0, microsecond=0) + datetime.timedelta(hours=1))
        wait_seconds = (next_hour - now).total_seconds()
        await asyncio.sleep(wait_seconds)


# ---------- 定时任务：每天结算（每天 12:00 America/New_York） ----------
async def daily_settlement_task(bot):
    """
    每天在 LOCAL_TZ 12:00 执行：
     - 计算当日发言排名，给前 N 名每人奖励
     - 记录日志并清空当天的发言统计
    """
    while True:
        now = datetime.datetime.now(LOCAL_TZ)
        # 目标时间：今天 12:00（如果已过则到明天）
        target = now.replace(hour=12, minute=0, second=0, microsecond=0)
        if now >= target:
            target = target + datetime.timedelta(days=1)
        wait_seconds = (target - now).total_seconds()
        print(f"[settlement] Next settlement（12:00）in {target.isoformat()}，Waiting {int(wait_seconds)} s")
        await asyncio.sleep(wait_seconds)

        # 执行结算（结算 target 的前一天）
        date_iso = (target - datetime.timedelta(days=1)).date().isoformat()
        print(f"[settlement] Execute settlement, target date：{date_iso}")
        ranking = await get_daily_message_ranking(date_iso=date_iso, limit=DAILY_SPEAK_TOP_N)
        if ranking:
            for i, (uid, uname, cnt) in enumerate(ranking):
                try:
                    await add_points_immediate(uid, DAILY_SPEAK_REWARD, reason=f"daily_speak_rank_{i + 1}")
                except Exception as e:
                    print(f"[settlement] For users {uid} Failed to issue rewards: {e}")
            # 给管理员或群组发送结算通知（可选）
            try:
                summary_lines = [f"📅 {date_iso} Speaker Ranking Settlement Results："]
                for i, (uid, uname, cnt) in enumerate(ranking):
                    display = f"@{uname}" if uname else str(uid)
                    summary_lines.append(f"{i + 1}. {display} — {cnt} Article — obtain {DAILY_SPEAK_REWARD} Points")
                summary = "\n".join(summary_lines)
                for gid in ALLOWED_GROUPS:
                    await bot.send_message(gid, summary)
            except Exception as e:
                print("[settlement] Failed to send settlement notification:", e)
        else:
            print("[settlement] No speaking records for the day; skip speaking rewards.")

        # 清空那天的统计
        try:
            await reset_daily_messages(date_iso=date_iso)
            print(f"[settlement] 清理完成：{date_iso}")
        except Exception as e:
            print("[settlement] 清理 daily_messages 失败:", e)


# ---------- 主逻辑 ----------
async def main():
    await init_db()

    # 使用代理时传 proxy=PROXY，若不需要代理则移除该参数
    # Telethon proxy 格式可能和你的代理不同，请按需调整（或移除）
    bot = TelegramClient('bot_session', API_ID, API_HASH, proxy=PROXY)
    await bot.start(bot_token=BOT_TOKEN)

    # 启动每小时发送币价任务（对每个白名单群）
    for gid in ALLOWED_GROUPS:
        asyncio.create_task(send_price_periodically(bot, gid))

    # 启动每日结算任务（12:00 America/New_York）
    asyncio.create_task(daily_settlement_task(bot))

    @bot.on(events.ChatAction)
    async def welcome(event: events.ChatAction.Event):
        # 欢迎新成员并尝试记录邀请者（若存在）
        if not (event.user_added or event.user_joined):
            return

        # 尝试获取 inviter id（兼容多种 Telethon 结构）
        inviter_id = None
        try:
            # 优先 actor_id（某些 Telethon 版本和场景）
            if hasattr(event, "actor_id") and event.actor_id:
                inviter_id = event.actor_id
            # 再尝试 action_message.from_id（另一些版本）
            elif hasattr(event, "action_message") and event.action_message:
                maybe = getattr(event.action_message, "from_id", None)
                if isinstance(maybe, types.PeerUser):
                    inviter_id = maybe.user_id
                elif isinstance(maybe, int):
                    inviter_id = maybe
                # 在某些情形下 from_id 是 a User 或 MessageEntity，处理兼容性
                elif hasattr(maybe, "user_id"):
                    inviter_id = getattr(maybe, "user_id", None)
        except Exception as e:
            print(f"[welcome] 获取邀请者ID时出错: {e}")
            inviter_id = None

        # 对 event.users 中的新成员逐个处理
        for user in event.users:
            try:
                new_uid = user.id
                new_name = user.username or user.first_name or "新成员"

                # 保证用户存在于 users 表中
                await add_or_update_user(new_uid, new_name)

                # 如果找到邀请者并且不是自己，则记录邀请
                if inviter_id and inviter_id != new_uid:
                    # 确保 inviter 在 users 表中（用户名未知时填 None）
                    await add_or_update_user(inviter_id, None)

                    # 调用 inc_invite 并检查返回值
                    success = await inc_invite(inviter_id, new_uid)
                    if success:
                        print(f"[welcome] 成功处理邀请: {inviter_id} 邀请了 {new_uid}")
                    else:
                        print(f"[welcome] 邀请处理失败或已存在: {inviter_id} -> {new_uid}")
                elif not inviter_id:
                    print(f"[welcome] 未找到邀请者，新成员 {new_uid} 可能是自己加入的")
                else:
                    print(f"[welcome] 跳过自己邀请自己: {new_uid}")

                # 欢迎消息
                try:
                    await event.reply(f"欢迎 @{new_name} 加入群聊！\n{get_command_list_text()}")
                except Exception:
                    # 某些场景 event.reply 不允许（例如系统消息），忽略
                    pass

            except Exception as e:
                print(f"[welcome] 处理新成员异常: user_id={user.id if user else 'unknown'}, error={e}")

    @bot.on(events.NewMessage)
    async def on_message(event):
        # 只处理白名单群组
        if event.chat_id not in ALLOWED_GROUPS:
            return

        text = (event.raw_text or "").strip()
        if not text:
            return

        # 去掉命令末尾的 @botname
        if "@" in text and text.startswith("/"):
            text = text.split("@")[0]

        lowered = text.lower()

        # 命令集合（注意使用小写比较）
        cmds_map = {
            "签到": ["/sign", "签到", "/tanda"],
            "发言": ["/daily_rank", "发言", "/kedudukan_harian"],
            "邀请": ["/my_invites", "邀请", "/daily_position"],
            "币价": ["/btc", "币价"]
        }

        for group_cmds in cmds_map.values():
            if lowered in [c.lower() for c in group_cmds]:
                # 传入原始命令（保持原有行为）
                await handle_command(event, lowered)
                return

        # 非命令，记录发言次数
        sender = await event.get_sender()
        if sender:
            await add_or_update_user(sender.id, sender.username or sender.first_name or "未知用户")
            await add_message_count(sender.id)

    @bot.on(events.CallbackQuery)
    async def callback(event):
        # 快速响应 UI
        try:
            await event.answer("Currently being processed...", alert=False)
        except Exception:
            pass

        cmd = event.data.decode() if event.data else ""
        if "@" in cmd and cmd.startswith("/"):
            cmd = cmd.split("@")[0]

        if cmd.startswith("price_"):
            symbol = cmd.split("_", 1)[1]
            await handle_price_command(event, symbol)
        else:
            await handle_command(event, cmd)

    print("🤖 机器人已启动，等待群聊消息...")
    await bot.run_until_disconnected()


if __name__ == "__main__":
    try:
        asyncio.run(main())
    except KeyboardInterrupt:
        print("机器人已停止")
