Files
remnawave-bedolaga-telegram…/app/services/remnawave_webhook_service.py
T
Fringg 977950b97f fix: address review issues in PR #2829 webhook intentional deletion guard
5 issues found by review agents and fixed:

1. Intentional deletion guard was at top of process_event, skipping ALL
   cleanup (subscription URLs, server counts, expired marking). Moved
   check into _handle_user_deleted so cleanup still runs but re-creation
   is suppressed via subscription_still_valid=False.

2. Variable shadowing: telegram_id loop var in any() generator shadowed
   the outer telegram_id local. Renamed to tid/uid.

3. No hard cap on in-memory dicts: added _MAX_INTENTIONAL_ENTRIES=10000
   with early return in mark_intentional_panel_deletion.

4. mark_intentional_panel_deletion called inside for-loop with single
   UUID — race window if webhook from first delete arrives before
   second UUID is marked. Moved to before the loop with all UUIDs.

5. Tests: @pytest.mark.anyio → @pytest.mark.anyio('asyncio') for
   consistency. Replaced process_event(db=None) test with direct
   mark+detect unit tests + hard cap test.
2026-04-02 06:34:35 +03:00

1412 lines
62 KiB
Python

"""
Service for processing incoming RemnaWave backend webhooks.
Handles all webhook scopes: user, user_hwid_devices, node, service, crm.
User events update subscription state and notify the user.
Admin events (node, service, crm) send alerts to the admin notification chat.
"""
from __future__ import annotations
import html
import re
from datetime import UTC, datetime
from typing import Any
import structlog
from aiogram import Bot
from aiogram.types import InlineKeyboardButton, InlineKeyboardMarkup
from sqlalchemy import delete, inspect as sa_inspect
from sqlalchemy.exc import PendingRollbackError
from sqlalchemy.ext.asyncio import AsyncSession
from sqlalchemy.orm.exc import StaleDataError
from app.config import settings
from app.database.crud.subscription import (
deactivate_subscription,
decrement_subscription_server_counts,
expire_subscription,
get_subscription_by_user_id,
is_recently_updated_by_webhook,
reactivate_subscription,
update_subscription_usage,
)
from app.database.crud.user import get_user_by_id, get_user_by_remnawave_uuid, get_user_by_telegram_id
from app.database.models import Subscription, SubscriptionServer, SubscriptionStatus, User
from app.localization.texts import get_texts
from app.services.admin_notification_service import AdminNotificationService
from app.services.notification_delivery_service import NotificationType, notification_delivery_service
from app.utils.miniapp_buttons import build_miniapp_or_callback_button
logger = structlog.get_logger(__name__)
# Mapping from locale text_key to NotificationType for unified delivery
_TEXT_KEY_TO_NOTIFICATION_TYPE: dict[str, NotificationType] = {
'WEBHOOK_SUB_EXPIRED': NotificationType.WEBHOOK_SUB_EXPIRED,
'WEBHOOK_SUB_DISABLED': NotificationType.WEBHOOK_SUB_DISABLED,
'WEBHOOK_SUB_ENABLED': NotificationType.WEBHOOK_SUB_ENABLED,
'WEBHOOK_SUB_LIMITED': NotificationType.WEBHOOK_SUB_LIMITED,
'WEBHOOK_SUB_TRAFFIC_RESET': NotificationType.WEBHOOK_SUB_TRAFFIC_RESET,
'WEBHOOK_SUB_DELETED': NotificationType.WEBHOOK_SUB_DELETED,
'WEBHOOK_SUB_REVOKED': NotificationType.WEBHOOK_SUB_REVOKED,
'WEBHOOK_SUB_EXPIRES_72H': NotificationType.WEBHOOK_SUB_EXPIRING,
'WEBHOOK_SUB_EXPIRES_48H': NotificationType.WEBHOOK_SUB_EXPIRING,
'WEBHOOK_SUB_EXPIRES_24H': NotificationType.WEBHOOK_SUB_EXPIRING,
'WEBHOOK_SUB_EXPIRED_24H_AGO': NotificationType.WEBHOOK_SUB_EXPIRED,
'WEBHOOK_SUB_FIRST_CONNECTED': NotificationType.WEBHOOK_SUB_FIRST_CONNECTED,
'WEBHOOK_SUB_BANDWIDTH_THRESHOLD': NotificationType.WEBHOOK_SUB_BANDWIDTH_THRESHOLD,
'WEBHOOK_USER_NOT_CONNECTED': NotificationType.WEBHOOK_USER_NOT_CONNECTED,
'WEBHOOK_DEVICE_ADDED': NotificationType.WEBHOOK_DEVICE_ADDED,
'WEBHOOK_DEVICE_DELETED': NotificationType.WEBHOOK_DEVICE_DELETED,
}
# Mapping from locale text_key to the Settings toggle that controls it
_TEXT_KEY_TO_SETTING: dict[str, str] = {
'WEBHOOK_SUB_EXPIRED': 'WEBHOOK_NOTIFY_SUB_EXPIRED',
'WEBHOOK_SUB_DISABLED': 'WEBHOOK_NOTIFY_SUB_STATUS',
'WEBHOOK_SUB_ENABLED': 'WEBHOOK_NOTIFY_SUB_STATUS',
'WEBHOOK_SUB_LIMITED': 'WEBHOOK_NOTIFY_SUB_LIMITED',
'WEBHOOK_SUB_TRAFFIC_RESET': 'WEBHOOK_NOTIFY_TRAFFIC_RESET',
'WEBHOOK_SUB_DELETED': 'WEBHOOK_NOTIFY_SUB_DELETED',
'WEBHOOK_SUB_REVOKED': 'WEBHOOK_NOTIFY_SUB_REVOKED',
'WEBHOOK_SUB_EXPIRES_72H': 'WEBHOOK_NOTIFY_SUB_EXPIRING',
'WEBHOOK_SUB_EXPIRES_48H': 'WEBHOOK_NOTIFY_SUB_EXPIRING',
'WEBHOOK_SUB_EXPIRES_24H': 'WEBHOOK_NOTIFY_SUB_EXPIRING',
'WEBHOOK_SUB_EXPIRED_24H_AGO': 'WEBHOOK_NOTIFY_SUB_EXPIRED',
'WEBHOOK_SUB_FIRST_CONNECTED': 'WEBHOOK_NOTIFY_FIRST_CONNECTED',
'WEBHOOK_SUB_BANDWIDTH_THRESHOLD': 'WEBHOOK_NOTIFY_BANDWIDTH_THRESHOLD',
'WEBHOOK_USER_NOT_CONNECTED': 'WEBHOOK_NOTIFY_NOT_CONNECTED',
'WEBHOOK_DEVICE_ADDED': 'WEBHOOK_NOTIFY_DEVICES',
'WEBHOOK_DEVICE_DELETED': 'WEBHOOK_NOTIFY_DEVICES',
}
# Admin event display names for notification messages
_ADMIN_NODE_EVENTS: dict[str, str] = {
'node.created': '🟢 Нода создана',
'node.modified': '🔧 Нода изменена',
'node.disabled': '🔴 Нода отключена',
'node.enabled': '🟢 Нода включена',
'node.deleted': '🗑️ Нода удалена',
'node.connection_lost': '🚨 Потеряно соединение с нодой',
'node.connection_restored': '✅ Соединение с нодой восстановлено',
'node.traffic_notify': '📊 Уведомление о трафике ноды',
}
_ADMIN_SERVICE_EVENTS: dict[str, str] = {
'service.panel_started': '🚀 Панель RemnaWave запущена',
'service.login_attempt_failed': '🔐 Неудачная попытка входа в панель',
'service.login_attempt_success': '🔓 Успешный вход в панель',
'service.subpage_config_changed': '📄 Конфиг страницы подписки изменён',
}
_ADMIN_CRM_EVENTS: dict[str, str] = {
'crm.infra_billing_node_payment_in_7_days': '💳 Оплата ноды через 7 дней',
'crm.infra_billing_node_payment_in_48hrs': '💳 Оплата ноды через 48 часов',
'crm.infra_billing_node_payment_in_24hrs': '⚠️ Оплата ноды через 24 часа',
'crm.infra_billing_node_payment_due_today': '🔴 Оплата ноды сегодня',
'crm.infra_billing_node_payment_overdue_24hrs': '❗ Просрочка оплаты ноды: 24 часа',
'crm.infra_billing_node_payment_overdue_48hrs': '❗ Просрочка оплаты ноды: 48 часов',
'crm.infra_billing_node_payment_overdue_7_days': '🚨 Просрочка оплаты ноды: 7 дней',
}
_ADMIN_ERROR_EVENTS: dict[str, str] = {
'errors.bandwidth_usage_threshold_reached_max_notifications': '⚠️ Достигнут лимит уведомлений о трафике',
}
_ADMIN_TORRENT_BLOCKER_EVENTS: dict[str, str] = {
'torrent_blocker.report': '🚫 Торрент-блокировщик: обнаружен торрент',
}
_ADMIN_NODE_CONNECTION_EVENTS = frozenset({'node.connection_lost', 'node.connection_restored'})
class RemnaWaveWebhookService:
"""Processes incoming webhooks from RemnaWave backend."""
# NOTE: In-memory guards. Only correct with a single-worker deployment.
# For multi-worker setups, move to Redis or another shared store.
_recent_recreations: dict[int, datetime] = {}
_RECREATION_GUARD_SECONDS: int = 120 # 2-minute cooldown
_intentional_panel_deletions_by_uuid: dict[str, datetime] = {}
_intentional_panel_deletions_by_telegram_id: dict[int, datetime] = {}
_INTENTIONAL_PANEL_DELETION_GUARD_SECONDS: int = 300
_MAX_INTENTIONAL_ENTRIES: int = 10_000
def __init__(self, bot: Bot) -> None:
self.bot = bot
self._admin_service = AdminNotificationService(bot)
# User-scoped handlers: require user resolution
self._user_handlers: dict[str, Any] = {
'user.expired': self._handle_user_expired,
'user.disabled': self._handle_user_disabled,
'user.enabled': self._handle_user_enabled,
'user.limited': self._handle_user_limited,
'user.traffic_reset': self._handle_user_traffic_reset,
'user.modified': self._handle_user_modified,
'user.deleted': self._handle_user_deleted,
'user.revoked': self._handle_user_revoked,
'user.created': self._handle_user_created,
'user.expires_in_72_hours': self._handle_expires_in_72h,
'user.expires_in_48_hours': self._handle_expires_in_48h,
'user.expires_in_24_hours': self._handle_expires_in_24h,
'user.expired_24_hours_ago': self._handle_expired_24h_ago,
'user.first_connected': self._handle_first_connected,
'user.bandwidth_usage_threshold_reached': self._handle_bandwidth_threshold,
'user.not_connected': self._handle_user_not_connected,
'user_hwid_devices.added': self._handle_device_added,
'user_hwid_devices.deleted': self._handle_device_deleted,
}
# Admin-scoped handlers: no user resolution, notify admin chat
self._admin_handlers: dict[str, str] = {
**_ADMIN_NODE_EVENTS,
**_ADMIN_SERVICE_EVENTS,
**_ADMIN_CRM_EVENTS,
**_ADMIN_ERROR_EVENTS,
**_ADMIN_TORRENT_BLOCKER_EVENTS,
}
def is_admin_event(self, event_name: str) -> bool:
"""Check if the event is admin-scoped (no DB session needed)."""
return event_name in self._admin_handlers
@classmethod
def _prune_intentional_panel_deletions(cls) -> None:
if not cls._intentional_panel_deletions_by_uuid and not cls._intentional_panel_deletions_by_telegram_id:
return
now = datetime.now(UTC)
uuid_keys = [
key
for key, created_at in cls._intentional_panel_deletions_by_uuid.items()
if (now - created_at).total_seconds() >= cls._INTENTIONAL_PANEL_DELETION_GUARD_SECONDS
]
for key in uuid_keys:
del cls._intentional_panel_deletions_by_uuid[key]
telegram_keys = [
key
for key, created_at in cls._intentional_panel_deletions_by_telegram_id.items()
if (now - created_at).total_seconds() >= cls._INTENTIONAL_PANEL_DELETION_GUARD_SECONDS
]
for key in telegram_keys:
del cls._intentional_panel_deletions_by_telegram_id[key]
@classmethod
def mark_intentional_panel_deletion(
cls,
*,
panel_uuids: list[str] | None = None,
telegram_id: int | None = None,
) -> None:
cls._prune_intentional_panel_deletions()
total = len(cls._intentional_panel_deletions_by_uuid) + len(cls._intentional_panel_deletions_by_telegram_id)
if total >= cls._MAX_INTENTIONAL_ENTRIES:
logger.warning('Intentional deletion guard at capacity, skipping', total=total)
return
now = datetime.now(UTC)
for panel_uuid in panel_uuids or []:
normalized = (panel_uuid or '').strip()
if normalized:
cls._intentional_panel_deletions_by_uuid[normalized] = now
if telegram_id is not None:
cls._intentional_panel_deletions_by_telegram_id[int(telegram_id)] = now
@classmethod
def _is_intentional_panel_deletion_event(cls, data: dict[str, Any]) -> bool:
cls._prune_intentional_panel_deletions()
candidate_uuids: list[str] = []
candidate_telegram_ids: list[int] = []
for value in (data.get('uuid'), data.get('userUuid')):
if value:
candidate_uuids.append(str(value).strip())
telegram_id = data.get('telegramId')
if telegram_id:
try:
candidate_telegram_ids.append(int(telegram_id))
except (TypeError, ValueError):
pass
nested_user = data.get('user')
if isinstance(nested_user, dict):
nested_uuid = nested_user.get('uuid')
if nested_uuid:
candidate_uuids.append(str(nested_uuid).strip())
nested_tid = nested_user.get('telegramId')
if nested_tid:
try:
candidate_telegram_ids.append(int(nested_tid))
except (TypeError, ValueError):
pass
return any(uid in cls._intentional_panel_deletions_by_uuid for uid in candidate_uuids) or any(
tid in cls._intentional_panel_deletions_by_telegram_id for tid in candidate_telegram_ids
)
async def process_event(self, db: AsyncSession | None, event_name: str, data: dict) -> bool:
"""Route event to the appropriate handler.
Returns True if the event was processed, False if skipped/unknown.
db may be None for admin events that don't require database access.
"""
# Check admin-scoped handlers (no DB needed)
if event_name in self._admin_handlers:
return await self._process_admin_event(event_name, data)
# Check user-scoped handlers (require DB session)
user_handler = self._user_handlers.get(event_name)
if user_handler:
if db is None:
logger.error('RemnaWave webhook: DB session required for user event', event_name=event_name)
return False
return await self._process_user_event(db, event_name, data, user_handler)
logger.debug('Unhandled RemnaWave webhook event', event_name=event_name)
return False
async def _process_user_event(self, db: AsyncSession, event_name: str, data: dict, handler: Any) -> bool:
"""Resolve user and execute user-scoped handler."""
user, subscription = await self._resolve_user_and_subscription(db, data)
if not user:
logger.warning(
'RemnaWave webhook: user not found for event , data telegramId= uuid',
event_name=event_name,
data=data.get('telegramId'),
data_2=data.get('uuid'),
)
return False
user_id = user.id
try:
await handler(db, user, subscription, data)
return True
except (StaleDataError, PendingRollbackError):
logger.warning(
'RemnaWave webhook : entity already deleted for user (concurrent deletion)',
event_name=event_name,
user_id=user_id,
)
try:
await db.rollback()
except Exception:
pass
return True
except Exception:
logger.exception(
'Error processing RemnaWave webhook event for user', event_name=event_name, user_id=user_id
)
try:
await db.rollback()
except Exception:
logger.debug('Rollback after webhook handler error also failed')
return False
async def _process_admin_event(self, event_name: str, data: dict) -> bool:
"""Format and send admin notification for infrastructure events."""
# Invalidate subscription page config cache on subpage config changes
if event_name == 'service.subpage_config_changed':
try:
from app.handlers.subscription.common import invalidate_app_config_cache
invalidate_app_config_cache()
logger.info(
'Webhook: subpage config changed — app config cache invalidated',
action=data.get('subpageConfig', {}).get('action')
if isinstance(data.get('subpageConfig'), dict)
else None,
)
except Exception:
logger.warning('Failed to invalidate app config cache on subpage_config_changed')
if event_name in _ADMIN_NODE_CONNECTION_EVENTS and not settings.REMNAWAVE_WEBHOOK_NOTIFY_NODE_CONNECTION_STATUS:
logger.debug('RemnaWave node connection notifications disabled, skipping event', event_name=event_name)
return True
if not self._admin_service.is_enabled:
logger.debug('Admin notifications disabled, skipping event', event_name=event_name)
return True
title = self._admin_handlers.get(event_name, event_name)
# Build message from event data (escape all untrusted values to prevent HTML injection)
lines = [f'<b>{title}</b>']
# Extract common fields
name = html.escape(data.get('name') or data.get('nodeName') or data.get('username') or '')
if name:
lines.append(f'Имя: <code>{name}</code>')
address = html.escape(data.get('address') or data.get('ip') or '')
if address:
lines.append(f'Адрес: <code>{address}</code>')
port = data.get('port')
if port:
lines.append(f'Порт: <code>{html.escape(str(port))}</code>')
version = html.escape(data.get('version') or data.get('panelVersion') or '')
if version:
lines.append(f'Версия: <code>{version}</code>')
# CRM billing fields
provider_name = html.escape(data.get('providerName') or '')
if provider_name:
lines.append(f'Провайдер: <code>{provider_name}</code>')
amount = html.escape(str(data.get('amount') or data.get('price') or ''))
if amount:
lines.append(f'Сумма: <code>{amount}</code>')
due_date = html.escape(data.get('dueDate') or data.get('paymentDate') or data.get('nextBillingAt') or '')
if due_date:
lines.append(f'Дата: <code>{due_date}</code>')
login_url = data.get('loginUrl') or ''
if login_url and self._is_valid_url(login_url):
lines.append(f'Панель: {html.escape(login_url)}')
# Login attempt fields
login_data = data.get('loginAttempt')
if isinstance(login_data, dict):
login_user = html.escape(login_data.get('username') or '')
if login_user:
lines.append(f'Пользователь: <code>{login_user}</code>')
login_ip = html.escape(login_data.get('ip') or '')
if login_ip:
lines.append(f'IP: <code>{login_ip}</code>')
login_ua = html.escape(login_data.get('userAgent') or '')
if login_ua:
lines.append(f'User-Agent: <code>{login_ua[:100]}</code>')
login_desc = html.escape(login_data.get('description') or '')
if login_desc:
lines.append(f'Описание: {login_desc}')
else:
ip_addr = html.escape(data.get('ipAddress') or data.get('ip') or '')
if ip_addr and not address:
lines.append(f'IP: <code>{ip_addr}</code>')
message = html.escape(data.get('message') or data.get('description') or '')
if message:
lines.append(f'Сообщение: {message}')
# Subpage config fields
subpage = data.get('subpageConfig')
if isinstance(subpage, dict):
action = subpage.get('action', '')
action_labels = {'CREATED': 'Создан', 'UPDATED': 'Обновлён', 'DELETED': 'Удалён'}
lines.append(f'Действие: {action_labels.get(action, html.escape(str(action)))}')
sub_uuid = subpage.get('uuid', '')
if sub_uuid:
lines.append(f'UUID: <code>{html.escape(str(sub_uuid))}</code>')
# Torrent blocker fields
if event_name == 'torrent_blocker.report':
node_data = data.get('node')
user_data = data.get('user')
if isinstance(node_data, dict):
node_name = html.escape(node_data.get('name') or '')
if node_name:
lines.append(f'Нода: <code>{node_name}</code>')
node_addr = html.escape(node_data.get('address') or '')
if node_addr:
lines.append(f'Адрес: <code>{node_addr}</code>')
if isinstance(user_data, dict):
username = html.escape(user_data.get('username') or '')
if username:
lines.append(f'Пользователь: <code>{username}</code>')
user_status = html.escape(user_data.get('status') or '')
if user_status:
lines.append(f'Статус: <code>{user_status}</code>')
try:
await self._admin_service.send_webhook_notification('\n'.join(lines))
return True
except Exception:
logger.exception('Failed to send admin notification for event', event_name=event_name)
return False
# ------------------------------------------------------------------
# User resolution
# ------------------------------------------------------------------
async def _resolve_user_and_subscription(
self, db: AsyncSession, data: dict
) -> tuple[User | None, Subscription | None]:
"""Find bot user by telegramId or uuid from webhook payload.
Handles both user-scope events (top-level telegramId/uuid) and
device-scope events (userUuid, or nested user.telegramId/user.uuid).
In multi-tariff mode, resolves subscription by remnawave_uuid from payload
(each subscription has its own Remnawave user).
"""
user: User | None = None
remnawave_uuid: str | None = None
# Extract Remnawave UUID from payload (used for subscription lookup in multi-tariff)
remnawave_uuid = data.get('uuid') or data.get('userUuid')
if not remnawave_uuid:
nested_user = data.get('user')
if isinstance(nested_user, dict):
remnawave_uuid = nested_user.get('uuid')
# Try top-level telegramId first
telegram_id = data.get('telegramId')
if telegram_id:
try:
user = await get_user_by_telegram_id(db, int(telegram_id))
except (ValueError, TypeError):
pass
# Try top-level uuid
if not user and remnawave_uuid:
user = await get_user_by_remnawave_uuid(db, remnawave_uuid)
# Try nested user object (e.g. user_hwid_devices events)
if not user:
nested_user = data.get('user')
if isinstance(nested_user, dict):
nested_tid = nested_user.get('telegramId')
if nested_tid:
try:
user = await get_user_by_telegram_id(db, int(nested_tid))
except (ValueError, TypeError):
pass
if not user:
nested_uuid = nested_user.get('uuid')
if nested_uuid:
user = await get_user_by_remnawave_uuid(db, nested_uuid)
# Multi-tariff: try finding user through subscription's remnawave_uuid
if not user and remnawave_uuid and settings.is_multi_tariff_enabled():
from sqlalchemy import select as sa_select
from sqlalchemy.orm import selectinload as sa_selectinload
sub_result = await db.execute(
sa_select(Subscription)
.options(
sa_selectinload(Subscription.user)
.selectinload(User.subscriptions)
.selectinload(Subscription.tariff),
sa_selectinload(Subscription.tariff),
)
.where(Subscription.remnawave_uuid == remnawave_uuid)
.limit(1)
)
found_sub = sub_result.scalar_one_or_none()
if found_sub and found_sub.user:
return found_sub.user, found_sub
if not user:
return None, None
# In multi-tariff mode, find subscription by remnawave_uuid (per-subscription)
if settings.is_multi_tariff_enabled() and remnawave_uuid:
from sqlalchemy import select
from sqlalchemy.orm import selectinload
result = await db.execute(
select(Subscription)
.options(selectinload(Subscription.tariff))
.where(
Subscription.remnawave_uuid == remnawave_uuid,
Subscription.user_id == user.id,
)
)
subscription = result.scalar_one_or_none()
if subscription:
return user, subscription
# Fallback 1: search ALL user's subscriptions by remnawave_uuid
# (covers recently merged accounts where user_id might differ)
logger.warning(
'Webhook: подписка не найдена по remnawave_uuid + user_id, '
'fallback на поиск по remnawave_uuid среди всех подписок пользователя',
remnawave_uuid=remnawave_uuid,
user_id=user.id,
)
fallback1_result = await db.execute(
select(Subscription)
.options(selectinload(Subscription.tariff))
.where(Subscription.remnawave_uuid == remnawave_uuid)
.limit(1)
)
fallback1_sub = fallback1_result.scalar_one_or_none()
if fallback1_sub:
if fallback1_sub.user_id == user.id:
return user, fallback1_sub
# Subscription belongs to a different user (transferred or merged)
logger.warning(
'Webhook: подписка найдена по remnawave_uuid, '
'но принадлежит другому пользователю — игнорируем (IDOR prevention)',
remnawave_uuid=remnawave_uuid,
webhook_user_id=user.id,
subscription_user_id=fallback1_sub.user_id,
subscription_id=fallback1_sub.id,
)
# Do NOT return cross-user subscription — would mutate another user's data
return user, None
# Fallback 2: all lookups exhausted
logger.warning(
'Webhook: подписка не найдена ни по одному методу поиска, возвращаем (user, None)',
remnawave_uuid=remnawave_uuid,
user_id=user.id,
)
if settings.is_multi_tariff_enabled():
# In multi-tariff mode, don't fall back to arbitrary subscription
return user, None
subscription = await get_subscription_by_user_id(db, user.id)
return user, subscription
# ------------------------------------------------------------------
# Notification helpers
# ------------------------------------------------------------------
@staticmethod
def _is_valid_url(value: str) -> bool:
"""Basic URL validation to prevent stored XSS via crafted URLs."""
if not value or len(value) > 2048:
return False
return bool(re.match(r'^https?://', value))
@staticmethod
def _is_valid_link(value: str) -> bool:
"""Validate URL or deep link (happ://, vless://, ss://, etc.)."""
if not value or len(value) > 4096:
return False
return bool(re.match(r'^[a-zA-Z][a-zA-Z0-9+\-.]*://', value))
def _get_renew_keyboard(self, user: User, subscription_id: int | None = None) -> InlineKeyboardMarkup:
texts = get_texts(user.language)
button_text = texts.get('WEBHOOK_RENEW_BUTTON', 'Renew subscription')
extend_callback = (
f'se:{subscription_id}' if settings.is_multi_tariff_enabled() and subscription_id else 'subscription_extend'
)
return InlineKeyboardMarkup(
inline_keyboard=[
[build_miniapp_or_callback_button(text=button_text, callback_data=extend_callback)],
]
)
def _get_subscription_keyboard(self, user: User) -> InlineKeyboardMarkup:
texts = get_texts(user.language)
button_text = texts.get('MY_SUBSCRIPTION_BUTTON', 'My subscription')
return InlineKeyboardMarkup(
inline_keyboard=[
[build_miniapp_or_callback_button(text=button_text, callback_data='menu_subscription')],
]
)
def _get_connect_keyboard(self, user: User) -> InlineKeyboardMarkup:
texts = get_texts(user.language)
button_text = texts.get('CONNECT_BUTTON', 'Connect')
return InlineKeyboardMarkup(
inline_keyboard=[
[build_miniapp_or_callback_button(text=button_text, callback_data='subscription_connect')],
]
)
def _get_traffic_keyboard(self, user: User) -> InlineKeyboardMarkup:
texts = get_texts(user.language)
buy_text = texts.get('BUY_TRAFFIC_BUTTON', 'Buy traffic')
sub_text = texts.get('MY_SUBSCRIPTION_BUTTON', 'My subscription')
return InlineKeyboardMarkup(
inline_keyboard=[
[build_miniapp_or_callback_button(text=buy_text, callback_data='buy_traffic')],
[build_miniapp_or_callback_button(text=sub_text, callback_data='menu_subscription')],
]
)
async def _notify_user(
self,
user: User,
text_key: str,
*,
reply_markup: InlineKeyboardMarkup | None = None,
format_kwargs: dict[str, Any] | None = None,
subscription: Subscription | None = None,
) -> None:
"""Send a notification to user via appropriate channel.
Telegram users receive a bot message; email-only users receive
an email and/or WebSocket notification through the unified
notification delivery service.
Respects WEBHOOK_NOTIFY_USER_ENABLED master toggle and
per-event toggles from Settings.
"""
if not settings.WEBHOOK_NOTIFY_USER_ENABLED:
logger.debug('Webhook user notifications disabled globally, skipping', text_key=text_key)
return
setting_key = _TEXT_KEY_TO_SETTING.get(text_key)
if setting_key and not getattr(settings, setting_key, True):
logger.debug('Webhook notification disabled via', text_key=text_key, setting_key=setting_key)
return
texts = get_texts(user.language)
message = texts.get(text_key)
if not message:
logger.warning('Missing locale key for language', text_key=text_key, language=user.language)
return
# Inject tariff_label for multi-tariff subscription identification
if format_kwargs is None:
format_kwargs = {}
if 'tariff_label' not in format_kwargs:
tariff_label = ''
if settings.is_multi_tariff_enabled() and subscription:
# Access tariff only if already eagerly loaded to avoid
# MissingGreenlet from lazy loading in async context
loaded_tariff = sa_inspect(subscription).dict.get('tariff')
if loaded_tariff is not None:
tariff_label = f' «{loaded_tariff.name}»'
format_kwargs['tariff_label'] = tariff_label
if format_kwargs:
try:
message = message.format(**format_kwargs)
except (KeyError, IndexError):
logger.warning('Failed to format message with kwargs', text_key=text_key, format_kwargs=format_kwargs)
return
# Append "Close" button to every webhook notification keyboard
close_text = texts.get('WEBHOOK_CLOSE_BUTTON', '✖️ Закрыть')
close_row = [InlineKeyboardButton(text=close_text, callback_data='webhook:close')]
if reply_markup:
reply_markup = InlineKeyboardMarkup(
inline_keyboard=[*reply_markup.inline_keyboard, close_row],
)
else:
reply_markup = InlineKeyboardMarkup(inline_keyboard=[close_row])
notification_type = _TEXT_KEY_TO_NOTIFICATION_TYPE.get(text_key)
if not notification_type:
logger.warning('No NotificationType mapping for text_key', text_key=text_key)
return
context = {'text_key': text_key, **(format_kwargs or {})}
try:
await notification_delivery_service.send_notification(
user=user,
notification_type=notification_type,
context=context,
bot=self.bot,
telegram_message=message,
telegram_markup=reply_markup,
)
except Exception:
logger.exception('Notification delivery failed for user , text_key', user_id=user.id, text_key=text_key)
# ------------------------------------------------------------------
# Webhook timestamp helper
# ------------------------------------------------------------------
@staticmethod
def _stamp_webhook_update(subscription: Subscription) -> None:
"""Mark subscription as recently updated by webhook to prevent sync overwrite."""
subscription.last_webhook_update_at = datetime.now(UTC)
# ------------------------------------------------------------------
# User event handlers
# ------------------------------------------------------------------
async def _handle_user_expired(
self, db: AsyncSession, user: User, subscription: Subscription | None, data: dict
) -> None:
if not subscription:
# Подписка уже удалена из БД — фантомный хук от панели, игнорируем
logger.info('Webhook user.expired: подписка не найдена в БД (уже удалена), пропуск', user_id=user.id)
return
# Суточные подписки управляются DailySubscriptionService.
# Remnawave может прислать user.expired если sync не дошёл (старый end_date),
# но локально подписка ещё жива — не экспайрим её.
tariff = sa_inspect(subscription).dict.get('tariff')
is_active_daily = (
tariff is not None
and getattr(tariff, 'is_daily', False)
and not getattr(subscription, 'is_daily_paused', False)
)
if is_active_daily:
logger.info(
'Webhook: пропуск expire для суточной подписки (управляет DailySubscriptionService)',
subscription_id=subscription.id,
user_id=user.id,
)
self._stamp_webhook_update(subscription)
await db.commit()
return
self._stamp_webhook_update(subscription)
if subscription.status != SubscriptionStatus.EXPIRED.value:
await expire_subscription(db, subscription)
logger.info('Webhook: subscription expired for user', subscription_id=subscription.id, user_id=user.id)
else:
await db.commit()
await self._notify_user(
user,
'WEBHOOK_SUB_EXPIRED',
reply_markup=self._get_renew_keyboard(user, subscription.id),
subscription=subscription,
)
async def _handle_user_disabled(
self, db: AsyncSession, user: User, subscription: Subscription | None, data: dict
) -> None:
if not subscription:
# Подписка уже удалена из БД — фантомный хук от панели, игнорируем
logger.info('Webhook user.disabled: подписка не найдена в БД (уже удалена), пропуск', user_id=user.id)
return
# Суточные подписки управляются DailySubscriptionService — не деактивируем
tariff = sa_inspect(subscription).dict.get('tariff')
is_active_daily = (
tariff is not None
and getattr(tariff, 'is_daily', False)
and not getattr(subscription, 'is_daily_paused', False)
)
if is_active_daily:
logger.info(
'Webhook: пропуск disabled для суточной подписки',
subscription_id=subscription.id,
user_id=user.id,
)
self._stamp_webhook_update(subscription)
await db.commit()
return
# Защита от echo-webhook: если подписка была недавно реактивирована
# (канал-реподписка ставит last_webhook_update_at), пропускаем
if subscription.status == SubscriptionStatus.ACTIVE.value and is_recently_updated_by_webhook(subscription):
logger.info(
'Webhook user.disabled: подписка недавно реактивирована, пропуск echo-webhook',
subscription_id=subscription.id,
user_id=user.id,
)
self._stamp_webhook_update(subscription)
await db.commit()
return
self._stamp_webhook_update(subscription)
if subscription.status != SubscriptionStatus.DISABLED.value:
await deactivate_subscription(db, subscription)
logger.info('Webhook: subscription disabled for user', subscription_id=subscription.id, user_id=user.id)
else:
await db.commit()
await self._notify_user(
user, 'WEBHOOK_SUB_DISABLED', reply_markup=self._get_subscription_keyboard(user), subscription=subscription
)
async def _handle_user_enabled(
self, db: AsyncSession, user: User, subscription: Subscription | None, data: dict
) -> None:
if not subscription:
logger.info('Webhook user.enabled: подписка не найдена в БД (уже удалена), пропуск', user_id=user.id)
return
self._stamp_webhook_update(subscription)
if subscription.status in (SubscriptionStatus.DISABLED.value, SubscriptionStatus.LIMITED.value):
await reactivate_subscription(db, subscription)
logger.info('Webhook: subscription re-enabled for user', subscription_id=subscription.id, user_id=user.id)
else:
await db.commit()
await self._notify_user(
user, 'WEBHOOK_SUB_ENABLED', reply_markup=self._get_connect_keyboard(user), subscription=subscription
)
async def _handle_user_limited(
self, db: AsyncSession, user: User, subscription: Subscription | None, data: dict
) -> None:
if not subscription:
logger.info('Webhook user.limited: подписка не найдена в БД (уже удалена), пропуск', user_id=user.id)
return
self._stamp_webhook_update(subscription)
if subscription.status in (SubscriptionStatus.ACTIVE.value, SubscriptionStatus.TRIAL.value):
subscription.status = SubscriptionStatus.LIMITED.value
subscription.updated_at = datetime.now(UTC)
await db.commit()
await db.refresh(subscription)
logger.info(
'Webhook: subscription limited (traffic) for user', subscription_id=subscription.id, user_id=user.id
)
else:
await db.commit()
await self._notify_user(
user, 'WEBHOOK_SUB_LIMITED', reply_markup=self._get_traffic_keyboard(user), subscription=subscription
)
async def _handle_user_traffic_reset(
self, db: AsyncSession, user: User, subscription: Subscription | None, data: dict
) -> None:
if not subscription:
logger.info('Webhook user.traffic_reset: подписка не найдена в БД (уже удалена), пропуск', user_id=user.id)
return
self._stamp_webhook_update(subscription)
await update_subscription_usage(db, subscription, 0.0)
# Re-enable if was disabled/limited due to traffic limit
if subscription.status in (SubscriptionStatus.DISABLED.value, SubscriptionStatus.LIMITED.value):
await reactivate_subscription(db, subscription)
logger.info('Webhook: traffic reset for subscription , user', subscription_id=subscription.id, user_id=user.id)
await self._notify_user(
user,
'WEBHOOK_SUB_TRAFFIC_RESET',
reply_markup=self._get_subscription_keyboard(user),
subscription=subscription,
)
async def _handle_user_modified(
self, db: AsyncSession, user: User, subscription: Subscription | None, data: dict
) -> None:
"""Sync subscription fields from webhook payload without notifying user."""
if not subscription:
return
changed = False
# Sync traffic limit
traffic_limit_bytes = data.get('trafficLimitBytes')
if traffic_limit_bytes is not None:
try:
new_limit_gb = int(traffic_limit_bytes) // (1024**3)
if subscription.traffic_limit_gb != new_limit_gb:
subscription.traffic_limit_gb = new_limit_gb
changed = True
except (ValueError, TypeError):
pass
# Sync used traffic
used_traffic_bytes = data.get('usedTrafficBytes')
if used_traffic_bytes is not None:
try:
new_used_gb = round(int(used_traffic_bytes) / (1024**3), 2)
subscription.traffic_used_gb = new_used_gb
changed = True
except (ValueError, TypeError):
pass
# Sync expire date — panel is the source of truth for user.modified events
expire_at = data.get('expireAt')
if expire_at:
try:
parsed_dt = datetime.fromisoformat(expire_at.replace('Z', '+00:00'))
new_end_date = parsed_dt.astimezone(UTC)
if subscription.end_date != new_end_date:
old_end_date = subscription.end_date
subscription.end_date = new_end_date
changed = True
if old_end_date and new_end_date < old_end_date:
logger.info(
'Webhook: end_date обновлена назад (панель авторитетна): → ',
subscription_id=subscription.id,
old_end_date=old_end_date,
new_end_date=new_end_date,
)
except (ValueError, TypeError):
pass
# Sync status from panel
panel_status = data.get('status')
if panel_status:
now = datetime.now(UTC)
end_date = subscription.end_date
if panel_status == 'ACTIVE' and end_date and end_date > now:
if subscription.status != SubscriptionStatus.ACTIVE.value:
subscription.status = SubscriptionStatus.ACTIVE.value
changed = True
logger.info(
'Webhook: subscription reactivated (→ active) for user',
subscription_id=subscription.id,
subscription_status=subscription.status,
user_id=user.id,
)
elif panel_status == 'DISABLED':
if subscription.status != SubscriptionStatus.DISABLED.value:
subscription.status = SubscriptionStatus.DISABLED.value
changed = True
# Sync subscription URL (validate to prevent stored XSS)
subscription_url = data.get('subscriptionUrl')
if (
subscription_url
and self._is_valid_url(subscription_url)
and subscription.subscription_url != subscription_url
):
subscription.subscription_url = subscription_url
changed = True
# Sync subscription crypto link (for HAPP_CRYPT4_LINK)
subscription_crypto_link = data.get('subscriptionCryptoLink') or (data.get('happ') or {}).get('cryptoLink', '')
if subscription_crypto_link and self._is_valid_link(subscription_crypto_link):
if subscription.subscription_crypto_link != subscription_crypto_link:
subscription.subscription_crypto_link = subscription_crypto_link
changed = True
elif subscription_url and subscription.subscription_crypto_link:
# URL обновился, а крипто-ссылка не пришла — сбрасываем старую
subscription.subscription_crypto_link = None
changed = True
# Always stamp to protect from sync overwrite, even if no fields changed
self._stamp_webhook_update(subscription)
if changed:
subscription.updated_at = datetime.now(UTC)
logger.info(
'Webhook: subscription modified (synced from panel) for user',
subscription_id=subscription.id,
user_id=user.id,
)
await db.commit()
async def _handle_user_deleted(
self, db: AsyncSession, user: User, subscription: Subscription | None, data: dict
) -> None:
user_id = user.id
sub_id = subscription.id if subscription else None
# Evict stale entries from the recreation loop guard to prevent unbounded growth
if self._recent_recreations:
now = datetime.now(UTC)
expired_keys = [
k
for k, v in self._recent_recreations.items()
if (now - v).total_seconds() >= self._RECREATION_GUARD_SECONDS
]
for k in expired_keys:
del self._recent_recreations[k]
# Guard against webhook loop: if we recently attempted panel recreation for this
# subscription (from a previous user.deleted), skip to prevent unbounded
# recreate→delete→recreate cycles. Uses an in-memory guard (not the generic
# last_webhook_update_at stamp which fires on ANY webhook event).
if sub_id and sub_id in self._recent_recreations:
elapsed = (datetime.now(UTC) - self._recent_recreations[sub_id]).total_seconds()
if elapsed < self._RECREATION_GUARD_SECONDS:
logger.warning(
'Webhook user.deleted: skipping — panel recreation was attempted recently (recreation loop guard)',
sub_id=sub_id,
user_id=user_id,
elapsed=round(elapsed, 1),
)
return
# Stamp immediately (before any await) so concurrent coroutines see the guard
if sub_id:
self._recent_recreations[sub_id] = datetime.now(UTC)
if subscription:
self._stamp_webhook_update(subscription)
# Decrement server counters BEFORE clearing connected_squads
await decrement_subscription_server_counts(db, subscription)
# Re-fetch after potential rollback inside decrement_subscription_server_counts
try:
await db.refresh(subscription)
except Exception:
# Subscription was cascade-deleted, re-fetch user and skip subscription updates
logger.warning(
'Webhook: subscription already deleted for user , skipping subscription cleanup',
sub_id=sub_id,
user_id=user_id,
)
subscription = None
try:
await db.rollback()
except Exception:
pass
try:
user = await get_user_by_id(db, user_id)
except Exception:
logger.error('Webhook: user not found after rollback', user_id=user_id)
return
if not user:
logger.error('Webhook: user not found after rollback', user_id=user_id)
return
# Intentional admin deletion: cleanup runs (fields cleared above), but skip re-creation
is_intentional = self._is_intentional_panel_deletion_event(data)
if is_intentional:
logger.info(
'Webhook user.deleted: intentional admin deletion, cleanup done, skipping re-creation',
sub_id=sub_id,
user_id=user_id,
)
# Check if subscription has a future end_date — likely a spurious user.deleted
# (e.g., RemnaWave sends user.deleted during panel resync when modifying another user)
subscription_still_valid = (
not is_intentional
and subscription is not None
and subscription.end_date is not None
and subscription.end_date > datetime.now(UTC)
)
if subscription:
if subscription_still_valid:
# Subscription is still valid — don't mark as expired.
# Clear only panel linkage fields (URLs, UUID) but keep status and squads
# so that re-creation can restore VPN access.
logger.warning(
'Webhook user.deleted: subscription has future end_date, '
'keeping active status and attempting panel re-creation',
sub_id=sub_id,
user_id=user_id,
end_date=subscription.end_date,
status=subscription.status,
)
subscription.subscription_url = None
subscription.subscription_crypto_link = None
subscription.remnawave_short_uuid = None
# Keep connected_squads — needed for panel re-creation
subscription.updated_at = datetime.now(UTC)
else:
# Subscription expired or has no end_date — safe to mark as expired
if subscription.status != SubscriptionStatus.EXPIRED.value:
subscription.status = SubscriptionStatus.EXPIRED.value
logger.info(
'Webhook: subscription marked expired (user deleted in panel) for user',
sub_id=sub_id,
user_id=user_id,
)
subscription.subscription_url = None
subscription.subscription_crypto_link = None
subscription.remnawave_short_uuid = None
subscription.connected_squads = []
subscription.updated_at = datetime.now(UTC)
# In multi-tariff mode clear per-subscription UUID here
if settings.is_multi_tariff_enabled():
subscription.remnawave_uuid = None
# Remove SubscriptionServer link rows (panel user no longer exists)
await db.execute(delete(SubscriptionServer).where(SubscriptionServer.subscription_id == sub_id))
# Clear remnawave linkage — only in single-tariff mode (multi-tariff uses per-subscription UUIDs)
if not settings.is_multi_tariff_enabled():
if user.remnawave_uuid:
user.remnawave_uuid = None
# In multi-tariff mode, subscription.remnawave_uuid was cleared above.
# If subscription was None (fallback path), extract panel UUID from data and
# clear it from the matching subscription manually.
elif subscription is None:
panel_uuid = data.get('uuid') or data.get('userUuid')
if panel_uuid:
await db.refresh(user, ['subscriptions'])
for sub in getattr(user, 'subscriptions', None) or []:
if getattr(sub, 'remnawave_uuid', None) == panel_uuid:
sub.remnawave_uuid = None
sub.remnawave_short_uuid = None
break
await db.commit()
if subscription_still_valid:
# Attempt to re-create user in panel to restore VPN access.
# If recreation fails, fall back to expiring the subscription
# so it doesn't stay in ACTIVE-but-no-panel limbo.
recreated = await self._attempt_panel_recreation(db, user, subscription)
if not recreated:
subscription.status = SubscriptionStatus.EXPIRED.value
subscription.connected_squads = []
subscription.updated_at = datetime.now(UTC)
await db.commit()
await self._notify_user(
user,
'WEBHOOK_SUB_DELETED',
reply_markup=self._get_renew_keyboard(
user, getattr(subscription, 'id', None) if subscription else None
),
subscription=subscription,
)
else:
await self._notify_user(
user,
'WEBHOOK_SUB_DELETED',
reply_markup=self._get_renew_keyboard(
user, getattr(subscription, 'id', None) if subscription else None
),
subscription=subscription,
)
async def _attempt_panel_recreation(self, db: AsyncSession, user: User, subscription: Subscription) -> bool:
"""Re-create user in RemnaWave panel after spurious user.deleted webhook.
Called when a user.deleted webhook arrives but the subscription still has a
future end_date, indicating the deletion was likely spurious (e.g., RemnaWave
resync when modifying another user). Attempts to restore VPN access by
creating/updating the user in the panel.
Returns True if recreation succeeded, False otherwise.
"""
# Update the recreation guard timestamp to the actual recreation start time
if subscription.id is not None:
self._recent_recreations[subscription.id] = datetime.now(UTC)
try:
from app.services.subscription_service import SubscriptionService
service = SubscriptionService()
if not service.is_configured:
logger.warning(
'RemnaWave not configured, cannot re-create panel user after user.deleted',
user_id=user.id,
)
return False
remnawave_user = await service.create_remnawave_user(db, subscription)
if remnawave_user:
logger.info(
'Webhook user.deleted: successfully re-created user in panel',
user_id=user.id,
subscription_id=subscription.id,
new_uuid=remnawave_user.uuid,
)
return True
logger.error(
'Webhook user.deleted: failed to re-create user in panel',
user_id=user.id,
subscription_id=subscription.id,
)
return False
except Exception as e:
logger.error(
'Webhook user.deleted: error re-creating user in panel',
user_id=user.id,
subscription_id=subscription.id,
error=e,
)
return False
async def _handle_user_revoked(
self, db: AsyncSession, user: User, subscription: Subscription | None, data: dict
) -> None:
if not subscription:
logger.info('Webhook user.revoked: подписка не найдена в БД, пропуск', user_id=user.id)
return
new_url = data.get('subscriptionUrl')
new_crypto_link = data.get('subscriptionCryptoLink') or (data.get('happ') or {}).get('cryptoLink', '')
changed = False
if new_url and self._is_valid_url(new_url) and subscription.subscription_url != new_url:
subscription.subscription_url = new_url
changed = True
if new_crypto_link and self._is_valid_link(new_crypto_link):
if subscription.subscription_crypto_link != new_crypto_link:
subscription.subscription_crypto_link = new_crypto_link
changed = True
elif new_url and subscription.subscription_crypto_link:
subscription.subscription_crypto_link = None
changed = True
# Always stamp to protect from sync overwrite
self._stamp_webhook_update(subscription)
if changed:
subscription.updated_at = datetime.now(UTC)
logger.info(
'Webhook: subscription credentials revoked/updated for user',
subscription_id=subscription.id,
user_id=user.id,
)
await db.commit()
await self._notify_user(
user, 'WEBHOOK_SUB_REVOKED', reply_markup=self._get_connect_keyboard(user), subscription=subscription
)
async def _handle_user_created(
self, db: AsyncSession, user: User, subscription: Subscription | None, data: dict
) -> None:
logger.info('Webhook: user created externally in panel (uuid=)', user_id=user.id, data=data.get('uuid'))
async def _handle_expires_in_72h(
self, db: AsyncSession, user: User, subscription: Subscription | None, data: dict
) -> None:
if not subscription:
logger.info('Webhook expires_72h: подписка не найдена в БД, пропуск', user_id=user.id)
return
await self._notify_user(
user,
'WEBHOOK_SUB_EXPIRES_72H',
reply_markup=self._get_renew_keyboard(user, subscription.id),
subscription=subscription,
)
async def _handle_expires_in_48h(
self, db: AsyncSession, user: User, subscription: Subscription | None, data: dict
) -> None:
if not subscription:
logger.info('Webhook expires_48h: подписка не найдена в БД, пропуск', user_id=user.id)
return
await self._notify_user(
user,
'WEBHOOK_SUB_EXPIRES_48H',
reply_markup=self._get_renew_keyboard(user, subscription.id),
subscription=subscription,
)
async def _handle_expires_in_24h(
self, db: AsyncSession, user: User, subscription: Subscription | None, data: dict
) -> None:
if not subscription:
logger.info('Webhook expires_24h: подписка не найдена в БД, пропуск', user_id=user.id)
return
await self._notify_user(
user,
'WEBHOOK_SUB_EXPIRES_24H',
reply_markup=self._get_renew_keyboard(user, subscription.id),
subscription=subscription,
)
async def _handle_expired_24h_ago(
self, db: AsyncSession, user: User, subscription: Subscription | None, data: dict
) -> None:
if not subscription:
logger.info('Webhook expired_24h_ago: подписка не найдена в БД, пропуск', user_id=user.id)
return
await self._notify_user(
user,
'WEBHOOK_SUB_EXPIRED_24H_AGO',
reply_markup=self._get_renew_keyboard(user, subscription.id),
subscription=subscription,
)
async def _handle_first_connected(
self, db: AsyncSession, user: User, subscription: Subscription | None, data: dict
) -> None:
logger.info('Webhook: user first VPN connection', user_id=user.id)
await self._notify_user(
user,
'WEBHOOK_SUB_FIRST_CONNECTED',
reply_markup=self._get_subscription_keyboard(user),
subscription=subscription,
)
async def _handle_bandwidth_threshold(
self, db: AsyncSession, user: User, subscription: Subscription | None, data: dict
) -> None:
# Extract threshold percentage from meta or data
percent = data.get('thresholdPercent') or data.get('threshold', '')
if not percent:
# Try to extract from meta
meta = data.get('meta', {})
if isinstance(meta, dict):
percent = meta.get('thresholdPercent', '80')
# Sanitize to numeric value only (prevent format string injection)
percent_str = re.sub(r'[^\d.]', '', str(percent)) or '80'
await self._notify_user(
user,
'WEBHOOK_SUB_BANDWIDTH_THRESHOLD',
reply_markup=self._get_traffic_keyboard(user),
format_kwargs={'percent': percent_str},
subscription=subscription,
)
async def _handle_user_not_connected(
self, db: AsyncSession, user: User, subscription: Subscription | None, data: dict
) -> None:
meta = data.get('_meta') or {}
hours = meta.get('notConnectedAfterHours')
logger.info(
'Webhook: user has not connected to VPN',
user_id=user.id,
not_connected_after_hours=hours,
)
format_kwargs: dict[str, Any] = {}
if hours is not None:
format_kwargs['hours'] = str(hours)
await self._notify_user(
user,
'WEBHOOK_USER_NOT_CONNECTED',
reply_markup=self._get_connect_keyboard(user),
format_kwargs=format_kwargs if format_kwargs else None,
subscription=subscription,
)
# ------------------------------------------------------------------
# Device event handlers (user_hwid_devices scope)
# ------------------------------------------------------------------
@staticmethod
def _extract_device_name(data: dict) -> str:
"""Extract device name from webhook payload.
RemnaWave sends device info in data['hwidUserDevice'] nested object.
Builds a composite name: "tag (platform)" or just "platform" or hwid short.
"""
device_obj = data.get('hwidUserDevice')
if not isinstance(device_obj, dict):
# Fallback: top-level fields
raw = data.get('deviceName') or data.get('tag') or data.get('hwid') or ''
return html.escape(str(raw)) if raw else ''
tag = (device_obj.get('tag') or device_obj.get('deviceName') or device_obj.get('name') or '').strip()
platform = (device_obj.get('platform') or '').strip()
hwid = (device_obj.get('hwid') or '').strip()
if tag and platform:
return html.escape(f'{tag} ({platform})')
if tag:
return html.escape(tag)
if platform and hwid:
# Show platform + short hwid suffix for identification
hwid_short = hwid[:8] if len(hwid) > 8 else hwid
return html.escape(f'{platform} ({hwid_short})')
if platform:
return html.escape(platform)
if hwid:
hwid_short = hwid[:12] if len(hwid) > 12 else hwid
return html.escape(hwid_short)
return ''
async def _handle_device_added(
self, db: AsyncSession, user: User, subscription: Subscription | None, data: dict
) -> None:
device_name = self._extract_device_name(data)
logger.info('Webhook: device added for user', user_id=user.id, device_name=device_name or '(empty)')
await self._notify_user(
user,
'WEBHOOK_DEVICE_ADDED',
reply_markup=self._get_subscription_keyboard(user),
format_kwargs={'device': device_name or ''},
subscription=subscription,
)
async def _handle_device_deleted(
self, db: AsyncSession, user: User, subscription: Subscription | None, data: dict
) -> None:
device_name = self._extract_device_name(data)
logger.info('Webhook: device deleted for user', user_id=user.id, device_name=device_name or '(empty)')
await self._notify_user(
user,
'WEBHOOK_DEVICE_DELETED',
reply_markup=self._get_subscription_keyboard(user),
format_kwargs={'device': device_name or ''},
subscription=subscription,
)