Update Regex to include Discord protocol format (#304)

Chris is traveling abroad and is unable to access. Therefore, Fisher
have become his proxy. An 'apprentice' if you will. Shame because Fisher
have years more software engineering experience compared to him.

Unfortunately, neither Fisher nor Chris cared to separate the version
bump into a separate commit so SENIOR Engineer Joe Banks to JUNIOR
Engineer Chris Lovering with INTERN Fisher has had to Get Involved.

Co-authored-by: ChrisLovering <[email protected]>
Co-authored-by: fisher60 <[email protected]>

Author: 3 people

pydis_core/utils/regex.py

@@ -3,14 +3,16 @@
 import re
 
 DISCORD_INVITE = re.compile(
-    r"(https?:\/\/)?(www\.)?"                    # Optional http(s) and www.
+    r"(https?:\/\/)?(discord:\/*)?"              # Optional protocols
+    r"(www\.)?"                                  # Optional www
+    r"[@#]*"                                     # Optional @ or # symbols
     r"(\B|discord(app)?)"                        # Optional discord(app)
     r"([.,]|dot)"                                # Various characters to cover dots
     r"("
         r"(gg|me)"                               # TLDs that embed within discord
         r"|com(\/|slash|\\)invite"               # Only match com/invite
     r")"
-    r"(/|slash|\\+)"                              # / or 'slash' or 1+ of \
+    r"(/|slash|\\+)"                             # / or 'slash' or 1+ of \
     r"(?P<invite>\S+)",                          # the invite code itself
     flags=re.IGNORECASE
 )

tests/pydis_core/utils/test_regex.py

@@ -49,6 +49,11 @@ def test_discord_invite_positives(self):
         self.assertEqual(search_regex("https://discord.gg/python with whitespace"), "python")
         self.assertEqual(search_regex(" https://discord.gg/python "), "python")
 
+        self.assertEqual(search_regex("discord:#@discordapp.com/invite/python"), "python")
+        self.assertEqual(search_regex("discord:/#@discordapp.com/invite/python"), "python")
+        self.assertEqual(search_regex("discord://#@discordapp.com/invite/python"), "python")
+        self.assertEqual(search_regex("discord://@#discordapp.com/invite/python"), "python")
+
     def test_discord_invite_negatives(self):
         """Test the DISCORD_INVITE regex on a set of strings we would expect to not capture."""