from apps.workspace.models import Workspace as WorkspaceModel
from apps.cuser.models import CuserRole as CuserRoleModel
from apps.customer.models import CustomerResponsible as CustomerResponsibleModel, Customer as CustomerModel


def get_user_workspace(request) -> WorkspaceModel:
    jwt_payload = request.auth.payload
    cuser_role_uuid = jwt_payload.get('cuser_role_uuid')
    # use_redis
    cuser_role = CuserRoleModel.objects.get(uuid=cuser_role_uuid)

    return cuser_role.sys_ws
    '''role_type_name = cuser_role.role_type.role_name_en

    if role_type_name == 'workspace_operator':
        operator = CustomerResponsibleModel.objects.get(cuser_role=cuser_role)
        workspace = operator.sys_ws
        return workspace
    else:
        customer = CustomerModel.objects.get(cuser_role=cuser_role)
        return customer.sys_ws'''


def get_user_workspace_uuid(request) -> str:
    jwt_payload = request.auth.payload
    workspace_uuid = jwt_payload.get('workspace_uuid')
    return str(workspace_uuid)