| import asyncio |
| import logging |
| import time |
| import os |
|
|
| from bot.telegram_api import send_message, edit_message |
| from bot.onboarding import ( |
| handle_start, |
| handle_token_command, |
| handle_reset, |
| handle_help, |
| handle_stats, |
| ) |
| from bot.streaming import stream_reply |
| from db import models as dbm |
|
|
| log = logging.getLogger("webhook") |
|
|
| _last_msg = {} |
| RATE_SECONDS = 2.0 |
|
|
| ADMIN_TELEGRAM_IDS = { |
| int(x.strip()) |
| for x in os.environ.get("ADMIN_TELEGRAM_IDS", "").split(",") |
| if x.strip().isdigit() |
| } |
|
|
|
|
| async def handle_update(update: dict): |
| msg = update.get("message") or update.get("edited_message") |
| if not msg: |
| return |
|
|
| chat = msg.get("chat", {}) |
| chat_id = chat.get("id") |
| from_user = msg.get("from", {}) |
| user_id = from_user.get("id") |
| username = from_user.get("username") |
| first_name = from_user.get("first_name") |
|
|
| if chat_id is None or user_id is None: |
| return |
|
|
| text = (msg.get("text") or "").strip() |
|
|
| if not text: |
| await send_message(chat_id, "⚠️ Only text messages are supported.") |
| return |
|
|
| dbm.upsert_user(user_id, username, first_name) |
|
|
| if text.startswith("/start"): |
| await handle_start(chat_id, user_id, username) |
| return |
|
|
| if text.startswith("/token"): |
| parts = text.split(maxsplit=1) |
| arg = parts[1].strip() if len(parts) > 1 else "" |
| await handle_token_command(chat_id, user_id, arg) |
| return |
|
|
| if text.startswith("/reset"): |
| await handle_reset(chat_id, user_id) |
| return |
|
|
| if text.startswith("/help"): |
| await handle_help(chat_id) |
| return |
|
|
| if text.startswith("/stats"): |
| await handle_stats(chat_id, user_id) |
| return |
|
|
| if text.startswith("/admin"): |
| if user_id in ADMIN_TELEGRAM_IDS: |
| await send_message( |
| chat_id, |
| "**Admin panel:** open the Space URL and switch to the 🔐 Admin Panel tab.", |
| ) |
| else: |
| await send_message(chat_id, "⛔ You are not an admin.") |
| return |
|
|
| now = time.time() |
| last = _last_msg.get(user_id, 0) |
| if now - last < RATE_SECONDS: |
| await send_message(chat_id, "⏳ Slow down a moment.") |
| return |
| _last_msg[user_id] = now |
|
|
| if not dbm.user_has_valid_token(user_id): |
| await send_message( |
| chat_id, |
| "🔒 You need a valid access token.\n" |
| "Send `/token YOUR_TOKEN_HERE` to activate.", |
| ) |
| return |
|
|
| try: |
| daily_limit = int(dbm.get_setting("daily_turn_limit", "100")) |
| except Exception: |
| daily_limit = 100 |
|
|
| user = dbm.get_user(user_id) |
| if user and user.get("total_turns", 0) >= daily_limit: |
| await send_message( |
| chat_id, |
| "⚠️ Daily message limit reached. Try again tomorrow.", |
| ) |
| return |
|
|
| |
| dbm.add_message(user_id, "user", text) |
| dbm.archive_append(user_id, "user", text) |
|
|
| try: |
| await stream_reply(chat_id, user_id, text) |
| except Exception: |
| log.exception("stream_reply crashed") |
| try: |
| await send_message(chat_id, "⚠️ Something went wrong generating a reply.") |
| except Exception: |
| pass |
