feat(permissions): centralize workspace role capability checks
This commit is contained in:
@@ -1,9 +1,15 @@
|
||||
from rest_framework import permissions
|
||||
|
||||
from apps.workspaces.models import Workspace, WorkspaceMembership
|
||||
from rest_framework import permissions
|
||||
|
||||
from apps.workspaces.models import Workspace, WorkspaceMembership
|
||||
from apps.workspaces.services import (
|
||||
WORKSPACE_EDIT,
|
||||
WORKSPACE_MEMBERS_CHANGE_ROLE,
|
||||
WORKSPACE_VIEW,
|
||||
has_workspace_capability,
|
||||
)
|
||||
|
||||
|
||||
class IsWorkspaceOwner(permissions.BasePermission):
|
||||
class IsWorkspaceOwner(permissions.BasePermission):
|
||||
"""
|
||||
Permission check:
|
||||
- User must be the explicit 'owner' on the Workspace model.
|
||||
@@ -11,98 +17,86 @@ class IsWorkspaceOwner(permissions.BasePermission):
|
||||
"""
|
||||
message = "Access denied. You must be the Workspace Owner to perform this action."
|
||||
|
||||
def has_object_permission(self, request, view, obj):
|
||||
if not request.user or not request.user.is_authenticated:
|
||||
return False
|
||||
|
||||
if isinstance(obj, Workspace):
|
||||
workspace = obj
|
||||
elif isinstance(obj, WorkspaceMembership):
|
||||
workspace = obj.workspace
|
||||
elif hasattr(obj, 'workspace'):
|
||||
workspace = obj.workspace
|
||||
else:
|
||||
return False
|
||||
|
||||
if workspace.owner == request.user:
|
||||
return True
|
||||
|
||||
return WorkspaceMembership.objects.filter(
|
||||
workspace=workspace,
|
||||
user=request.user,
|
||||
role=WorkspaceMembership.Role.OWNER,
|
||||
is_active=True
|
||||
).exists()
|
||||
|
||||
|
||||
class IsWorkspaceAdmin(permissions.BasePermission):
|
||||
def has_object_permission(self, request, view, obj):
|
||||
if not request.user or not request.user.is_authenticated:
|
||||
return False
|
||||
|
||||
if isinstance(obj, Workspace):
|
||||
workspace = obj
|
||||
elif isinstance(obj, WorkspaceMembership):
|
||||
workspace = obj.workspace
|
||||
elif hasattr(obj, "workspace"):
|
||||
workspace = obj.workspace
|
||||
else:
|
||||
return False
|
||||
|
||||
return workspace.owner_id == request.user.id
|
||||
|
||||
|
||||
class IsWorkspaceAdmin(permissions.BasePermission):
|
||||
"""
|
||||
Permission check:
|
||||
- User's role in the workspace is either 'ADMIN' or 'OWNER'.
|
||||
"""
|
||||
message = "Access denied. You must be a Workspace Admin or Owner to perform this action."
|
||||
|
||||
def has_object_permission(self, request, view, obj):
|
||||
if not request.user or not request.user.is_authenticated:
|
||||
return False
|
||||
|
||||
if isinstance(obj, Workspace):
|
||||
workspace = obj
|
||||
elif isinstance(obj, WorkspaceMembership):
|
||||
workspace = obj.workspace
|
||||
elif hasattr(obj, 'workspace'):
|
||||
workspace = obj.workspace
|
||||
else:
|
||||
return False
|
||||
|
||||
if workspace.owner == request.user:
|
||||
return True
|
||||
|
||||
allowed_roles = [
|
||||
WorkspaceMembership.Role.OWNER,
|
||||
WorkspaceMembership.Role.ADMIN,
|
||||
]
|
||||
|
||||
return WorkspaceMembership.objects.filter(
|
||||
workspace=workspace,
|
||||
user=request.user,
|
||||
role__in=allowed_roles,
|
||||
is_active=True
|
||||
).exists()
|
||||
|
||||
|
||||
class IsWorkspaceMember(permissions.BasePermission):
|
||||
def has_object_permission(self, request, view, obj):
|
||||
if not request.user or not request.user.is_authenticated:
|
||||
return False
|
||||
|
||||
if isinstance(obj, Workspace):
|
||||
workspace = obj
|
||||
elif isinstance(obj, WorkspaceMembership):
|
||||
workspace = obj.workspace
|
||||
elif hasattr(obj, "workspace"):
|
||||
workspace = obj.workspace
|
||||
else:
|
||||
return False
|
||||
|
||||
return has_workspace_capability(request.user, workspace, WORKSPACE_EDIT)
|
||||
|
||||
|
||||
class IsWorkspaceMember(permissions.BasePermission):
|
||||
"""
|
||||
Permission check:
|
||||
- User's role in the workspace is 'OWNER', 'ADMIN', or 'MEMBER'.
|
||||
"""
|
||||
message = "Access denied. You must be an active member of this workspace."
|
||||
|
||||
def has_object_permission(self, request, view, obj):
|
||||
if not request.user or not request.user.is_authenticated:
|
||||
return False
|
||||
|
||||
if isinstance(obj, Workspace):
|
||||
workspace = obj
|
||||
elif isinstance(obj, WorkspaceMembership):
|
||||
workspace = obj.workspace
|
||||
elif hasattr(obj, 'workspace'):
|
||||
workspace = obj.workspace
|
||||
else:
|
||||
return False
|
||||
|
||||
if workspace.owner == request.user:
|
||||
return True
|
||||
|
||||
allowed_roles = [
|
||||
WorkspaceMembership.Role.OWNER,
|
||||
WorkspaceMembership.Role.ADMIN,
|
||||
WorkspaceMembership.Role.MEMBER,
|
||||
]
|
||||
|
||||
return WorkspaceMembership.objects.filter(
|
||||
workspace=workspace,
|
||||
user=request.user,
|
||||
role__in=allowed_roles,
|
||||
is_active=True
|
||||
).exists()
|
||||
def has_object_permission(self, request, view, obj):
|
||||
if not request.user or not request.user.is_authenticated:
|
||||
return False
|
||||
|
||||
if isinstance(obj, Workspace):
|
||||
workspace = obj
|
||||
elif isinstance(obj, WorkspaceMembership):
|
||||
workspace = obj.workspace
|
||||
elif hasattr(obj, "workspace"):
|
||||
workspace = obj.workspace
|
||||
else:
|
||||
return False
|
||||
|
||||
return has_workspace_capability(request.user, workspace, WORKSPACE_VIEW)
|
||||
|
||||
|
||||
class CanWorkspaceManageMembers(permissions.BasePermission):
|
||||
message = "Access denied. You do not have permission to manage workspace members."
|
||||
|
||||
def has_object_permission(self, request, view, obj):
|
||||
if not request.user or not request.user.is_authenticated:
|
||||
return False
|
||||
|
||||
if isinstance(obj, Workspace):
|
||||
workspace = obj
|
||||
elif isinstance(obj, WorkspaceMembership):
|
||||
workspace = obj.workspace
|
||||
elif hasattr(obj, "workspace"):
|
||||
workspace = obj.workspace
|
||||
else:
|
||||
return False
|
||||
|
||||
return has_workspace_capability(
|
||||
request.user,
|
||||
workspace,
|
||||
WORKSPACE_MEMBERS_CHANGE_ROLE,
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user