from django.db.models.signals import post_delete, pre_save
from django.dispatch import receiver

from vbv_lernwelt.course.models import CourseSessionUser
from vbv_lernwelt.sso.exceptions import MyVbvKeycloakDeleteError, MyVbvKeycloakPostError
from vbv_lernwelt.sso.models import SsoSyncError
from vbv_lernwelt.sso.role_sync.services import (
    add_roles_to_user,
    remove_roles_from_user,
    update_roles_for_user,
)


@receiver(post_delete, sender=CourseSessionUser, dispatch_uid="delete_sso_roles")
def remove_sso_roles(sender, instance, **kwargs):
    try:
        remove_roles_from_user(
            instance.user, [(instance.course_session.course.slug, instance.role)]
        )
    except MyVbvKeycloakDeleteError as e:
        additional_data = getattr(e, "additional_data", {})
        SsoSyncError.objects.create(
            user=instance.user, action=SsoSyncError.Action.REMOVE, data=additional_data
        )


@receiver(pre_save, sender=CourseSessionUser, dispatch_uid="update_sso_roles")
def update_sso_roles(sender, instance: CourseSessionUser, **kwargs):
    try:
        if instance.created_at is None:
            add_roles_to_user(
                instance.user, [(instance.course_session.course.slug, instance.role)]
            )
        else:
            old_csu = CourseSessionUser.objects.get(pk=instance.pk)
            if old_csu.role != instance.role:
                update_roles_for_user(
                    instance.user,
                    add_course_roles=[
                        (instance.course_session.course.slug, instance.role)
                    ],
                    remove_course_roles=[
                        (instance.course_session.course.slug, old_csu.role)
                    ],
                )
    except MyVbvKeycloakDeleteError as e:
        additional_data = getattr(e, "additional_data", {})
        SsoSyncError.objects.create(
            user=instance.user, action=SsoSyncError.Action.REMOVE, data=additional_data
        )
    except MyVbvKeycloakPostError as e:
        additional_data = getattr(e, "additional_data", {})
        SsoSyncError.objects.create(
            user=instance.user, action=SsoSyncError.Action.ADD, data=additional_data
        )