# Copyright (C) 2020 by Animath
# SPDX-License-Identifier: GPL-3.0-or-later

import re

from bootstrap_datepicker_plus import DateTimePickerInput
from django import forms
from django.core.exceptions import ObjectDoesNotExist, ValidationError
from django.db.models import Q
from django.utils.translation import gettext_lazy as _

from .models import Participation, Phase, Question, Team, Video


class TeamForm(forms.ModelForm):
    """
    Form to create a team, with the name and the trigram,
    and if the team accepts that Animath diffuse the videos.
    """
    def clean_trigram(self):
        trigram = self.cleaned_data["trigram"].upper()
        if not re.match("[A-Z]{3}", trigram):
            raise ValidationError(_("The trigram must be composed of three uppercase letters."))
        return trigram

    class Meta:
        model = Team
        fields = ('name', 'trigram', 'grant_animath_access_videos',)


class JoinTeamForm(forms.ModelForm):
    """
    Form to join a team by the access code.
    """
    def clean_access_code(self):
        access_code = self.cleaned_data["access_code"]
        if not Team.objects.filter(access_code=access_code).exists():
            raise ValidationError(_("No team was found with this access code."))
        return access_code

    def clean(self):
        cleaned_data = super().clean()
        if "access_code" in cleaned_data:
            team = Team.objects.get(access_code=cleaned_data["access_code"])
            self.instance = team
        return cleaned_data

    class Meta:
        model = Team
        fields = ('access_code',)


class ParticipationForm(forms.ModelForm):
    """
    Form to update the problem of a team participation.
    """
    class Meta:
        model = Participation
        fields = ('problem',)


class RequestValidationForm(forms.Form):
    """
    Form to ask about validation.
    """
    _form_type = forms.CharField(
        initial="RequestValidationForm",
        widget=forms.HiddenInput(),
    )

    engagement = forms.BooleanField(
        label=_("I engage myself to participate to the whole \"Correspondances\"."),
        required=True,
    )


class ValidateParticipationForm(forms.Form):
    """
    Form to let administrators to accept or refuse a team.
    """
    _form_type = forms.CharField(
        initial="ValidateParticipationForm",
        widget=forms.HiddenInput(),
    )

    message = forms.CharField(
        label=_("Message to address to the team:"),
        widget=forms.Textarea(),
    )


class UploadVideoForm(forms.ModelForm):
    """
    Form to upload a video, for a solution or a synthesis.
    """
    class Meta:
        model = Video
        fields = ('link',)

    def clean(self):
        if Phase.current_phase().phase_number != 1 and Phase.current_phase().phase_number != 4 and self.instance.link:
            self.add_error("link", _("You can't upload your video after the deadline."))
        return super().clean()


class ReceiveParticipationForm(forms.ModelForm):
    """
    Update the received participation of a participation.
    """
    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
        self.fields["received_participation"].queryset = Participation.objects.filter(
            ~Q(pk=self.instance.pk) & Q(problem=self.instance.problem, valid=True)
        )

    class Meta:
        model = Participation
        fields = ('received_participation',)


class SendParticipationForm(forms.ModelForm):
    """
    Update the sent participation of a participation.
    """
    sent_participation = forms.ModelChoiceField(
        queryset=Participation.objects,
        label=lambda: _("Send to team"),
    )

    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
        try:
            self.fields["sent_participation"].initial = self.instance.sent_participation
        except ObjectDoesNotExist:     # No sent participation
            pass
        self.fields["sent_participation"].queryset = Participation.objects.filter(
            ~Q(pk=self.instance.pk) & Q(problem=self.instance.problem, valid=True)
        )

    def clean(self, commit=True):
        cleaned_data = super().clean()
        if "sent_participation" in cleaned_data:
            participation = cleaned_data["sent_participation"]
            participation.received_participation = self.instance
            self.instance = participation
        return cleaned_data

    class Meta:
        model = Participation
        fields = ('sent_participation',)


class QuestionForm(forms.ModelForm):
    """
    Create or update a question.
    """
    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
        self.fields["question"].widget.attrs.update({"placeholder": _("How did you get the idea to ...?")})

    def clean(self):
        if Phase.current_phase().phase_number != 2:
            self.add_error(None, _("You can only create or update a question during the second phase."))
        return super().clean()

    class Meta:
        model = Question
        fields = ('question',)


class PhaseForm(forms.ModelForm):
    """
    Form to update the calendar of a phase.
    """
    class Meta:
        model = Phase
        fields = ('start', 'end',)
        widgets = {
            'start': DateTimePickerInput(format='%d/%m/%Y %H:%M'),
            'end': DateTimePickerInput(format='%d/%m/%Y %H:%M'),
        }

    def clean(self):
        # Ensure that dates are in a right order
        cleaned_data = super().clean()
        start = cleaned_data["start"]
        end = cleaned_data["end"]
        if end <= start:
            self.add_error("end", _("Start date must be before the end date."))
        if Phase.objects.filter(phase_number__lt=self.instance.phase_number, end__gt=start).exists():
            self.add_error("start", _("This phase must start after the previous phases."))
        if Phase.objects.filter(phase_number__gt=self.instance.phase_number, start__lt=end).exists():
            self.add_error("end", _("This phase must end after the next phases."))
        return cleaned_data