mirror of
https://github.com/BEDOLAGA-DEV/remnawave-bedolaga-telegram-bot.git
synced 2026-02-09 21:50:37 +00:00
c868ef3b69
- Added a new handler to delete ban notifications upon user interaction. - Introduced a delete button in ban notifications for better user experience. - Updated ban notification messages to include node information more prominently. - Refactored the BanNotificationService to send messages with the delete button included.
145 lines
5.8 KiB
Python
145 lines
5.8 KiB
Python
"""
|
||
API эндпоинты для приема уведомлений от ban системы
|
||
"""
|
||
from __future__ import annotations
|
||
|
||
import logging
|
||
|
||
from fastapi import APIRouter, Depends, HTTPException, status
|
||
from sqlalchemy.ext.asyncio import AsyncSession
|
||
|
||
from app.webapi.dependencies import get_db_session, require_api_token
|
||
from app.webapi.schemas.ban_notifications import (
|
||
BanNotificationRequest,
|
||
BanNotificationResponse,
|
||
)
|
||
from app.services.ban_notification_service import ban_notification_service
|
||
|
||
|
||
logger = logging.getLogger(__name__)
|
||
|
||
router = APIRouter()
|
||
|
||
|
||
@router.post(
|
||
"/send",
|
||
response_model=BanNotificationResponse,
|
||
summary="Отправить уведомление от ban системы",
|
||
description=(
|
||
"Эндпоинт для отправки уведомлений пользователям от системы мониторинга ban. "
|
||
"Поддерживает уведомления о блокировке, разблокировке и предупреждения."
|
||
),
|
||
)
|
||
async def send_ban_notification(
|
||
request: BanNotificationRequest,
|
||
db: AsyncSession = Depends(get_db_session),
|
||
_token=Depends(require_api_token),
|
||
) -> BanNotificationResponse:
|
||
"""
|
||
Отправить уведомление пользователю от ban системы
|
||
|
||
- **punishment**: Уведомление о блокировке за превышение лимита устройств
|
||
- **enabled**: Уведомление о снятии блокировки
|
||
- **warning**: Предупреждение пользователю
|
||
|
||
Требует API ключ в заголовке X-API-Key или Authorization: Bearer <token>
|
||
"""
|
||
logger.info(
|
||
f"Получен запрос на отправку уведомления типа '{request.notification_type}' "
|
||
f"для пользователя {request.username} ({request.user_identifier}), "
|
||
f"node_name={request.node_name!r}"
|
||
)
|
||
|
||
try:
|
||
if request.notification_type == "punishment":
|
||
if request.ip_count is None or request.limit is None or request.ban_minutes is None:
|
||
raise HTTPException(
|
||
status_code=status.HTTP_400_BAD_REQUEST,
|
||
detail="Для типа 'punishment' требуются поля: ip_count, limit, ban_minutes"
|
||
)
|
||
|
||
success, message, telegram_id = await ban_notification_service.send_punishment_notification(
|
||
db=db,
|
||
user_identifier=request.user_identifier,
|
||
username=request.username,
|
||
ip_count=request.ip_count,
|
||
limit=request.limit,
|
||
ban_minutes=request.ban_minutes,
|
||
node_name=request.node_name,
|
||
)
|
||
|
||
elif request.notification_type == "enabled":
|
||
success, message, telegram_id = await ban_notification_service.send_enabled_notification(
|
||
db=db,
|
||
user_identifier=request.user_identifier,
|
||
username=request.username,
|
||
)
|
||
|
||
elif request.notification_type == "warning":
|
||
if not request.warning_message:
|
||
raise HTTPException(
|
||
status_code=status.HTTP_400_BAD_REQUEST,
|
||
detail="Для типа 'warning' требуется поле: warning_message"
|
||
)
|
||
|
||
success, message, telegram_id = await ban_notification_service.send_warning_notification(
|
||
db=db,
|
||
user_identifier=request.user_identifier,
|
||
username=request.username,
|
||
warning_message=request.warning_message,
|
||
)
|
||
|
||
elif request.notification_type == "network_wifi":
|
||
if request.ban_minutes is None:
|
||
raise HTTPException(
|
||
status_code=status.HTTP_400_BAD_REQUEST,
|
||
detail="Для типа 'network_wifi' требуется поле: ban_minutes"
|
||
)
|
||
|
||
success, message, telegram_id = await ban_notification_service.send_network_wifi_notification(
|
||
db=db,
|
||
user_identifier=request.user_identifier,
|
||
username=request.username,
|
||
ban_minutes=request.ban_minutes,
|
||
network_type=request.network_type,
|
||
node_name=request.node_name,
|
||
)
|
||
|
||
elif request.notification_type == "network_mobile":
|
||
if request.ban_minutes is None:
|
||
raise HTTPException(
|
||
status_code=status.HTTP_400_BAD_REQUEST,
|
||
detail="Для типа 'network_mobile' требуется поле: ban_minutes"
|
||
)
|
||
|
||
success, message, telegram_id = await ban_notification_service.send_network_mobile_notification(
|
||
db=db,
|
||
user_identifier=request.user_identifier,
|
||
username=request.username,
|
||
ban_minutes=request.ban_minutes,
|
||
network_type=request.network_type,
|
||
node_name=request.node_name,
|
||
)
|
||
|
||
else:
|
||
raise HTTPException(
|
||
status_code=status.HTTP_400_BAD_REQUEST,
|
||
detail=f"Неизвестный тип уведомления: {request.notification_type}"
|
||
)
|
||
|
||
return BanNotificationResponse(
|
||
success=success,
|
||
message=message,
|
||
telegram_id=telegram_id,
|
||
sent=success
|
||
)
|
||
|
||
except HTTPException:
|
||
raise
|
||
except Exception as e:
|
||
logger.exception(f"Ошибка при отправке уведомления: {e}")
|
||
raise HTTPException(
|
||
status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
|
||
detail=f"Внутренняя ошибка сервера: {str(e)}"
|
||
)
|