from rest_framework import permissions
from dry_rest_permissions.generics import DRYPermissions

from ..rest_framework.viewsets import ServiceGenericViewSet


class ServiceDRYPermissions(DRYPermissions):
    """
    Customização para o DRYPermissions utilizar a nossa camada service,
    e não o model!
    """

    def has_permission(self, request, view: ServiceGenericViewSet):
        """
        Overrides the standard function and figures out methods to call for global permissions.
        """
        if not self.global_permissions:
            return True

        assert view.service is not None, (
                "global_permissions set to true without a service "
                "set on the view for '%s'" % view.__class__.__name__
        )

        service = view.get_service()

        action_method_name = None
        if hasattr(view, 'action'):
            action = self._get_action(view.action)
            action_method_name = "has_{action}_permission".format(action=action)
            # If the specific action permission exists then use it, otherwise use general.
            if hasattr(service, action_method_name):
                return getattr(service, action_method_name)(request=request)

        if request.method in permissions.SAFE_METHODS:
            assert hasattr(service, 'has_read_permission'), \
                self._get_error_message(service, 'has_read_permission', action_method_name)
            return service.has_read_permission(request=request)
        else:
            assert hasattr(service, 'has_write_permission'), \
                self._get_error_message(service, 'has_write_permission', action_method_name)
            return service.has_write_permission(request=request)

    def has_object_permission(self, request, view: ServiceGenericViewSet, obj):
        """
        Overrides the standard function and figures out methods to call for object permissions.
        """
        if not self.object_permissions:
            return True

        service = view.get_service()

        action_method_name = None
        if hasattr(view, 'action'):
            action = self._get_action(view.action)
            action_method_name = "has_object_{action}_permission".format(action=action)
            # If the specific action permission exists then use it, otherwise use general.
            if hasattr(service, action_method_name):
                return getattr(service, action_method_name)(request=request, obj=obj)

        if request.method in permissions.SAFE_METHODS:
            assert hasattr(service, 'has_object_read_permission'), \
                self._get_error_message(service, 'has_object_read_permission', action_method_name)
            return service.has_object_read_permission(request=request, obj=obj)
        else:
            assert hasattr(service, 'has_object_write_permission'), \
                self._get_error_message(service, 'has_object_write_permission', action_method_name)
            return service.has_object_write_permission(request=request, obj=obj)

serializer_class = view.get_serializer_class()

assert serializer_class.Meta.model is not None, (
            "global_permissions set to true without a model "
            "set on the serializer for '%s'" % view.__class__.__name__
        )

model_class = serializer_class.Meta.model

class DRYPermissions(permissions.BasePermission):
    def _get_permission_target(self,view, obj=None):
        if obj:
            return obj

        serializer_class = view.get_serializer_class()

        assert serializer_class.Meta.model is not None, (
                "global_permissions set to true without a model "
                "set on the serializer for '%s'" % view.__class__.__name__
            )

        model_class = serializer_class.Meta.model
        return model_class

target = self._get_permission_target(view, obj)

if hasattr(target, action_method_name):
    return getattr(target, action_method_name)(request=request)

target = self._get_permission_target(view, obj)

if hasattr(target, action_method_name):
    return getattr(target, action_method_name)(request=request, obj=obj)