Add error handling to authentik.lib.utils.email.mask_email

authentik/lib/utils/email.py

@@ -15,26 +15,46 @@ def mask_email(email: str | None) -> str | None:
     if not email:
         return None
 
-    local, domain = email.split("@")
-    domain_parts = domain.split(".")
-    limit = 2
-
-    # Mask local part (keep first char)
-    if len(local) <= limit:
-        masked_local = "*" * len(local)
-    else:
-        masked_local = local[0] + "*" * (len(local) - 1)
-
-    # Mask each domain part except the last one (TLD)
-    masked_domain_parts = []
-    for _i, part in enumerate(domain_parts[:-1]):  # Process all parts except TLD
-        if len(part) <= limit:
-            masked_part = "*" * len(part)
-        else:
-            masked_part = part[0] + "*" * (len(part) - 1)
-        masked_domain_parts.append(masked_part)
 
-    # Add TLD unchanged
-    masked_domain_parts.append(domain_parts[-1])
+    try:
+        # Basic email format validation
+        if email.count("@") != 1:
+            raise ValueError("Invalid email format: Must contain exactly one '@' symbol")
+
+        local, domain = email.split("@")
+        if not local or not domain:
+            raise ValueError("Invalid email format: Local and domain parts cannot be empty")
+
+        domain_parts = domain.split(".")
+        if len(domain_parts) < 2:  # noqa: PLR2004
+            raise ValueError("Invalid email format: Domain must contain at least one dot")
 
-    return f"{masked_local}@{'.'.join(masked_domain_parts)}"
+        limit = 2
+
+        # Mask local part (keep first char)
+        if len(local) <= limit:
+            masked_local = "*" * len(local)
+        else:
+            masked_local = local[0] + "*" * (len(local) - 1)
+
+        # Mask each domain part except the last one (TLD)
+        masked_domain_parts = []
+        for _i, part in enumerate(domain_parts[:-1]):  # Process all parts except TLD
+            if not part:  # Check for empty parts (consecutive dots)
+                raise ValueError("Invalid email format: Domain parts cannot be empty")
+            if len(part) <= limit:
+                masked_part = "*" * len(part)
+            else:
+                masked_part = part[0] + "*" * (len(part) - 1)
+            masked_domain_parts.append(masked_part)
+
+        # Add TLD unchanged
+        if not domain_parts[-1]:  # Check if TLD is empty
+            raise ValueError("Invalid email format: TLD cannot be empty")
+        masked_domain_parts.append(domain_parts[-1])
+
+        return f"{masked_local}@{'.'.join(masked_domain_parts)}"
+    except ValueError:
+        raise
+    except Exception as e:
+        raise ValueError(f"Invalid email format: {str(e)}") from e