# This file is part of photo21
# Copyright (C) 2022  Amicale des élèves de l'ENS Paris-Saclay
# SPDX-License-Identifier: GPL-3.0-or-later

from allauth.account.models import EmailAddress
from allauth.socialaccount.signals import pre_social_login
from django.conf import settings
from django.contrib.auth.models import Group
from django.dispatch import receiver


@receiver(pre_social_login)
def sync_user_fields(sender, request, sociallogin, **kwargs):
    if not sociallogin.is_existing:
        return

    user = sociallogin.user
    data = sociallogin.account.extra_data
    changed = False

    email = data.get("email")
    if email and user.email != email:
        user.email = email
        changed = True
        EmailAddress.objects.filter(user=user).update(email=email)

    username = data.get("preferred_username")
    if username and user.username != username:
        user.username = username
        changed = True

    display_name = data.get("name") or data.get("preferred_username")
    if display_name and user.first_name != display_name:
        user.first_name = display_name
        changed = True

    staff_groups = settings.OAUTH_STAFF_GROUPS
    if staff_groups:
        oauth_groups = set(data.get("groups", []))
        is_staff = bool(oauth_groups & set(staff_groups))
        if user.is_staff != is_staff:
            user.is_staff = is_staff
            changed = True

    if changed:
        user.save()

    group_map = settings.OAUTH_GROUP_MAP
    if group_map:
        oauth_groups = set(data.get("groups", []))
        for oauth_group, django_group_name in group_map.items():
            django_group, _ = Group.objects.get_or_create(name=django_group_name)
            if oauth_group in oauth_groups:
                user.groups.add(django_group)
            else:
                user.groups.remove(django_group)