import uuid
from django.db import models
from django.contrib.auth.models import AbstractUser, UserManager, Group, Permission
from django.core.validators import RegexValidator, MinLengthValidator, MaxValueValidator, MinValueValidator

from django.conf import settings
from django.db import models
from django.utils import timezone
from datetime import timedelta

from vontor_cz.models import SoftDeleteModel

from django.contrib.auth.models import UserManager

import logging

logger = logging.getLogger(__name__)

class CustomUserManager(UserManager):
    # Inherit get_by_natural_key and all auth behaviors
    use_in_migrations = True

class ActiveUserManager(CustomUserManager):
    def get_queryset(self):
        return super().get_queryset().filter(is_active=True)

class CustomUser(SoftDeleteModel, AbstractUser):
    groups = models.ManyToManyField(
        Group,
        related_name="customuser_set",  # <- přidáš related_name
        blank=True,
        help_text="The groups this user belongs to.",
        related_query_name="customuser",
    )
    user_permissions = models.ManyToManyField(
        Permission,
        related_name="customuser_set",  # <- přidáš related_name
        blank=True,
        help_text="Specific permissions for this user.",
        related_query_name="customuser",
    )

    class Role(models.TextChoices):
        ADMIN = "admin", "Admin"
        MANAGER = "mod", "Moderator"
        CUSTOMER = "regular", "Regular"

    role = models.CharField(max_length=20, choices=Role.choices, default=Role.CUSTOMER)

    

    phone_number = models.CharField(
        null=True,
        blank=True,

        unique=True,
        max_length=16,
        validators=[RegexValidator(r'^\+?\d{9,15}$', message="Zadejte platné telefonní číslo.")]
    )
    
    email_verified = models.BooleanField(default=False)
    email = models.EmailField(unique=True, db_index=True)

    gdpr = models.BooleanField(default=False)
    is_active = models.BooleanField(default=False)

    create_time = models.DateTimeField(auto_now_add=True)


    city = models.CharField(null=True, blank=True, max_length=100)
    street = models.CharField(null=True, blank=True, max_length=200)

    postal_code = models.CharField(
        blank=True,
        null=True,

        max_length=5,
        validators=[
            RegexValidator(
                regex=r'^\d{5}$',
                message="Postal code must contain exactly 5 digits.",
                code='invalid_postal_code'
            )
        ]
    )

    USERNAME_FIELD = "username"
    REQUIRED_FIELDS = [
        "email"
    ]

    # Ensure default manager has get_by_natural_key
    objects = CustomUserManager()
    # Optional convenience manager for active users only
    active = ActiveUserManager()
    
    def delete(self, *args, **kwargs):
        self.is_active = False

        return super().delete(*args, **kwargs)
    
    def save(self, *args, **kwargs):
        is_new = self._state.adding  # True if object hasn't been saved yet

        # Pre-save flags for new users
        if is_new:
            if self.is_superuser or self.role == "admin":
                # ensure admin flags are consistent
                self.is_active = True
                self.is_staff = True
                self.is_superuser = True
                self.role = "admin"
            else:
                self.is_staff = False

        # First save to obtain a primary key
        super().save(*args, **kwargs)

        # Assign group after we have a PK
        if is_new:
            from django.contrib.auth.models import Group
            group, _ = Group.objects.get_or_create(name=self.role)
            # Use add/set now that PK exists
            self.groups.set([group])

        return super().save(*args, **kwargs)