from datetime import datetime

from fastapi import APIRouter, Depends
from pydantic import BaseModel
from typing import Optional, List

from backend.services.permission.role_service import RoleService
from common.token.token_manager import TokenManager

router = APIRouter()
token_manager = TokenManager()
role_service = RoleService()


class UpdateRoleRequest(BaseModel):
    role_id: str
    role_key: str
    role_name: str
    role_description: Optional[str] = None
    role_level: int
    custom_role_id: Optional[str] = None


class RoleResponse(BaseModel):
    id: str
    role_key: str
    role_name: str
    role_description: Optional[str] = None
    permission_ids: List[str]
    role_level: int
    created_at: datetime
    updated_at: datetime


@router.post(
    "/update",
    response_model=RoleResponse,
    operation_id="update-role",
    summary="Update Role",
    description="Update an existing role by id. Only Admin role allowed."
)
async def update_role(
        req: UpdateRoleRequest,
        #_: bool = Depends(token_manager.has_all_permissions([DefaultPermissionEnum.INVITE_COLLABORATOR.value.permission_key]))
) -> RoleResponse:
    doc = await role_service.update_role(req.role_id, req.role_key, req.role_name, req.role_description, req.role_level, req.custom_role_id)
    return RoleResponse(**doc.dict())