T3MP3ST public launch build

bench/cve-zero/samples/GHSA-fqvv-jvhr-g5jc/source/src/firefighter/raid/utils.py

@@ -0,0 +1,52 @@
+from __future__ import annotations
+
+import logging
+from functools import cache
+from typing import Any
+
+logger = logging.getLogger(__name__)
+
+
+@cache
+def get_domain_from_email(email: str) -> str:
+    """Returns the domain from an email address.
+
+    Removes any subdomain(s) and the @, applies .lower().
+
+    Args:
+        email: The email address to extract the domain from.
+
+    Returns:
+        The domain part of the email address.
+
+    Raises:
+        ValueError: If the email is not well-formed.
+
+    Examples:
+      - `john.doe@example.com` => `example.com`
+      - `alice.bob@test.example.ORG` => `example.org`
+      - `webmaster@localhost` => `localhost`
+    """
+    # If there is not exactly one @, the email is invalid
+    if email.count("@") != 1:
+        msg = f"Invalid email: {email}"
+        raise ValueError(msg)
+    domain = email.rsplit("@", maxsplit=1)[-1]
+    if not domain:
+        msg = f"Invalid email: {email}"
+        raise ValueError(msg)
+    domain_parts = domain.split(".")
+
+    return (".".join(domain_parts[-2:]) if len(domain_parts) > 2 else domain).lower()
+
+
+def normalize_cache_value(value: Any) -> str:
+    """Normalize cache values for loop-prevention keys."""
+    if value is None:
+        return ""
+    if isinstance(value, str):
+        return value.strip().lower()
+    try:
+        return str(int(value))
+    except (TypeError, ValueError):
+        return str(value).strip().lower()