from django.db.models import Count
from rest_framework import status
from rest_framework.exceptions import ValidationError

from api.models import Role
from api.utils import log_critical_error, get_exception_detail
from api.views.admin.base import AdminApiBase
from api.translation import __


class RoleList(AdminApiBase):
    def post(self, request):
        try:
            self.api_log_skip_outgoing_data = True
            data = self.get_data(request)
            return self.api_response(data={"data": data})
        except ValidationError as e:
            return self.api_response(
                error=e.detail["error"],
                errors=e.detail["errors"],
                code=status.HTTP_400_BAD_REQUEST
            )
        except Exception as e:
            e_details = get_exception_detail(e)
            log_critical_error(
                user_id=None,
                descr=e_details["descr"],
                url=e_details["url"],
                trace=e_details["trace"]
            )
            return self.api_response(
                error=__("message.internal_server_error"),
                code=status.HTTP_500_INTERNAL_SERVER_ERROR
            )

    def get_permission_data(self, role):
        data = []

        for val in role.rolepermission_set.all():
            department_ids = list(
                val.rolepermissiondepartment_set.values_list("department_id", flat=True)
            )

            data.append({
                "permission": val.permission,
                "grant_to_all_department": val.grant_to_all_department,
                "department_ids": department_ids,
            })

        return data

    def get_data(self, request):
        roles = (
            Role.objects
            .prefetch_related(
                "rolepermission_set__rolepermissiondepartment_set"
            )
            .annotate(
                user_count=Count("userrole", distinct=True)
            )
            .order_by("name")
        )

        return_data = []

        for val in roles:
            data = {}
            data["id"] = val.id
            data["name"] = val.name
            data["color"] = val.color
            data["type"] = val.type
            data["is_only_self"] = val.is_only_self
            data["created_at"] = val.created_at
            data["user_count"] = val.user_count
            data["permissions"] = self.get_permission_data(val)

            return_data.append(data)

        return return_data