from django.shortcuts import get_object_or_404
from rest_framework.decorators import api_view, permission_classes
from rest_framework.generics import get_object_or_404
from rest_framework.permissions import IsAuthenticated
from rest_framework.response import Response

from vbv_lernwelt.course.models import Course, CourseSessionUser
from vbv_lernwelt.learning_mentor.models import LearningMentor


@api_view(["GET"])
@permission_classes([IsAuthenticated])
def get_cockpit_type(request, course_id: int):
    course = get_object_or_404(Course, id=course_id)

    cockpit_type = None

    if LearningMentor.objects.filter(mentor=request.user, course=course).exists():
        cockpit_type = "mentor"
    elif CourseSessionUser.objects.filter(
        user=request.user,
        course_session__course=course,
        role=CourseSessionUser.Role.EXPERT,
    ).exists():
        cockpit_type = "expert"

    return Response({"type": cockpit_type})