"""Extension et gestion des profils utilisateurs.

Les profils peuvent enregistrer les informations suivantes:

* le type de template,
* la couleur de la barre de navigation
* si la barre de navigation doit être cachée ou non
"""

from django.contrib.contenttypes.models import ContentType
from django.contrib.auth import get_user_model
from django.db import models
from jarvis.people.models import Gymnast

User = get_user_model()


class Profile(models.Model):
    """Classe étendant les informations des utilisateurs/entraineurs de l'application.

    References:
    * https://simpleisbetterthancomplex.com/tutorial/2016/07/22/how-to-extend-django-user-model.html
    """

    TEMPLATE_CHOICES = ((0, "Dark"), (1, "Light"))
    SIDEBAR_CHOICES = (
        (0, "Purple"),
        (1, "Blue"),
        (2, "Green"),
        (3, "Orange"),
        (4, "Red"),
    )

    user = models.OneToOneField(User, on_delete=models.CASCADE)
    template_color = models.PositiveSmallIntegerField(
        choices=TEMPLATE_CHOICES, verbose_name="Template", default=0
    )
    sidebar_color = models.PositiveSmallIntegerField(
        choices=SIDEBAR_CHOICES, verbose_name="Sidebar", default=0
    )
    is_sidebar_minified = models.BooleanField(default=False)

    def __str__(self):
        return f"{self.user.first_name} {self.user.last_name}"


class Notification(models.Model):
    """Classe permettant de définir quelles notification un utilisateur veut recevoir."""

    class Meta:
        unique_together = ("user", "gymnast", "functionality")

    user = models.ForeignKey(User, on_delete=models.CASCADE)
    gymnast = models.ForeignKey(
        Gymnast, on_delete=models.CASCADE, related_name="notifications"
    )
    functionality = models.ForeignKey(
        ContentType,
        on_delete=models.CASCADE,
        limit_choices_to={"app_label": "followup"},
    )

    def __str__(self):
        return f"{self.user} will be notified for add/update {self.functionality} to {self.gymnast}"