from django import forms
from .models import Truck, Shipment, Bid, Profile, Country, OTPCode
from django.utils.translation import gettext_lazy as _
from django.contrib.auth.forms import UserCreationForm
from django.contrib.auth.models import User

class UserRegistrationForm(UserCreationForm):
    email = forms.EmailField(required=True, widget=forms.EmailInput(attrs={'class': 'form-control'}))
    confirm_email = forms.EmailField(required=True, widget=forms.EmailInput(attrs={'class': 'form-control'}), label=_("Confirm Email"))
    role = forms.ChoiceField(choices=Profile.ROLE_CHOICES, widget=forms.Select(attrs={'class': 'form-select'}))
    country_code = forms.ChoiceField(choices=[], widget=forms.Select(attrs={'class': 'form-select'}))
    phone_number = forms.CharField(max_length=20, required=True, widget=forms.TextInput(attrs={'class': 'form-control', 'placeholder': '123456789'}))

    class Meta(UserCreationForm.Meta):
        model = User
        fields = UserCreationForm.Meta.fields + ('email',)

    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
        # Dynamic country codes from the database
        countries = Country.objects.all()
        if countries.exists():
            self.fields['country_code'].choices = [(c.code, str(c)) for c in countries]
            default_country = countries.filter(is_default=True).first()
            if default_country:
                self.fields['country_code'].initial = default_country.code
        else:
            # Fallback if no countries are defined yet
            self.fields['country_code'].choices = [('966', 'Saudi Arabia (+966)')]
            self.fields['country_code'].initial = '966'

        for field in self.fields.values():
            if not isinstance(field.widget, (forms.CheckboxInput, forms.Select)):
                field.widget.attrs.update({'class': 'form-control'})

    def clean_email(self):
        email = self.cleaned_data.get('email')
        if User.objects.filter(email=email).exists():
            raise forms.ValidationError(_("This email is already in use."))
        return email

    def clean_phone_number(self):
        phone_number = self.cleaned_data.get('phone_number')
        country_code = self.cleaned_data.get('country_code')
        # We check uniqueness of phone_number in Profile
        if Profile.objects.filter(phone_number=phone_number).exists():
            raise forms.ValidationError(_("This phone number is already in use."))
        return phone_number

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

        if email and confirm_email and email != confirm_email:
            self.add_error('confirm_email', _("The two email fields didn't match."))
        
        return cleaned_data

class OTPVerifyForm(forms.Form):
    otp_code = forms.CharField(max_length=6, widget=forms.TextInput(attrs={'class': 'form-control', 'placeholder': '123456'}))

class TruckForm(forms.ModelForm):
    class Meta:
        model = Truck
        fields = [
            'truck_type', 'truck_type_ar', 'model', 'model_ar', 'year', 'plate_no', 
            'load_capacity', 'load_capacity_ar', 'color', 'color_ar', 'registration_expiry_date',
            'truck_picture', 'registration_front', 'registration_back', 'driver_license'
        ]
        widgets = {
            'truck_type': forms.TextInput(attrs={'class': 'form-control', 'placeholder': _('e.g. Flatbed')}),
            'truck_type_ar': forms.TextInput(attrs={'class': 'form-control', 'placeholder': _('مثلا سطحة')}),
            'model': forms.TextInput(attrs={'class': 'form-control'}),
            'model_ar': forms.TextInput(attrs={'class': 'form-control'}),
            'year': forms.NumberInput(attrs={'class': 'form-control'}),
            'plate_no': forms.TextInput(attrs={'class': 'form-control'}),
            'load_capacity': forms.TextInput(attrs={'class': 'form-control'}),
            'load_capacity_ar': forms.TextInput(attrs={'class': 'form-control'}),
            'color': forms.TextInput(attrs={'class': 'form-control'}),
            'color_ar': forms.TextInput(attrs={'class': 'form-control'}),
            'registration_expiry_date': forms.DateInput(attrs={'class': 'form-control', 'type': 'date'}),
            'truck_picture': forms.FileInput(attrs={'class': 'form-control'}),
            'registration_front': forms.FileInput(attrs={'class': 'form-control'}),
            'registration_back': forms.FileInput(attrs={'class': 'form-control'}),
            'driver_license': forms.FileInput(attrs={'class': 'form-control'}),
        }

class ShipmentForm(forms.ModelForm):
    class Meta:
        model = Shipment
        fields = ['description', 'weight', 'origin', 'destination', 'delivery_date']
        widgets = {
            'description': forms.Textarea(attrs={'class': 'form-control', 'rows': 3}),
            'weight': forms.TextInput(attrs={'class': 'form-control'}),
            'origin': forms.TextInput(attrs={'class': 'form-control'}),
            'destination': forms.TextInput(attrs={'class': 'form-control'}),
            'delivery_date': forms.DateInput(attrs={'class': 'form-control', 'type': 'date'}),
        }

class BidForm(forms.ModelForm):
    class Meta:
        model = Bid
        fields = ['truck', 'amount', 'comments']
        widgets = {
            'truck': forms.Select(attrs={'class': 'form-select'}),
            'amount': forms.NumberInput(attrs={'class': 'form-control', 'step': '0.01'}),
            'comments': forms.Textarea(attrs={'class': 'form-control', 'rows': 3}),
        }
    
    def __init__(self, *args, **kwargs):
        user = kwargs.pop('user', None)
        super().__init__(*args, **kwargs)
        if user:
            # Only allow bidding with approved trucks
            self.fields['truck'].queryset = Truck.objects.filter(owner=user, is_approved=True)
            if not self.fields['truck'].queryset.exists():
                self.fields['truck'].help_text = _("You must have an approved truck to place a bid.")