from django import forms
from django.core import validators
from .models import File, Message, PublicMessage, PublicFile, Friendship
from django.contrib.auth import get_user_model
from django.contrib.auth.forms import UserCreationForm

class CustomUserCreationForm(UserCreationForm):
    class Meta(UserCreationForm.Meta):
        labels = {
            'username': '用户名',
            'password1': '密码',
            'password2': '确认密码'
        }
        help_texts = {
            'username': '150个字符以内。可以包含中文、字母、数字和 @/./+/-/_。',
            'password1': '您的密码不能与其他个人信息太相似。\n您的密码必须包含至少 8 个字符。\n您的密码不能是常用密码。\n您的密码不能全部是数字。',
            'password2': '再次输入密码以确认。'
        }

    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
        # 修改用户名字段的正则验证
        self.fields['username'].validators = [
            validators.RegexValidator(
                r'^[\w\u4e00-\u9fa5@.+-_]+$',
                '用户名只能包含中文、字母、数字和 @/./+/-/_。'
            )
        ]

class FileUploadForm(forms.ModelForm):
    is_public = forms.BooleanField(
        required=False,
        label='设为公开文件',
        help_text='如果勾选，此文件将对所有用户可见'
    )

    class Meta:
        model = File
        fields = ['file', 'description', 'shared_with', 'is_public']
        labels = {
            'file': '文件',
            'description': '描述',
            'shared_with': '共享用户'
        }

    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
        self.fields['shared_with'].required = False

class MessageForm(forms.ModelForm):
    class Meta:
        model = Message
        fields = ['recipients', 'content', 'file']
        labels = {
            'recipients': '收件人',
            'content': '消息内容',
            'file': '附件'
        }

class PublicMessageForm(forms.ModelForm):
    class Meta:
        model = PublicMessage
        fields = ['content']
        labels = {
            'content': '留言内容'
        }
        widgets = {
            'content': forms.Textarea(attrs={'rows': 3, 'placeholder': '您的留言...'}),
        }

class FriendRequestForm(forms.Form):
    username = forms.CharField(
        label='用户名',
        max_length=150,
        widget=forms.TextInput(attrs={
            'placeholder': '输入要添加的好友用户名',
            'class': 'w-full px-3 py-2 border rounded-md'
        })
    )

    def clean_username(self):
        username = self.cleaned_data['username']
        User = get_user_model()
        try:
            user = User.objects.get(username=username)
        except User.DoesNotExist:
            raise forms.ValidationError('用户不存在')
        return user

class FriendActionForm(forms.Form):
    ACTION_CHOICES = [
        ('accept', '接受'),
        ('reject', '拒绝'),
    ]

    action = forms.ChoiceField(
        choices=ACTION_CHOICES,
        widget=forms.RadioSelect,
        label='操作'
    )

class PublicFileForm(forms.ModelForm):
    class Meta:
        model = PublicFile
        fields = ['file', 'description']
        labels = {
            'file': '文件',
            'description': '描述'
        }
        widgets = {
            'description': forms.Textarea(attrs={'rows': 3, 'placeholder': '文件描述...'}),
        }