alexei.dolgolyov
82e400ddcd
Chat language: - Added language_code field to TelegramChat model + migration - Saved from message.from.language_code on webhook/polling - Displayed as badge on bot chat cards and target receiver items - Resolved from DB in target API response (works for existing receivers) - Shown in chat picker dropdown (desc includes language) EntitySelect improvements: - Tracker-target link selector shows all targets, already-linked ones appear disabled with "Already linked" hint - Receiver chat picker shows already-added chats as disabled Dev scripts: - scripts/restart-backend.sh and restart-frontend.sh - Updated .claude/docs/dev-servers.md to reference scripts
105 lines
3.7 KiB
Python
105 lines
3.7 KiB
Python
"""Telegram webhook handler for bot commands."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import logging
|
|
from typing import Any
|
|
|
|
import aiohttp
|
|
from fastapi import APIRouter, Depends, Header, HTTPException, Request
|
|
from sqlmodel import select
|
|
from sqlmodel.ext.asyncio.session import AsyncSession
|
|
|
|
from notify_bridge_core.notifications.telegram.client import TelegramClient
|
|
|
|
from ..database.engine import get_session
|
|
from ..database.models import TelegramBot
|
|
from ..services.telegram import save_chat_from_webhook
|
|
from .handler import handle_command, send_media_group, send_reply
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
router = APIRouter(prefix="/api/telegram", tags=["telegram-webhook"])
|
|
|
|
# Webhook secret — set via NOTIFY_BRIDGE_TELEGRAM_WEBHOOK_SECRET env var
|
|
_webhook_secret: str | None = None
|
|
|
|
|
|
def set_webhook_secret(secret: str | None) -> None:
|
|
global _webhook_secret
|
|
_webhook_secret = secret
|
|
|
|
|
|
@router.post("/webhook/{webhook_id}")
|
|
async def telegram_webhook(
|
|
webhook_id: str,
|
|
request: Request,
|
|
x_telegram_bot_api_secret_token: str | None = Header(default=None),
|
|
session: AsyncSession = Depends(get_session),
|
|
):
|
|
"""Handle incoming Telegram messages — route commands to handlers."""
|
|
# Validate webhook secret if configured
|
|
if _webhook_secret:
|
|
if x_telegram_bot_api_secret_token != _webhook_secret:
|
|
raise HTTPException(status_code=403, detail="Invalid webhook secret")
|
|
|
|
# Find bot by opaque webhook path ID (not by token — token must not appear in URLs)
|
|
bot_result = await session.exec(
|
|
select(TelegramBot).where(TelegramBot.webhook_path_id == webhook_id)
|
|
)
|
|
bot = bot_result.first()
|
|
if not bot:
|
|
raise HTTPException(status_code=403, detail="Unknown webhook")
|
|
|
|
try:
|
|
update = await request.json()
|
|
except Exception:
|
|
return {"ok": True, "error": "invalid_json"}
|
|
|
|
message = update.get("message")
|
|
if not message:
|
|
return {"ok": True, "skipped": "no_message"}
|
|
|
|
chat_info = message.get("chat", {})
|
|
chat_id = str(chat_info.get("id", ""))
|
|
text = message.get("text", "")
|
|
|
|
if not chat_id or not text:
|
|
return {"ok": True, "skipped": "empty"}
|
|
|
|
# Auto-persist chat from incoming message
|
|
from_user = message.get("from", {})
|
|
msg_language = from_user.get("language_code", "")
|
|
try:
|
|
await save_chat_from_webhook(session, bot.id, chat_info, language_code=msg_language)
|
|
await session.commit()
|
|
except Exception:
|
|
_LOGGER.warning("Failed to auto-save chat %s", chat_id, exc_info=True)
|
|
|
|
# Handle commands
|
|
if text.startswith("/"):
|
|
message_id = message.get("message_id")
|
|
cmd_response = await handle_command(bot, chat_id, text, language_code=msg_language)
|
|
if cmd_response is not None:
|
|
if isinstance(cmd_response, list):
|
|
await send_media_group(bot.token, chat_id, cmd_response, reply_to_message_id=message_id)
|
|
else:
|
|
await send_reply(bot.token, chat_id, cmd_response, reply_to_message_id=message_id)
|
|
return {"ok": True}
|
|
|
|
return {"ok": True, "skipped": "not_a_command"}
|
|
|
|
|
|
async def register_webhook(bot_token: str, webhook_url: str, secret: str | None = None) -> dict:
|
|
"""Register webhook URL with Telegram Bot API via TelegramClient."""
|
|
async with aiohttp.ClientSession() as http:
|
|
client = TelegramClient(http, bot_token)
|
|
return await client.set_webhook(webhook_url, secret=secret)
|
|
|
|
|
|
async def unregister_webhook(bot_token: str) -> dict:
|
|
"""Remove webhook from Telegram Bot API via TelegramClient."""
|
|
async with aiohttp.ClientSession() as http:
|
|
client = TelegramClient(http, bot_token)
|
|
return await client.delete_webhook()
|