from django.contrib import admin
from django.core.exceptions import PermissionDenied
from user_roles.services import RoleService


class RoleBasedAdminMixin:
    """
    Миксин для ModelAdmin с проверкой ролей.

    Использование:
        class MyAdmin(RoleBasedAdminMixin, admin.ModelAdmin):
            required_roles = ['owner', 'manager']
    """
    required_roles = []  # Роли, которые имеют доступ

    def has_module_permission(self, request):
        """Проверка доступа к модулю"""
        if not super().has_module_permission(request):
            return False

        if not self.required_roles:
            return True  # Нет ограничений

        return RoleService.user_has_role(request.user, *self.required_roles)

    def has_view_permission(self, request, obj=None):
        """Проверка доступа на просмотр"""
        if not super().has_view_permission(request, obj):
            return False

        if not self.required_roles:
            return True

        return RoleService.user_has_role(request.user, *self.required_roles)

    def has_add_permission(self, request):
        """Проверка доступа на добавление"""
        if not super().has_add_permission(request):
            return False

        if not self.required_roles:
            return True

        return RoleService.user_has_role(request.user, *self.required_roles)

    def has_change_permission(self, request, obj=None):
        """Проверка доступа на изменение"""
        if not super().has_change_permission(request, obj):
            return False

        if not self.required_roles:
            return True

        return RoleService.user_has_role(request.user, *self.required_roles)

    def has_delete_permission(self, request, obj=None):
        """Проверка доступа на удаление"""
        if not super().has_delete_permission(request, obj):
            return False

        if not self.required_roles:
            return True

        return RoleService.user_has_role(request.user, *self.required_roles)


class OwnerOnlyAdminMixin(RoleBasedAdminMixin):
    """Миксин для админки, доступной только владельцу"""
    required_roles = ['owner']


class ManagerOwnerAdminMixin(RoleBasedAdminMixin):
    """Миксин для админки, доступной менеджеру и владельцу"""
    required_roles = ['owner', 'manager']