import uuid
from django.db import models
from django.contrib.auth.models import AbstractBaseUser, BaseUserManager, PermissionsMixin

class UserStatus(models.TextChoices):
    ACTIVE = 'ACTIVE', '正常'
    INACTIVE = 'INACTIVE', '未激活'
    BANNED = 'BANNED', '封禁'

class CustomUserManager(BaseUserManager):
    def create_user(self, username, phone=None, password=None, **extra_fields):
        if not username:
            raise ValueError('The Username field must be set')
        user = self.model(username=username, phone=phone, **extra_fields)
        if password:
            user.set_password(password)
        user.save(using=self._db)
        return user

    def create_superuser(self, username, phone=None, password=None, **extra_fields):
        extra_fields.setdefault('is_staff', True)
        extra_fields.setdefault('is_superuser', True)
        return self.create_user(username, phone, password, **extra_fields)

class User(AbstractBaseUser, PermissionsMixin):
    id = models.UUIDField(primary_key=True, default=uuid.uuid4, editable=False)
    username = models.CharField(max_length=64, unique=True)
    phone = models.CharField(max_length=20, unique=True, null=True, blank=True)
    email = models.EmailField(max_length=128, unique=True, null=True, blank=True)
    nickname = models.CharField(max_length=64, null=True, blank=True)
    avatar_url = models.CharField(max_length=512, null=True, blank=True)
    face_url = models.CharField(max_length=512, null=True, blank=True)
    wx_openid = models.CharField(max_length=128, unique=True, null=True, blank=True)
    wx_unionid = models.CharField(max_length=128, unique=True, null=True, blank=True)
    bio = models.TextField(null=True, blank=True)
    location = models.CharField(max_length=128, null=True, blank=True)
    rating = models.DecimalField(max_digits=3, decimal_places=2, default=5.00)
    completed_tasks = models.IntegerField(default=0)

    status = models.CharField(max_length=16, choices=UserStatus.choices, default=UserStatus.ACTIVE)

    is_deleted = models.BooleanField(default=False)
    is_recommended = models.BooleanField(default=False, help_text='管理员推荐')
    recommend_priority = models.IntegerField(default=0, help_text='推荐优先级, 越大越靠前')
    created_at = models.DateTimeField(auto_now_add=True)
    updated_at = models.DateTimeField(auto_now=True)

    is_active = models.BooleanField(default=True)
    is_staff = models.BooleanField(default=False)

    objects = CustomUserManager()
    
    @property
    def roles(self):
        return list(self.user_roles.values_list('role__code', flat=True))

    USERNAME_FIELD = 'username'

    REQUIRED_FIELDS = ['phone']

    class Meta:
        db_table = 'users'

class Role(models.Model):
    id = models.UUIDField(primary_key=True, default=uuid.uuid4, editable=False)
    code = models.CharField(max_length=64, unique=True)
    name = models.CharField(max_length=64)
    description = models.TextField(null=True, blank=True)
    is_system = models.BooleanField(default=False)
    sort_order = models.IntegerField(default=0)
    created_at = models.DateTimeField(auto_now_add=True)

    class Meta:
        db_table = 'roles'

class PermissionType(models.TextChoices):
    MENU = 'MENU', '菜单'
    BUTTON = 'BUTTON', '按钮'
    API = 'API', '接口'

class Permission(models.Model):
    id = models.UUIDField(primary_key=True, default=uuid.uuid4, editable=False)
    code = models.CharField(max_length=128, unique=True)
    name = models.CharField(max_length=128)
    type = models.CharField(max_length=16, choices=PermissionType.choices)
    path = models.CharField(max_length=256, null=True, blank=True)
    method = models.CharField(max_length=16, null=True, blank=True)
    parent = models.ForeignKey('self', on_delete=models.SET_NULL, null=True, blank=True)
    sort_order = models.IntegerField(default=0)
    icon = models.CharField(max_length=64, null=True, blank=True)
    created_at = models.DateTimeField(auto_now_add=True)

    class Meta:
        db_table = 'permissions'

class UserRole(models.Model):
    id = models.UUIDField(primary_key=True, default=uuid.uuid4, editable=False)
    user = models.ForeignKey(User, on_delete=models.CASCADE, related_name='user_roles')
    role = models.ForeignKey(Role, on_delete=models.CASCADE)
    granted_by = models.ForeignKey(User, on_delete=models.SET_NULL, null=True, related_name='granted_roles')
    created_at = models.DateTimeField(auto_now_add=True)

    class Meta:
        db_table = 'user_roles'
        unique_together = ('user', 'role')

class RolePermission(models.Model):
    id = models.UUIDField(primary_key=True, default=uuid.uuid4, editable=False)
    role = models.ForeignKey(Role, on_delete=models.CASCADE, related_name='role_permissions')
    permission = models.ForeignKey(Permission, on_delete=models.CASCADE)
    created_at = models.DateTimeField(auto_now_add=True)

    class Meta:
        db_table = 'role_permissions'
        unique_together = ('role', 'permission')

class Enterprise(models.Model):
    class EnterpriseStatus(models.TextChoices):
        PENDING = 'PENDING', '待审核'
        VERIFIED = 'VERIFIED', '已认证'
        REJECTED = 'REJECTED', '已拒绝'

    id = models.UUIDField(primary_key=True, default=uuid.uuid4, editable=False)
    user = models.OneToOneField(User, on_delete=models.CASCADE)
    company_name = models.CharField(max_length=256)
    credit_code = models.CharField(max_length=64, unique=True, null=True, blank=True)
    status = models.CharField(max_length=16, choices=EnterpriseStatus.choices, default=EnterpriseStatus.PENDING)

    logo_url = models.CharField(max_length=512, null=True, blank=True)

    business_license = models.CharField(max_length=512, null=True, blank=True)
    contact_name = models.CharField(max_length=64, null=True, blank=True)
    contact_phone = models.CharField(max_length=20, null=True, blank=True)
    landline = models.CharField(max_length=32, null=True, blank=True, help_text='座机号码')
    contact_email = models.CharField(max_length=128, null=True, blank=True)
    address = models.TextField(null=True, blank=True)
    description = models.TextField(null=True, blank=True)
    is_deleted = models.BooleanField(default=False)
    created_at = models.DateTimeField(auto_now_add=True)
    updated_at = models.DateTimeField(auto_now=True)

    class Meta:
        db_table = 'enterprises'

class EnterpriseMember(models.Model):
    class MemberRole(models.TextChoices):
        ADMIN = 'ADMIN', '管理员'
        MEMBER = 'MEMBER', '普通成员'
        GUEST = 'GUEST', '外部观察员'

    id = models.UUIDField(primary_key=True, default=uuid.uuid4, editable=False)
    enterprise = models.ForeignKey(Enterprise, on_delete=models.CASCADE, related_name='members')
    user = models.ForeignKey(User, on_delete=models.CASCADE, related_name='enterprise_memberships')
    role = models.CharField(max_length=16, choices=MemberRole.choices, default=MemberRole.MEMBER)
    joined_at = models.DateTimeField(auto_now_add=True)

    class Meta:
        db_table = 'enterprise_members'
        unique_together = ('enterprise', 'user')