from django import forms
from django.core.exceptions import ValidationError
from django.utils.translation import gettext_lazy as _
from dcim.models import Site
from netbox.forms import NetBoxModelFilterSetForm, NetBoxModelForm
from tenancy.models import Tenant
from utilities.forms.fields import (
    DynamicModelChoiceField,
    DynamicModelMultipleChoiceField,
    SlugField,
)
from utilities.forms.rendering import FieldSet
from virtualization.models import Cluster

from .choices import EndpointAuthMethodChoices, EndpointProviderChoices, SyncStatusChoices
from .models import VCenterEndpoint


class VCenterEndpointForm(NetBoxModelForm):
    slug = SlugField()
    tenant = DynamicModelChoiceField(
        queryset=Tenant.objects.all(),
        required=False,
    )
    site = DynamicModelChoiceField(
        queryset=Site.objects.all(),
        required=False,
    )
    cluster = DynamicModelChoiceField(
        queryset=Cluster.objects.all(),
    )
    password = forms.CharField(
        label=_("Password"),
        required=False,
        widget=forms.PasswordInput(render_value=False),
        help_text=_("Leave blank to keep the currently stored password."),
    )

    fieldsets = (
        FieldSet("name", "slug", "enabled", "comments", name=_("Endpoint")),
        FieldSet(
            "host",
            "port",
            "username",
            "password",
            "validate_ssl",
            name=_("Connection"),
        ),
        FieldSet("tenant", "site", "cluster", name=_("NetBox target")),
        FieldSet(
            "include_name_regex",
            "exclude_name_regex",
            "sync_powered_off",
            "sync_interfaces",
            "sync_ip_addresses",
            "sync_primary_ips",
            "update_existing",
            name=_("Sync behavior"),
        ),
        FieldSet("default_ipv4_prefix_length", "default_ipv6_prefix_length", "sync_interval_minutes", name=_("Automation")),
        FieldSet("tags", name=_("Tags")),
    )

    class Meta:
        model = VCenterEndpoint
        fields = (
            "name",
            "slug",
            "enabled",
            "host",
            "port",
            "username",
            "password",
            "validate_ssl",
            "tenant",
            "site",
            "cluster",
            "include_name_regex",
            "exclude_name_regex",
            "sync_powered_off",
            "sync_interfaces",
            "sync_ip_addresses",
            "sync_primary_ips",
            "update_existing",
            "default_ipv4_prefix_length",
            "default_ipv6_prefix_length",
            "sync_interval_minutes",
            "comments",
            "tags",
        )

    def clean(self):
        super().clean()
        password = self.cleaned_data.get("password")

        if not self.instance.pk and not password:
            raise ValidationError({"password": _("A password is required for new endpoints.")})

        return self.cleaned_data

    def save(self, commit=True):
        obj = super().save(commit=False)
        password = self.cleaned_data.get("password")

        obj.provider = EndpointProviderChoices.PROVIDER_VMWARE
        obj.auth_method = EndpointAuthMethodChoices.METHOD_PASSWORD
        obj.token_name = ""

        if password:
            obj.set_password(password)

        if commit:
            obj.save()
            self.save_m2m()

        return obj


class VCenterEndpointFilterForm(NetBoxModelFilterSetForm):
    model = VCenterEndpoint

    enabled = forms.NullBooleanField(
        required=False,
        label=_("Enabled"),
    )

    tenant_id = DynamicModelMultipleChoiceField(
        queryset=Tenant.objects.all(),
        required=False,
        label=_("Tenant"),
    )
    site_id = DynamicModelMultipleChoiceField(
        queryset=Site.objects.all(),
        required=False,
        label=_("Site"),
    )
    cluster_id = DynamicModelMultipleChoiceField(
        queryset=Cluster.objects.all(),
        required=False,
        label=_("Cluster"),
    )
    last_status = forms.MultipleChoiceField(
        choices=SyncStatusChoices,
        required=False,
        label=_("Last status"),
    )
    provider = forms.MultipleChoiceField(
        choices=EndpointProviderChoices,
        required=False,
        label=_("Provider"),
    )

    fieldsets = (
        FieldSet("q", "enabled", "provider", "tenant_id", "site_id", "cluster_id", "last_status", name=_("Endpoint")),
    )