class SlotForm(forms.ModelForm):
    extra_guests = forms.IntegerField(label="Personnes supplémentaires", required=False, initial=0, min_value=0)

    class Meta:
        model = Reservation
        fields = ["start", "end", "extra_guests"]
        widgets = {
            'start': forms.DateTimeInput(attrs={'type': 'datetime-local'}, format='%Y-%m-%dT%H:%M'),
            'end': forms.DateTimeInput(attrs={'type': 'datetime-local'}, format='%Y-%m-%dT%H:%M'),
        }
        help_texts = {"start": "Choisir un horaire entre 14h et 16h",
                      "end": "Choisir un horaire entre 16h et 18h"}

    def __init__(self, *args, **kwargs):
        self.house = kwargs.pop("house", None)
        super().__init__(*args, **kwargs)

    def clean(self):
        cleaned_data = super().clean()
        start = cleaned_data.get("start")
        end = cleaned_data.get("end")
        guests = cleaned_data.get("extra_guests", 0) + settings.MINI_RESERVATION

        day_reservation_exists = Reservation.objects.filter(start__date=start.date(), house=self.house,
                                                            slot_in_day=True).exists()
        late_night_reservation_exists = Reservation.objects.filter(end__date=start.date(), house=self.house,
                                                                   late=True).exists()
        # Début du créneau du form plus grand ou égal que la fin d'un existant
        # Fin du créneau du form plus petit ou égal que le début d'un existant
        long_term_reservation_exists = Reservation.objects.filter(house=self.house, end__gte=start,
                                                                  start__lte=end).exists()

        # Vérifier que start et end sont définis
        if not start or not end:
            raise ValidationError("Les dates de début et de fin doivent être définies.")

        # Vérifier que le créneau est d'au moins 2 heures
        if end - start < timedelta(hours=2):
            raise ValidationError("La réservation doit être d'au moins 2 heures.")

        # Vérifier que start et end sont sur la même journée
        if start.date() != end.date():
            raise ValidationError("La réservation doit commencer et se terminer le même jour.")

        # Vérifier que le créneau est entre 14h et 18h
        if not (start.time() >= time(14, 0) and end.time() <= time(18, 0)):
            raise ValidationError("La réservation doit être entre 14h et 18h.")

        # Vérifier que l'utilisateur ne réserve pas pour trop de personnes
        if guests > self.house.capacity:
            raise ValidationError(
                f"Capacité maximum de {self.house.capacity} personnes, vous essayez de réserver pour {guests} "
                f"personnes.")

        # Vérifier si une réservation existe ce jour
        if day_reservation_exists or late_night_reservation_exists or long_term_reservation_exists:
            raise ValidationError("Une réservation est déjà prévue ce jour")

        return cleaned_data

from datetime import time, datetime

from django import forms
from django.conf import settings
from django.core.exceptions import ValidationError

from spa.models import Reservation


class NightReservationForm(forms.Form):
    start = forms.DateField(widget=forms.DateInput(attrs={'type': 'date'}, format='%Y-%m-%d'), label="Arrivée")
    end = forms.DateField(widget=forms.DateInput(attrs={'type': 'date'}, format='%Y-%m-%d'), label="Départ")
    late = forms.BooleanField(label="Départ tardif",
                              help_text="Départ à 16h au lieu de 12h. Supp 25€/personne",
                              widget=forms.CheckboxInput(attrs={"style": "border: 1px solid #000;"}), required=False)
    # Options
    extra_guests = forms.IntegerField(initial=0, required=False, label="Personnes supplémentaires",
                                      help_text="supp 80€ / personne", min_value=0)
    massage_1h = forms.IntegerField(initial=0, required=False, label="Massage 1h", help_text="supp 50€/personne",
                                    min_value=0)
    massage_30min = forms.IntegerField(initial=0, required=False, label="Massage 30 min", help_text="supp 30€/personne",
                                       min_value=0)
    cocktails = forms.IntegerField(initial=0, required=False, label="Fours et Cocktails (2 personnes)",
                                   help_text="supp 40€ pour 2 personnes", min_value=0)
    breakfast = forms.IntegerField(initial=0, required=False, label="Petit Déjeuner",
                                   help_text="supp 35€ pour 2 personnes", min_value=0)
    decoration = forms.BooleanField(label="Décoration", help_text="supp 50€",
                                    widget=forms.CheckboxInput(attrs={"style": "border: 1px solid #000;"}),
                                    required=False)

    def __init__(self, *args, **kwargs):
        self.house = kwargs.pop("house", None)
        super().__init__(*args, **kwargs)

    def clean(self):
        cleaned_data = super().clean()
        start = cleaned_data.get("start")
        end = cleaned_data.get("end")
        late = cleaned_data.get("late")
        guests = cleaned_data.get("extra_guests", 0) + settings.MINI_RESERVATION

        # Vérifier que start et end sont définis
        if not start or not end:
            raise ValidationError("Les dates de début et de fin doivent être définies.")

        # Vérifier que la date de fin est après la date de début
        if end <= start:
            raise ValidationError("La date de fin doit être postérieure à la date de début.")

        # Vérifier la capacité de l'établissement
        if guests > self.house.capacity:
            raise ValidationError(
                f"Capacité maximum de {self.house.capacity} personnes, vous essayez de réserver pour {guests} "
                f"personnes.")

        # Si départ tardif, mais qu'il y a une réservation en journée le lendemain ce n'est pas possible
        if late:
            if Reservation.objects.filter(start__date=end, house=self.house, slot_in_day=True).exists():
                raise ValidationError("Il n'est pas possible de faire un départ tardif sur cette période.")

        # Convertir les dates de début et de fin en datetime avec les heures spécifiques
        start_dt = datetime.combine(start, time(19, 0))  # Début à 19h00
        end_dt = datetime.combine(end, time(16, 0) if late else time(12, 0))

        # Vérifier s'il n'existe pas déjà une réservation
        if Reservation.objects.filter(house=self.house, start__lt=end_dt, end__gt=start_dt).exists():
            raise ValidationError("Il y a déjà une réservation pour ces dates.")

        return cleaned_data