Add config identity_min_length and apply non-admin validation

1 files changed, 22 insertions, 0 deletions

diff --git a/users/views/identity.py b/users/views/identity.py
index 5524c4c..4dae6d5 100644
--- a/users/views/identity.py
+++ b/users/views/identity.py
@@ -2,6 +2,7 @@ import string
 
 from django import forms
 from django.contrib.auth.decorators import login_required
+from django.core import validators
 from django.http import Http404, JsonResponse
 from django.shortcuts import redirect
 from django.utils.decorators import method_decorator
@@ -144,10 +145,23 @@ class CreateIdentity(FormView):
                 (domain.domain, domain.domain)
                 for domain in Domain.available_for_user(user)
             ]
+            self.user = user
 
         def clean_username(self):
             # Remove any leading @ and force it lowercase
             value = self.cleaned_data["username"].lstrip("@").lower()
+
+            if not self.user.admin:
+                # Apply username min length
+                limit = int(Config.system.identity_min_length)
+                validators.MinLengthValidator(limit)(value)
+
+                # Apply username restrictions
+                if value in Config.system.restricted_usernames.split():
+                    raise forms.ValidationError(
+                        "This username is restricted to administrators only."
+                    )
+
             # Validate it's all ascii characters
             for character in value:
                 if character not in string.ascii_letters + string.digits + "_-":
@@ -167,6 +181,14 @@ class CreateIdentity(FormView):
             ):
                 raise forms.ValidationError(f"{username}@{domain} is already taken")
 
+            if not self.user.admin and (
+                Identity.objects.filter(users=self.user).count()
+                >= Config.system.identity_max_per_user
+            ):
+                raise forms.ValidationError(
+                    f"You are not allowed more than {Config.system.identity_max_per_user} identities"
+                )
+
     def get_form(self):
         form_class = self.get_form_class()
         return form_class(user=self.request.user, **self.get_form_kwargs())