v2.0
This commit is contained in:
811
app/services/sync_service.py
Normal file
811
app/services/sync_service.py
Normal file
@@ -0,0 +1,811 @@
|
||||
from __future__ import annotations
|
||||
|
||||
import secrets
|
||||
import math
|
||||
from dataclasses import dataclass, field
|
||||
from datetime import datetime, timezone
|
||||
|
||||
from sqlalchemy import delete, func, or_, select
|
||||
from sqlalchemy.ext.asyncio import AsyncSession, async_sessionmaker
|
||||
|
||||
from app.config import Settings
|
||||
from app.db.base import utcnow
|
||||
from app.db.models import (
|
||||
ReferralCode,
|
||||
ReferralInvite,
|
||||
InternalSquad,
|
||||
RemnawaveUser,
|
||||
RemnawaveUserInternalSquad,
|
||||
SubscriptionRequestLog,
|
||||
TelegramUser,
|
||||
)
|
||||
from app.schemas.remnawave import RemnawaveUser as RemoteRemnawaveUser
|
||||
from app.schemas.remnawave import SubscriptionRequestHistory
|
||||
from app.services.remnawave_client import RemnawaveApiClient
|
||||
|
||||
|
||||
@dataclass(slots=True)
|
||||
class CachedUserView:
|
||||
record: RemnawaveUser
|
||||
internal_squads: list[str] = field(default_factory=list)
|
||||
recent_requests: list[SubscriptionRequestLog] = field(default_factory=list)
|
||||
|
||||
|
||||
@dataclass(slots=True)
|
||||
class ReferralSummary:
|
||||
total_invited: int = 0
|
||||
referral_code: str = ""
|
||||
applied_referral_code: str = ""
|
||||
recent_names: list[str] = field(default_factory=list)
|
||||
|
||||
|
||||
@dataclass(slots=True)
|
||||
class AdminStats:
|
||||
total_telegram_users: int = 0
|
||||
total_cached_users: int = 0
|
||||
active_cached_users: int = 0
|
||||
|
||||
|
||||
@dataclass(slots=True)
|
||||
class AdminTelegramUserListItem:
|
||||
telegram_id: int
|
||||
username: str | None
|
||||
first_name: str | None
|
||||
last_name: str | None
|
||||
language_code: str | None
|
||||
is_admin: bool
|
||||
is_blocked: bool
|
||||
last_seen_at: datetime
|
||||
created_at: datetime
|
||||
|
||||
@property
|
||||
def display_name(self) -> str:
|
||||
value = " ".join(part for part in [self.first_name, self.last_name] if part).strip()
|
||||
return value or self.username or str(self.telegram_id)
|
||||
|
||||
|
||||
@dataclass(slots=True)
|
||||
class AdminTelegramUsersPage:
|
||||
items: list[AdminTelegramUserListItem] = field(default_factory=list)
|
||||
page: int = 1
|
||||
page_size: int = 8
|
||||
total_items: int = 0
|
||||
total_pages: int = 1
|
||||
|
||||
|
||||
@dataclass(slots=True)
|
||||
class AdminTelegramUserProfile:
|
||||
user: TelegramUser
|
||||
referral_code: str = ""
|
||||
applied_referral_code: str = ""
|
||||
invited_count: int = 0
|
||||
recent_invited_names: list[str] = field(default_factory=list)
|
||||
accesses: list[CachedUserView] = field(default_factory=list)
|
||||
|
||||
|
||||
class SyncService:
|
||||
def __init__(
|
||||
self,
|
||||
*,
|
||||
settings: Settings,
|
||||
session_factory: async_sessionmaker[AsyncSession],
|
||||
remnawave_client: RemnawaveApiClient,
|
||||
) -> None:
|
||||
self._settings = settings
|
||||
self._session_factory = session_factory
|
||||
self._remnawave_client = remnawave_client
|
||||
|
||||
async def is_user_blocked(self, telegram_id: int) -> bool:
|
||||
async with self._session_factory() as session:
|
||||
is_blocked = await session.scalar(
|
||||
select(TelegramUser.is_blocked).where(
|
||||
TelegramUser.telegram_id == telegram_id
|
||||
)
|
||||
)
|
||||
return bool(is_blocked)
|
||||
|
||||
async def block_telegram_user(self, telegram_id: int) -> None:
|
||||
async with self._session_factory() as session:
|
||||
user = await session.scalar(
|
||||
select(TelegramUser).where(TelegramUser.telegram_id == telegram_id)
|
||||
)
|
||||
if user is not None:
|
||||
user.is_blocked = True
|
||||
user.updated_at = utcnow()
|
||||
await session.commit()
|
||||
|
||||
async def unblock_telegram_user(self, telegram_id: int) -> None:
|
||||
async with self._session_factory() as session:
|
||||
user = await session.scalar(
|
||||
select(TelegramUser).where(TelegramUser.telegram_id == telegram_id)
|
||||
)
|
||||
if user is not None:
|
||||
user.is_blocked = False
|
||||
user.updated_at = utcnow()
|
||||
await session.commit()
|
||||
|
||||
async def get_expiring_users(
|
||||
self,
|
||||
*,
|
||||
days_before: int,
|
||||
tolerance_hours: int = 6,
|
||||
) -> list[tuple[int, int, datetime]]:
|
||||
"""Return (remnawave_user.id, telegram_id, expire_at) for users expiring within window."""
|
||||
from datetime import timedelta
|
||||
|
||||
now = utcnow()
|
||||
window_start = now + timedelta(days=days_before) - timedelta(hours=tolerance_hours)
|
||||
window_end = now + timedelta(days=days_before) + timedelta(hours=tolerance_hours)
|
||||
|
||||
async with self._session_factory() as session:
|
||||
rows = await session.execute(
|
||||
select(
|
||||
RemnawaveUser.id,
|
||||
RemnawaveUser.telegram_id,
|
||||
RemnawaveUser.expire_at,
|
||||
)
|
||||
.where(
|
||||
RemnawaveUser.status == "ACTIVE",
|
||||
RemnawaveUser.telegram_id.isnot(None),
|
||||
RemnawaveUser.expire_at >= window_start,
|
||||
RemnawaveUser.expire_at <= window_end,
|
||||
)
|
||||
)
|
||||
return [(row[0], row[1], row[2]) for row in rows.all()]
|
||||
|
||||
async def register_telegram_user(
|
||||
self,
|
||||
*,
|
||||
telegram_id: int,
|
||||
username: str | None,
|
||||
first_name: str | None,
|
||||
last_name: str | None,
|
||||
language_code: str | None,
|
||||
) -> None:
|
||||
async with self._session_factory() as session:
|
||||
await self._upsert_telegram_user(
|
||||
session,
|
||||
telegram_id=telegram_id,
|
||||
username=username,
|
||||
first_name=first_name,
|
||||
last_name=last_name,
|
||||
language_code=language_code,
|
||||
)
|
||||
await self._get_or_create_referral_code(session, telegram_id=telegram_id)
|
||||
await session.commit()
|
||||
|
||||
async def apply_referral_code(
|
||||
self,
|
||||
*,
|
||||
telegram_id: int,
|
||||
username: str | None,
|
||||
first_name: str | None,
|
||||
last_name: str | None,
|
||||
language_code: str | None,
|
||||
referral_code: str,
|
||||
) -> str:
|
||||
async with self._session_factory() as session:
|
||||
invited = await self._upsert_telegram_user(
|
||||
session,
|
||||
telegram_id=telegram_id,
|
||||
username=username,
|
||||
first_name=first_name,
|
||||
last_name=last_name,
|
||||
language_code=language_code,
|
||||
)
|
||||
await self._get_or_create_referral_code(session, telegram_id=telegram_id)
|
||||
|
||||
existing = await session.scalar(
|
||||
select(ReferralInvite).where(
|
||||
ReferralInvite.invited_telegram_id == telegram_id
|
||||
)
|
||||
)
|
||||
if existing is not None:
|
||||
raise ValueError("Реферальный код уже применен к вашему аккаунту.")
|
||||
|
||||
inviter_telegram_id = await self._find_inviter_by_referral_code(session, referral_code)
|
||||
if inviter_telegram_id is None:
|
||||
raise ValueError("Реферальный код не найден.")
|
||||
if inviter_telegram_id == telegram_id:
|
||||
raise ValueError("Нельзя использовать собственный реферальный код.")
|
||||
|
||||
invite_created = await self._upsert_referral_invite(
|
||||
session,
|
||||
invited=invited,
|
||||
referred_by_telegram_id=inviter_telegram_id,
|
||||
)
|
||||
if not invite_created:
|
||||
raise ValueError("Не удалось применить реферальный код.")
|
||||
|
||||
await session.commit()
|
||||
return referral_code.strip().upper()
|
||||
|
||||
async def get_referral_summary(self, telegram_id: int) -> ReferralSummary:
|
||||
async with self._session_factory() as session:
|
||||
referral_code = await self._get_or_create_referral_code(
|
||||
session,
|
||||
telegram_id=telegram_id,
|
||||
)
|
||||
await session.commit()
|
||||
total_invited = await session.scalar(
|
||||
select(func.count(ReferralInvite.id)).where(
|
||||
ReferralInvite.inviter_telegram_id == telegram_id
|
||||
)
|
||||
)
|
||||
recent_rows = await session.execute(
|
||||
select(ReferralInvite.invited_display_name)
|
||||
.where(ReferralInvite.inviter_telegram_id == telegram_id)
|
||||
.order_by(ReferralInvite.created_at.desc())
|
||||
.limit(5)
|
||||
)
|
||||
recent_names = list(recent_rows.scalars().all())
|
||||
applied_referral_code = await session.scalar(
|
||||
select(ReferralCode.code)
|
||||
.join(
|
||||
ReferralInvite,
|
||||
ReferralInvite.inviter_telegram_id == ReferralCode.telegram_id,
|
||||
)
|
||||
.where(ReferralInvite.invited_telegram_id == telegram_id)
|
||||
)
|
||||
|
||||
return ReferralSummary(
|
||||
total_invited=int(total_invited or 0),
|
||||
referral_code=referral_code,
|
||||
applied_referral_code=applied_referral_code or "",
|
||||
recent_names=recent_names,
|
||||
)
|
||||
|
||||
async def get_admin_stats(self) -> AdminStats:
|
||||
async with self._session_factory() as session:
|
||||
total_telegram_users = await session.scalar(
|
||||
select(func.count(TelegramUser.id))
|
||||
)
|
||||
total_cached_users = await session.scalar(
|
||||
select(func.count(RemnawaveUser.id))
|
||||
)
|
||||
active_cached_users = await session.scalar(
|
||||
select(func.count(RemnawaveUser.id)).where(RemnawaveUser.status == "ACTIVE")
|
||||
)
|
||||
|
||||
return AdminStats(
|
||||
total_telegram_users=int(total_telegram_users or 0),
|
||||
total_cached_users=int(total_cached_users or 0),
|
||||
active_cached_users=int(active_cached_users or 0),
|
||||
)
|
||||
|
||||
async def get_admin_telegram_users_page(
|
||||
self,
|
||||
*,
|
||||
page: int,
|
||||
page_size: int = 8,
|
||||
) -> AdminTelegramUsersPage:
|
||||
resolved_page_size = max(1, min(page_size, 20))
|
||||
|
||||
async with self._session_factory() as session:
|
||||
total_items = int(
|
||||
await session.scalar(select(func.count(TelegramUser.id))) or 0
|
||||
)
|
||||
total_pages = max(1, math.ceil(total_items / resolved_page_size)) if total_items else 1
|
||||
resolved_page = min(max(page, 1), total_pages)
|
||||
|
||||
rows = list(
|
||||
(
|
||||
await session.scalars(
|
||||
select(TelegramUser)
|
||||
.order_by(
|
||||
TelegramUser.last_seen_at.desc(),
|
||||
TelegramUser.created_at.desc(),
|
||||
TelegramUser.id.desc(),
|
||||
)
|
||||
.offset((resolved_page - 1) * resolved_page_size)
|
||||
.limit(resolved_page_size)
|
||||
)
|
||||
).all()
|
||||
)
|
||||
|
||||
return AdminTelegramUsersPage(
|
||||
items=[
|
||||
AdminTelegramUserListItem(
|
||||
telegram_id=row.telegram_id,
|
||||
username=row.username,
|
||||
first_name=row.first_name,
|
||||
last_name=row.last_name,
|
||||
language_code=row.language_code,
|
||||
is_admin=row.is_admin,
|
||||
is_blocked=row.is_blocked,
|
||||
last_seen_at=row.last_seen_at,
|
||||
created_at=row.created_at,
|
||||
)
|
||||
for row in rows
|
||||
],
|
||||
page=resolved_page,
|
||||
page_size=resolved_page_size,
|
||||
total_items=total_items,
|
||||
total_pages=total_pages,
|
||||
)
|
||||
|
||||
async def get_admin_telegram_user_profile(
|
||||
self,
|
||||
*,
|
||||
telegram_id: int,
|
||||
) -> AdminTelegramUserProfile | None:
|
||||
async with self._session_factory() as session:
|
||||
user = await session.scalar(
|
||||
select(TelegramUser).where(TelegramUser.telegram_id == telegram_id)
|
||||
)
|
||||
if user is None:
|
||||
return None
|
||||
|
||||
referral_code = await session.scalar(
|
||||
select(ReferralCode.code).where(ReferralCode.telegram_id == telegram_id)
|
||||
)
|
||||
invited_count = await session.scalar(
|
||||
select(func.count(ReferralInvite.id)).where(
|
||||
ReferralInvite.inviter_telegram_id == telegram_id
|
||||
)
|
||||
)
|
||||
recent_rows = await session.execute(
|
||||
select(ReferralInvite.invited_display_name)
|
||||
.where(ReferralInvite.inviter_telegram_id == telegram_id)
|
||||
.order_by(ReferralInvite.created_at.desc())
|
||||
.limit(5)
|
||||
)
|
||||
recent_invited_names = list(recent_rows.scalars().all())
|
||||
applied_referral_code = await session.scalar(
|
||||
select(ReferralCode.code)
|
||||
.join(
|
||||
ReferralInvite,
|
||||
ReferralInvite.inviter_telegram_id == ReferralCode.telegram_id,
|
||||
)
|
||||
.where(ReferralInvite.invited_telegram_id == telegram_id)
|
||||
)
|
||||
|
||||
accesses = await self.get_cached_users_for_telegram(telegram_id)
|
||||
|
||||
return AdminTelegramUserProfile(
|
||||
user=user,
|
||||
referral_code=referral_code or "",
|
||||
applied_referral_code=applied_referral_code or "",
|
||||
invited_count=int(invited_count or 0),
|
||||
recent_invited_names=recent_invited_names,
|
||||
accesses=accesses,
|
||||
)
|
||||
|
||||
async def sync_users_for_telegram(
|
||||
self,
|
||||
*,
|
||||
telegram_id: int,
|
||||
username: str | None,
|
||||
first_name: str | None,
|
||||
last_name: str | None,
|
||||
language_code: str | None,
|
||||
) -> list[CachedUserView]:
|
||||
remote_users = await self._remnawave_client.get_users_by_telegram_id(telegram_id)
|
||||
|
||||
async with self._session_factory() as session:
|
||||
owner = await self._upsert_telegram_user(
|
||||
session,
|
||||
telegram_id=telegram_id,
|
||||
username=username,
|
||||
first_name=first_name,
|
||||
last_name=last_name,
|
||||
language_code=language_code,
|
||||
)
|
||||
await self._sync_remote_users_for_telegram(
|
||||
session,
|
||||
telegram_id=telegram_id,
|
||||
remote_users=remote_users,
|
||||
owner=owner,
|
||||
)
|
||||
|
||||
await session.commit()
|
||||
|
||||
return await self.get_cached_users_for_telegram(telegram_id)
|
||||
|
||||
async def refresh_cached_users_for_telegram(self, *, telegram_id: int) -> list[CachedUserView]:
|
||||
remote_users = await self._remnawave_client.get_users_by_telegram_id(telegram_id)
|
||||
|
||||
async with self._session_factory() as session:
|
||||
owner = await session.scalar(
|
||||
select(TelegramUser).where(TelegramUser.telegram_id == telegram_id)
|
||||
)
|
||||
await self._sync_remote_users_for_telegram(
|
||||
session,
|
||||
telegram_id=telegram_id,
|
||||
remote_users=remote_users,
|
||||
owner=owner,
|
||||
)
|
||||
await session.commit()
|
||||
|
||||
return await self.get_cached_users_for_telegram(telegram_id)
|
||||
|
||||
async def link_user_by_short_uuid(
|
||||
self,
|
||||
*,
|
||||
short_uuid: str,
|
||||
telegram_id: int,
|
||||
username: str | None,
|
||||
first_name: str | None,
|
||||
last_name: str | None,
|
||||
language_code: str | None,
|
||||
) -> CachedUserView:
|
||||
remote_user = await self._remnawave_client.get_user_by_short_uuid(short_uuid)
|
||||
|
||||
if remote_user.telegram_id and remote_user.telegram_id != telegram_id:
|
||||
raise ValueError("Этот аккаунт уже привязан к другому Telegram ID.")
|
||||
|
||||
updated_user = await self._remnawave_client.update_user(
|
||||
{
|
||||
"uuid": str(remote_user.uuid),
|
||||
"telegramId": telegram_id,
|
||||
}
|
||||
)
|
||||
|
||||
async with self._session_factory() as session:
|
||||
owner = await self._upsert_telegram_user(
|
||||
session,
|
||||
telegram_id=telegram_id,
|
||||
username=username,
|
||||
first_name=first_name,
|
||||
last_name=last_name,
|
||||
language_code=language_code,
|
||||
)
|
||||
await self._upsert_remnawave_user(session, updated_user, owner)
|
||||
if self._settings.sync_subscription_history:
|
||||
history = await self._remnawave_client.get_user_subscription_history(
|
||||
str(updated_user.uuid)
|
||||
)
|
||||
await self._upsert_history(session, str(updated_user.uuid), history)
|
||||
await session.commit()
|
||||
|
||||
user = await self.get_cached_user_by_uuid(str(updated_user.uuid), include_history=True)
|
||||
if user is None:
|
||||
raise ValueError("Не удалось получить локальный кэш пользователя после привязки.")
|
||||
return user
|
||||
|
||||
async def sync_user_by_identifier(
|
||||
self,
|
||||
*,
|
||||
identifier: str,
|
||||
include_history: bool = True,
|
||||
) -> CachedUserView:
|
||||
resolved = await self._remnawave_client.resolve_user(identifier)
|
||||
remote_user = await self._remnawave_client.get_user_by_uuid(str(resolved.uuid))
|
||||
history = None
|
||||
|
||||
if include_history and self._settings.sync_subscription_history:
|
||||
history = await self._remnawave_client.get_user_subscription_history(str(resolved.uuid))
|
||||
|
||||
async with self._session_factory() as session:
|
||||
await self._upsert_remnawave_user(session, remote_user, owner=None)
|
||||
if history is not None:
|
||||
await self._upsert_history(session, str(resolved.uuid), history)
|
||||
await session.commit()
|
||||
|
||||
user = await self.get_cached_user_by_uuid(str(resolved.uuid), include_history=include_history)
|
||||
if user is None:
|
||||
raise ValueError("Не удалось получить локальный кэш пользователя после синхронизации.")
|
||||
return user
|
||||
|
||||
async def sync_all_users(self, batch_size: int) -> int:
|
||||
start = 0
|
||||
total_synced = 0
|
||||
|
||||
while True:
|
||||
page = await self._remnawave_client.get_all_users(start=start, size=batch_size)
|
||||
if not page.users:
|
||||
break
|
||||
|
||||
async with self._session_factory() as session:
|
||||
for remote_user in page.users:
|
||||
await self._upsert_remnawave_user(session, remote_user, owner=None)
|
||||
total_synced += 1
|
||||
await session.commit()
|
||||
|
||||
start += len(page.users)
|
||||
if start >= page.total:
|
||||
break
|
||||
|
||||
return total_synced
|
||||
|
||||
async def get_cached_users_for_telegram(self, telegram_id: int) -> list[CachedUserView]:
|
||||
async with self._session_factory() as session:
|
||||
rows = await session.execute(
|
||||
select(RemnawaveUser, InternalSquad.name)
|
||||
.outerjoin(
|
||||
RemnawaveUserInternalSquad,
|
||||
RemnawaveUserInternalSquad.user_id == RemnawaveUser.id,
|
||||
)
|
||||
.outerjoin(
|
||||
InternalSquad,
|
||||
InternalSquad.uuid == RemnawaveUserInternalSquad.squad_uuid,
|
||||
)
|
||||
.where(RemnawaveUser.telegram_id == telegram_id)
|
||||
.order_by(RemnawaveUser.username.asc())
|
||||
)
|
||||
return self._build_views(rows.all())
|
||||
|
||||
async def get_cached_user_by_uuid(
|
||||
self,
|
||||
user_uuid: str,
|
||||
*,
|
||||
include_history: bool = False,
|
||||
) -> CachedUserView | None:
|
||||
async with self._session_factory() as session:
|
||||
rows = await session.execute(
|
||||
select(RemnawaveUser, InternalSquad.name)
|
||||
.outerjoin(
|
||||
RemnawaveUserInternalSquad,
|
||||
RemnawaveUserInternalSquad.user_id == RemnawaveUser.id,
|
||||
)
|
||||
.outerjoin(
|
||||
InternalSquad,
|
||||
InternalSquad.uuid == RemnawaveUserInternalSquad.squad_uuid,
|
||||
)
|
||||
.where(RemnawaveUser.rw_uuid == user_uuid)
|
||||
)
|
||||
views = self._build_views(rows.all())
|
||||
if not views:
|
||||
return None
|
||||
|
||||
view = views[0]
|
||||
if include_history:
|
||||
history_rows = await session.execute(
|
||||
select(SubscriptionRequestLog)
|
||||
.where(SubscriptionRequestLog.user_uuid == user_uuid)
|
||||
.order_by(SubscriptionRequestLog.request_at.desc())
|
||||
.limit(5)
|
||||
)
|
||||
view.recent_requests = list(history_rows.scalars().all())
|
||||
return view
|
||||
|
||||
async def _upsert_telegram_user(
|
||||
self,
|
||||
session: AsyncSession,
|
||||
*,
|
||||
telegram_id: int,
|
||||
username: str | None,
|
||||
first_name: str | None,
|
||||
last_name: str | None,
|
||||
language_code: str | None,
|
||||
) -> TelegramUser:
|
||||
record = await session.scalar(
|
||||
select(TelegramUser).where(TelegramUser.telegram_id == telegram_id)
|
||||
)
|
||||
|
||||
if record is None:
|
||||
record = TelegramUser(telegram_id=telegram_id)
|
||||
session.add(record)
|
||||
|
||||
record.username = username
|
||||
record.first_name = first_name
|
||||
record.last_name = last_name
|
||||
record.language_code = language_code
|
||||
record.is_admin = self._settings.is_admin(telegram_id)
|
||||
record.last_seen_at = utcnow()
|
||||
record.updated_at = utcnow()
|
||||
|
||||
await session.flush()
|
||||
return record
|
||||
|
||||
async def _sync_remote_users_for_telegram(
|
||||
self,
|
||||
session: AsyncSession,
|
||||
*,
|
||||
telegram_id: int,
|
||||
remote_users: list[RemoteRemnawaveUser],
|
||||
owner: TelegramUser | None,
|
||||
) -> None:
|
||||
active_uuids = {str(remote_user.uuid) for remote_user in remote_users}
|
||||
for remote_user in remote_users:
|
||||
await self._upsert_remnawave_user(session, remote_user, owner)
|
||||
|
||||
filters = [RemnawaveUser.telegram_id == telegram_id]
|
||||
if owner is not None:
|
||||
filters.append(RemnawaveUser.owner_telegram_user_id == owner.id)
|
||||
|
||||
stale_records = await session.scalars(
|
||||
select(RemnawaveUser).where(or_(*filters))
|
||||
)
|
||||
for record in stale_records:
|
||||
if record.rw_uuid in active_uuids:
|
||||
continue
|
||||
|
||||
if record.telegram_id == telegram_id:
|
||||
record.telegram_id = None
|
||||
if owner is not None and record.owner_telegram_user_id == owner.id:
|
||||
record.owner_telegram_user_id = None
|
||||
record.synced_at = utcnow()
|
||||
|
||||
async def _upsert_remnawave_user(
|
||||
self,
|
||||
session: AsyncSession,
|
||||
remote_user: RemoteRemnawaveUser,
|
||||
owner: TelegramUser | None,
|
||||
) -> RemnawaveUser:
|
||||
record = await session.scalar(
|
||||
select(RemnawaveUser).where(RemnawaveUser.rw_uuid == str(remote_user.uuid))
|
||||
)
|
||||
|
||||
if record is None:
|
||||
record = RemnawaveUser(rw_uuid=str(remote_user.uuid))
|
||||
session.add(record)
|
||||
|
||||
owner_id = owner.id if owner is not None else await self._find_owner_id(session, remote_user.telegram_id)
|
||||
|
||||
record.owner_telegram_user_id = owner_id
|
||||
record.rw_id = remote_user.id
|
||||
record.short_uuid = remote_user.short_uuid
|
||||
record.username = remote_user.username
|
||||
record.status = remote_user.status
|
||||
record.traffic_limit_bytes = remote_user.traffic_limit_bytes
|
||||
record.used_traffic_bytes = remote_user.user_traffic.used_traffic_bytes
|
||||
record.lifetime_used_traffic_bytes = remote_user.user_traffic.lifetime_used_traffic_bytes
|
||||
record.expire_at = self._normalize_datetime(remote_user.expire_at)
|
||||
record.telegram_id = remote_user.telegram_id
|
||||
record.email = remote_user.email
|
||||
record.description = remote_user.description
|
||||
record.tag = remote_user.tag
|
||||
record.hwid_device_limit = remote_user.hwid_device_limit
|
||||
record.external_squad_uuid = self._maybe_str(remote_user.external_squad_uuid)
|
||||
record.trojan_password = remote_user.trojan_password
|
||||
record.vless_uuid = str(remote_user.vless_uuid)
|
||||
record.ss_password = remote_user.ss_password
|
||||
record.last_triggered_threshold = remote_user.last_triggered_threshold
|
||||
record.sub_revoked_at = self._normalize_datetime(remote_user.sub_revoked_at)
|
||||
record.sub_last_user_agent = remote_user.sub_last_user_agent
|
||||
record.sub_last_opened_at = self._normalize_datetime(remote_user.sub_last_opened_at)
|
||||
record.last_traffic_reset_at = self._normalize_datetime(remote_user.last_traffic_reset_at)
|
||||
record.subscription_url = remote_user.subscription_url
|
||||
record.online_at = self._normalize_datetime(remote_user.user_traffic.online_at)
|
||||
record.first_connected_at = self._normalize_datetime(remote_user.user_traffic.first_connected_at)
|
||||
record.last_connected_node_uuid = self._maybe_str(remote_user.user_traffic.last_connected_node_uuid)
|
||||
record.remote_created_at = self._normalize_datetime(remote_user.created_at)
|
||||
record.remote_updated_at = self._normalize_datetime(remote_user.updated_at)
|
||||
record.synced_at = utcnow()
|
||||
|
||||
await session.flush()
|
||||
await session.execute(
|
||||
delete(RemnawaveUserInternalSquad).where(
|
||||
RemnawaveUserInternalSquad.user_id == record.id
|
||||
)
|
||||
)
|
||||
|
||||
for squad in remote_user.active_internal_squads:
|
||||
squad_uuid = str(squad.uuid)
|
||||
squad_record = await session.get(InternalSquad, squad_uuid)
|
||||
if squad_record is None:
|
||||
squad_record = InternalSquad(uuid=squad_uuid, name=squad.name, synced_at=utcnow())
|
||||
session.add(squad_record)
|
||||
else:
|
||||
squad_record.name = squad.name
|
||||
squad_record.synced_at = utcnow()
|
||||
|
||||
session.add(
|
||||
RemnawaveUserInternalSquad(
|
||||
user_id=record.id,
|
||||
squad_uuid=squad_uuid,
|
||||
)
|
||||
)
|
||||
|
||||
await session.flush()
|
||||
return record
|
||||
|
||||
async def _upsert_history(
|
||||
self,
|
||||
session: AsyncSession,
|
||||
user_uuid: str,
|
||||
history: SubscriptionRequestHistory,
|
||||
) -> None:
|
||||
for item in history.records:
|
||||
existing = await session.scalar(
|
||||
select(SubscriptionRequestLog).where(
|
||||
SubscriptionRequestLog.remote_id == item.id
|
||||
)
|
||||
)
|
||||
if existing is None:
|
||||
existing = SubscriptionRequestLog(remote_id=item.id, user_uuid=user_uuid)
|
||||
session.add(existing)
|
||||
|
||||
existing.user_uuid = user_uuid
|
||||
existing.request_at = self._normalize_datetime(item.request_at)
|
||||
existing.request_ip = item.request_ip
|
||||
existing.user_agent = item.user_agent
|
||||
existing.synced_at = utcnow()
|
||||
|
||||
async def _upsert_referral_invite(
|
||||
self,
|
||||
session: AsyncSession,
|
||||
*,
|
||||
invited: TelegramUser,
|
||||
referred_by_telegram_id: int | None,
|
||||
) -> bool:
|
||||
if referred_by_telegram_id is None or referred_by_telegram_id == invited.telegram_id:
|
||||
return False
|
||||
|
||||
existing = await session.scalar(
|
||||
select(ReferralInvite).where(
|
||||
ReferralInvite.invited_telegram_id == invited.telegram_id
|
||||
)
|
||||
)
|
||||
if existing is not None:
|
||||
return False
|
||||
|
||||
display_name = " ".join(
|
||||
part for part in [invited.first_name, invited.last_name] if part
|
||||
).strip() or invited.username or str(invited.telegram_id)
|
||||
|
||||
session.add(
|
||||
ReferralInvite(
|
||||
inviter_telegram_id=referred_by_telegram_id,
|
||||
invited_telegram_id=invited.telegram_id,
|
||||
invited_username=invited.username,
|
||||
invited_display_name=display_name,
|
||||
)
|
||||
)
|
||||
return True
|
||||
|
||||
async def _get_or_create_referral_code(
|
||||
self,
|
||||
session: AsyncSession,
|
||||
*,
|
||||
telegram_id: int,
|
||||
) -> str:
|
||||
existing = await session.scalar(
|
||||
select(ReferralCode).where(ReferralCode.telegram_id == telegram_id)
|
||||
)
|
||||
if existing is not None:
|
||||
return existing.code
|
||||
|
||||
alphabet = "ABCDEFGHJKLMNPQRSTUVWXYZ23456789"
|
||||
while True:
|
||||
code = "".join(secrets.choice(alphabet) for _ in range(8))
|
||||
duplicate = await session.scalar(
|
||||
select(ReferralCode.id).where(ReferralCode.code == code)
|
||||
)
|
||||
if duplicate is None:
|
||||
session.add(ReferralCode(telegram_id=telegram_id, code=code))
|
||||
await session.flush()
|
||||
return code
|
||||
|
||||
async def _find_inviter_by_referral_code(
|
||||
self,
|
||||
session: AsyncSession,
|
||||
referred_by_referral_code: str | None,
|
||||
) -> int | None:
|
||||
raw_code = (referred_by_referral_code or "").strip().upper()
|
||||
if not raw_code:
|
||||
return None
|
||||
|
||||
return await session.scalar(
|
||||
select(ReferralCode.telegram_id).where(ReferralCode.code == raw_code)
|
||||
)
|
||||
|
||||
async def _find_owner_id(self, session: AsyncSession, telegram_id: int | None) -> int | None:
|
||||
if telegram_id is None:
|
||||
return None
|
||||
|
||||
return await session.scalar(
|
||||
select(TelegramUser.id).where(TelegramUser.telegram_id == telegram_id)
|
||||
)
|
||||
|
||||
@staticmethod
|
||||
def _normalize_datetime(value: datetime | None) -> datetime | None:
|
||||
if value is None:
|
||||
return None
|
||||
if value.tzinfo is None:
|
||||
return value
|
||||
return value.astimezone(timezone.utc).replace(tzinfo=None)
|
||||
|
||||
@staticmethod
|
||||
def _maybe_str(value: object | None) -> str | None:
|
||||
return str(value) if value is not None else None
|
||||
|
||||
@staticmethod
|
||||
def _build_views(rows: list[tuple[RemnawaveUser, str | None]]) -> list[CachedUserView]:
|
||||
view_map: dict[int, CachedUserView] = {}
|
||||
for user, squad_name in rows:
|
||||
view = view_map.setdefault(user.id, CachedUserView(record=user))
|
||||
if squad_name and squad_name not in view.internal_squads:
|
||||
view.internal_squads.append(squad_name)
|
||||
return list(view_map.values())
|
||||
Reference in New Issue
Block a user