cyberia/research/events/backend/cyber_valley/users/models.py

from typing import TYPE_CHECKING, Any, ClassVar

from django.contrib.auth.base_user import AbstractBaseUser
from django.db import models

if TYPE_CHECKING:
    CharFieldType = models.CharField[str, str]
else:
    CharFieldType = models.CharField


class AddressField(CharFieldType):
    def __init__(self, *args: Any, **kwargs: Any) -> None:
        kwargs["max_length"] = 42
        super().__init__(*args, **kwargs)

    def get_prep_value(self, value: str) -> None | str:
        value = super().get_prep_value(value)
        return value if value is None else value.lower()


class Role(models.Model):
    """Role model for supporting multiple roles per user."""

    CUSTOMER = "customer"
    STAFF = "staff"
    CREATOR = "creator"
    LOCAL_PROVIDER = "localprovider"
    VERIFIED_SHAMAN = "verifiedshaman"
    MASTER = "master"

    ROLE_CHOICES = (
        (CUSTOMER, "Customer"),
        (STAFF, "Staff"),
        (CREATOR, "Creator"),
        (LOCAL_PROVIDER, "Local Provider"),
        (VERIFIED_SHAMAN, "Verified Shaman"),
        (MASTER, "Master"),
    )

    name = models.CharField(max_length=20, choices=ROLE_CHOICES, unique=True)

    class Meta:
        db_table = "users_role"

    def __str__(self) -> str:
        return self.name


class CyberValleyUser(AbstractBaseUser):
    CUSTOMER = "customer"
    STAFF = "staff"
    CREATOR = "creator"
    LOCAL_PROVIDER = "localprovider"
    VERIFIED_SHAMAN = "verifiedshaman"
    MASTER = "master"

    ROLE_CHOICES = (
        (CUSTOMER, "Customer"),
        (STAFF, "Staff"),
        (CREATOR, "Creator"),
        (LOCAL_PROVIDER, "Local Provider"),
        (VERIFIED_SHAMAN, "Verified Shaman"),
        (MASTER, "Master"),
    )

    address = AddressField(primary_key=True)
    role = models.CharField(max_length=20, choices=ROLE_CHOICES, default=CUSTOMER)
    roles = models.ManyToManyField(Role, blank=True, related_name="users")
    profile_manager_bps = models.PositiveSmallIntegerField(default=0)
    # Basis points for profile manager share
    # Applied after fixed bps, before profile recipients
    # XXX: This field is requred because of bug in simplejwt
    is_active = models.BooleanField(default=True)

    REQUIRED_FIELDS: ClassVar[list[str]] = []
    USERNAME_FIELD = "address"

    def has_role(self, *role_names: str) -> bool:
        """Check if user has any of the given roles."""
        if not role_names:
            return False
        return self.roles.filter(name__in=role_names).exists()

    @property
    def is_staff(self) -> bool:
        return self.has_role(self.STAFF, self.LOCAL_PROVIDER, self.MASTER)

    @property
    def is_creator(self) -> bool:
        return self.has_role(
            self.CREATOR,
            self.VERIFIED_SHAMAN,
            self.LOCAL_PROVIDER,
            self.MASTER,
        )

    @property
    def is_local_provider(self) -> bool:
        return self.has_role(self.LOCAL_PROVIDER)

    @property
    def is_verified_shaman(self) -> bool:
        return self.has_role(self.VERIFIED_SHAMAN)

    @property
    def is_master(self) -> bool:
        return self.has_role(self.MASTER)


class UserSocials(models.Model):
    class Network(models.TextChoices):
        TELEGRAM = "telegram"
        INSTAGRAM = "instagram"
        DISCORD = "discord"
        WHATSAPP = "whatsapp"

    user = models.ForeignKey(
        CyberValleyUser, on_delete=models.CASCADE, related_name="socials"
    )
    network = models.CharField(choices=Network)
    value = models.CharField()
    metadata = models.JSONField(null=True, blank=True, default=dict)
    created_at = models.DateTimeField(auto_now_add=True)
    updated_at = models.DateTimeField(auto_now=True)

    class Meta:
        unique_together = ("user", "network", "value")

    def __str__(self) -> str:
        return f"{self.user.address} - {self.network} - {self.value}"