Expand mastodon support

Author: egeakman

src/models/europython.py

@@ -93,16 +93,26 @@ def extract_twitter_url(text: str) -> str:
     @staticmethod
     def extract_mastodon_url(text: str) -> str:
         """
-        Extract the Mastodon URL from the answer, handle @username@instance format
+        Normalize Mastodon handle or URL to the format: https://<instance>/@<username>
         """
-        if not text.startswith(("https://", "http://")) and text.count("@") == 2:
-            mastodon_url = f"https://{text.split('@')[2]}/@{text.split('@')[1]}"
-        else:
-            mastodon_url = (
-                f"https://{text.removeprefix('https://').removeprefix('http://')}"
-            )
+        text = text.strip().split("?", 1)[0]
+
+        # Handle @username@instance or username@instance formats
+        if "@" in text and not text.startswith("http"):
+            parts = text.split("@")
+            if len(parts) == 3:  # @username@instance
+                _, username, instance = parts
+            elif len(parts) == 2:  # username@instance
+                username, instance = parts
+            else:
+                raise ValueError("Invalid Mastodon handle format")
+            return f"https://{instance}/@{username}"
+
+        # Handle full URLs
+        if text.startswith("http://"):
+            text = "https://" + text[len("http://") :]
 
-        return mastodon_url.split("?")[0]
+        return text
 
     @staticmethod
     def extract_linkedin_url(text: str) -> str:
@@ -126,7 +136,7 @@ def extract_bluesky_url(text: str) -> str:
         Returns a normalized BlueSky URL in the form https://bsky.app/profile/<USERNAME>.bsky.social,
         or uses the entire domain if it's custom (e.g., .dev).
         """
-        text = text.split("?", 1)[0].strip()
+        text = text.strip().split("?", 1)[0]
 
         if text.startswith("https://"):
             text = text[8:]

tests/test_social_media_extractions.py

@@ -13,6 +13,7 @@
             "https://mastodon.social/@username",
         ),
         ("@[email protected]", "https://mastodon.social/@username"),
+        ("[email protected]", "https://mastodon.social/@username"),
     ],
 )
 def test_extract_mastodon_url(input_string: str, result: str) -> None: