Solo_bot/database/models.py

import secrets
import uuid

from datetime import datetime

from sqlalchemy import (
    JSON,
    BigInteger,
    Boolean,
    Column,
    DateTime,
    Float,
    ForeignKey,
    Index,
    Integer,
    Numeric,
    String,
    Text,
    UniqueConstraint,
    text,
)
from sqlalchemy.dialects.postgresql import JSONB
from sqlalchemy.orm import Mapped, declarative_base, mapped_column, relationship


Base = declarative_base()


class DictLikeMixin:
    def __getitem__(self, key):
        return getattr(self, key)

    def get(self, key, default=None):
        return getattr(self, key, default)

    def to_dict(self):
        return {column.name: getattr(self, column.name) for column in self.__table__.columns}


class Identity(DictLikeMixin, Base):
    """Слой идентификации: к одному identity можно привязать email и/или Telegram (tg_id)."""

    __tablename__ = "identities"

    id = Column(String(36), primary_key=True, default=lambda: str(uuid.uuid4()))
    email = Column(String(255), unique=True, nullable=True, index=True)
    tg_id = Column(BigInteger, unique=True, nullable=True, index=True)
    api_token_hash = Column(String(64), nullable=True, index=True)
    token_issued_at = Column(DateTime, nullable=True)
    password_hash = Column(String(64), nullable=True)
    is_admin = Column(Boolean, nullable=False, server_default=text("false"))
    created_at = Column(DateTime, default=datetime.utcnow)
    updated_at = Column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)


class User(DictLikeMixin, Base):
    __tablename__ = "users"

    tg_id = Column(BigInteger, primary_key=True)
    identity_id = Column(
        String(36),
        ForeignKey("identities.id", ondelete="SET NULL", onupdate="CASCADE"),
        nullable=True,
        index=True,
    )
    username = Column(String)
    first_name = Column(String)
    last_name = Column(String)
    language_code = Column(String)
    is_bot = Column(Boolean, default=False)
    balance = Column(Float, default=0.0)
    trial = Column(Integer, default=0)
    preferred_currency = Column(String(10), nullable=False, server_default="RUB", index=True)
    source_code = Column(
        String,
        ForeignKey(
            "tracking_sources.code",
            ondelete="SET NULL",
            onupdate="CASCADE",
        ),
        nullable=True,
    )
    created_at = Column(DateTime, default=datetime.utcnow)
    updated_at = Column(DateTime, default=datetime.utcnow)


class Key(DictLikeMixin, Base):
    __tablename__ = "keys"

    tg_id = Column(BigInteger, ForeignKey("users.tg_id"), nullable=False, index=True)
    client_id = Column(String, primary_key=True)
    email = Column(String, unique=True)
    created_at = Column(BigInteger)
    expiry_time = Column(BigInteger)
    key = Column(String)
    server_id = Column(String)
    remnawave_link = Column(String)
    tariff_id = Column(Integer, ForeignKey("tariffs.id", ondelete="SET NULL"))
    is_frozen = Column(Boolean, default=False)
    alias = Column(String)
    notified = Column(Boolean, default=False)
    notified_24h = Column(Boolean, default=False)

    selected_device_limit = Column(Integer, nullable=True)
    selected_traffic_limit = Column(BigInteger, nullable=True)
    selected_price_rub = Column(Integer, nullable=True)

    current_device_limit = Column(Integer, nullable=True)
    current_traffic_limit = Column(BigInteger, nullable=True)


class Tariff(DictLikeMixin, Base):
    __tablename__ = "tariffs"

    id = Column(Integer, primary_key=True)
    name = Column(String)
    group_code = Column(String)
    duration_days = Column(Integer)
    price_rub = Column(Integer)
    traffic_limit = Column(BigInteger, nullable=True)
    device_limit = Column(Integer, nullable=True)
    is_active = Column(Boolean, default=True)
    created_at = Column(DateTime, default=datetime.utcnow)
    updated_at = Column(DateTime, default=datetime.utcnow)
    subgroup_title = Column(String, nullable=True)
    sort_order = Column(Integer, nullable=True)
    vless = Column(Boolean, default=False)
    external_squad: Mapped[str | None] = mapped_column(String(64), nullable=True)

    configurable = Column(Boolean, nullable=False, server_default="false")

    device_options = Column(JSONB, nullable=True)
    traffic_options_gb = Column(JSONB, nullable=True)

    device_step_rub = Column(Integer, nullable=True)
    device_overrides = Column(JSONB, nullable=True)

    traffic_step_rub = Column(Integer, nullable=True)
    traffic_overrides = Column(JSONB, nullable=True)


class Server(DictLikeMixin, Base):
    __tablename__ = "servers"

    id = Column(Integer, primary_key=True, autoincrement=True)
    cluster_name = Column(String)
    server_name = Column(String, unique=True)
    api_url = Column(String)
    subscription_url = Column(String)
    inbound_id = Column(String)
    panel_type = Column(String)
    max_keys = Column(Integer)
    tariff_group = Column(String)
    enabled = Column(Boolean, default=True)

    subgroups = relationship("ServerSubgroup", back_populates="server", cascade="all, delete-orphan")
    groups = relationship("ServerSpecialgroup", back_populates="server", cascade="all, delete-orphan")


class ServerSubgroup(DictLikeMixin, Base):
    __tablename__ = "server_subgroups"

    id = Column(Integer, primary_key=True, autoincrement=True)
    server_id = Column(Integer, ForeignKey("servers.id", ondelete="CASCADE"), index=True, nullable=False)
    group_code = Column(String, nullable=False)
    subgroup_title = Column(String, nullable=False)

    server = relationship("Server", back_populates="subgroups")

    __table_args__ = (UniqueConstraint("server_id", "subgroup_title", name="uq_server_subgroup"),)


class ServerSpecialgroup(DictLikeMixin, Base):
    __tablename__ = "server_specialgroups"

    id = Column(Integer, primary_key=True, autoincrement=True)
    server_id = Column(Integer, ForeignKey("servers.id", ondelete="CASCADE"), index=True, nullable=False)
    group_code = Column(String, nullable=False)

    server = relationship("Server")

    __table_args__ = (UniqueConstraint("server_id", "group_code", name="uq_server_group"),)


class Payment(DictLikeMixin, Base):
    __tablename__ = "payments"

    id = Column(Integer, primary_key=True, autoincrement=True)
    tg_id = Column(BigInteger, ForeignKey("users.tg_id"))
    amount = Column(Float)
    payment_system = Column(String)
    status = Column(String)
    created_at = Column(DateTime, default=datetime.utcnow)
    original_amount = Column(Numeric(18, 8), nullable=True)
    currency = Column(String(10), nullable=False, server_default="RUB")
    payment_id = Column(String(128), nullable=True, index=True)
    metadata_ = Column("metadata", JSONB, nullable=True)


class Coupon(DictLikeMixin, Base):
    __tablename__ = "coupons"

    id = Column(Integer, primary_key=True)
    code = Column(String, unique=True)
    amount = Column(Integer)
    usage_limit = Column(Integer)
    usage_count = Column(Integer, default=0)
    is_used = Column(Boolean, default=False)
    days = Column(Integer, nullable=True)
    new_users_only = Column(Boolean, nullable=False, server_default=text("false"))

    percent = Column(Integer, nullable=True)
    max_discount_amount = Column(Integer, nullable=True)
    min_order_amount = Column(Integer, nullable=True)


class CouponUsage(DictLikeMixin, Base):
    __tablename__ = "coupon_usages"

    coupon_id = Column(Integer, ForeignKey("coupons.id", ondelete="CASCADE"), primary_key=True)
    user_id = Column(BigInteger, primary_key=True)
    used_at = Column(DateTime, default=datetime.utcnow)


class Referral(DictLikeMixin, Base):
    __tablename__ = "referrals"

    referred_tg_id = Column(BigInteger, ForeignKey("users.tg_id", ondelete="CASCADE"), primary_key=True)
    referrer_tg_id = Column(BigInteger, ForeignKey("users.tg_id", ondelete="CASCADE"), primary_key=True)
    reward_issued = Column(Boolean, default=False)


class Notification(DictLikeMixin, Base):
    __tablename__ = "notifications"

    tg_id = Column(BigInteger, ForeignKey("users.tg_id", ondelete="CASCADE"), primary_key=True)
    notification_type = Column(String, primary_key=True)
    last_notification_time = Column(DateTime, default=datetime.utcnow)


class Gift(DictLikeMixin, Base):
    __tablename__ = "gifts"

    gift_id = Column(String, primary_key=True, default=lambda: uuid.uuid4().hex)
    sender_tg_id = Column(BigInteger, ForeignKey("users.tg_id"))
    recipient_tg_id = Column(BigInteger, ForeignKey("users.tg_id"), nullable=True)
    selected_months = Column(Integer)
    expiry_time = Column(DateTime)
    gift_link = Column(String)
    created_at = Column(DateTime, default=datetime.utcnow)
    is_used = Column(Boolean, default=False)
    is_unlimited = Column(Boolean, default=False)
    max_usages = Column(Integer, nullable=True)
    tariff_id: Mapped[int | None] = mapped_column(ForeignKey("tariffs.id"))

    selected_device_limit = Column(Integer, nullable=True)
    selected_traffic_gb = Column(Integer, nullable=True)
    selected_price_rub = Column(Integer, nullable=True)


class GiftUsage(DictLikeMixin, Base):
    __tablename__ = "gift_usages"

    gift_id = Column(String, ForeignKey("gifts.gift_id"), primary_key=True)
    tg_id = Column(BigInteger, primary_key=True)
    used_at = Column(DateTime, default=datetime.utcnow)


class ManualBan(DictLikeMixin, Base):
    __tablename__ = "manual_bans"

    tg_id = Column(BigInteger, primary_key=True)
    banned_at = Column(DateTime(timezone=True), default=datetime.utcnow)
    reason = Column(Text)
    banned_by = Column(BigInteger)
    until = Column(DateTime(timezone=True), nullable=True)


class TemporaryData(DictLikeMixin, Base):
    __tablename__ = "temporary_data"

    tg_id = Column(BigInteger, primary_key=True)
    state = Column(String)
    data = Column(JSON)
    updated_at = Column(DateTime, default=datetime.utcnow)


class BlockedUser(DictLikeMixin, Base):
    __tablename__ = "blocked_users"

    tg_id = Column(BigInteger, primary_key=True)


class TrackingSource(DictLikeMixin, Base):
    __tablename__ = "tracking_sources"

    id = Column(Integer, primary_key=True)
    name = Column(String)
    code = Column(String, unique=True)
    type = Column(String)
    created_by = Column(BigInteger)
    created_at = Column(DateTime, default=datetime.utcnow)


class AuditEvent(DictLikeMixin, Base):
    """События аудита (флоу пользователя)."""
    __tablename__ = "audit_events"
    __table_args__ = (
        Index("ix_audit_events_tg_created", "actor_tg_id", "created_at"),
        Index("ix_audit_events_identity_created", "actor_identity_id", "created_at"),
    )

    id = Column(Integer, primary_key=True, autoincrement=True)
    event_type = Column(String(64), nullable=False, index=True)
    channel = Column(String(32), nullable=False, index=True)
    actor_identity_id = Column(
        String(36),
        ForeignKey("identities.id", ondelete="SET NULL", onupdate="CASCADE"),
        nullable=True,
        index=True,
    )
    actor_tg_id = Column(BigInteger, nullable=True, index=True)
    path_or_handler = Column(String(255), nullable=False)
    entity_type = Column(String(64), nullable=True, index=True)
    entity_id = Column(String(255), nullable=True, index=True)
    result = Column(String(32), nullable=False, server_default=text("'success'"))
    reason = Column(Text, nullable=True)
    metadata_ = Column("metadata", JSONB, nullable=True)
    request_id = Column(String(64), nullable=True, index=True)
    created_at = Column(DateTime, default=datetime.utcnow, index=True)


class Admin(Base):
    __tablename__ = "admins"

    tg_id = Column(BigInteger, primary_key=True)
    token = Column(String, unique=True, nullable=True)
    description = Column(String, nullable=True)
    role = Column(String, nullable=False, default="admin")
    added_at = Column(DateTime, default=datetime.utcnow)

    @staticmethod
    def generate_token() -> str:
        return secrets.token_urlsafe(32)


class Setting(DictLikeMixin, Base):
    __tablename__ = "settings"

    key = Column(String, primary_key=True)
    value = Column(JSONB, nullable=True)
    description = Column(Text, nullable=True)
    created_at = Column(DateTime, default=datetime.utcnow)
    updated_at = Column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)


class WebPage(DictLikeMixin, Base):
    __tablename__ = "web_pages"

    slug = Column(String(64), primary_key=True)
    title = Column(String(255), nullable=True)


class WebTheme(DictLikeMixin, Base):
    __tablename__ = "web_themes"

    page_slug = Column(String(64), ForeignKey("web_pages.slug", ondelete="CASCADE"), primary_key=True)
    tokens = Column(JSONB, nullable=False, default=dict)


class WebBlock(DictLikeMixin, Base):
    __tablename__ = "web_blocks"

    id = Column(String(36), primary_key=True, default=lambda: str(uuid.uuid4()))
    page_slug = Column(String(64), ForeignKey("web_pages.slug", ondelete="CASCADE"), index=True, nullable=False)
    order = Column(Integer, nullable=False, default=0)
    type = Column(String(64), nullable=False)
    data = Column(JSONB, nullable=False, default=dict)