add web

2026-07-06 13:28:00 -04:00
commit 05211f73ef
441 changed files with 94128 additions and 0 deletions
@@ -0,0 +1,122 @@
+import requests
+from config import Config
+import urllib
+config = Config()
+
+class DiscordUserInfo:
+    UserId : int = None
+    Username : str = None
+    Discriminator : str = None
+    AvatarHash : str = None
+    GlobalName : str = None
+
+    def __init__(self, UserId : int, Username : str, AvatarHash : str, GlobalName : str = None, Discriminator : str = "0000") -> None:
+        self.UserId = UserId
+        self.Username = Username
+        self.Discriminator = Discriminator
+        self.AvatarHash = AvatarHash
+        self.GlobalName = GlobalName
+    
+    def GetAvatarURL(self) -> str:
+        """
+        GetAvatarURL returns the URL to the user's avatar.
+        """
+        if self.AvatarHash is None:
+            return f"https://cdn.discordapp.com/embed/avatars/{str(int(self.Discriminator) % 5)}.png"
+        return f"https://cdn.discordapp.com/avatars/{str(self.UserId)}/{self.AvatarHash}.png"
+    
+    def __repr__(self) -> str:
+        return f"<DiscordUserInfo {self.UserId} {self.Username}#{self.Discriminator}>"
+
+class UnexpectedStatusCode(Exception):
+    pass
+class MissingScope(Exception):
+    pass
+def ExchangeCodeForToken( AccessCode : str ) -> dict:
+    """
+    ExchangeCodeForToken exchanges a Discord OAuth2 access code for a Discord OAuth2 token.
+    """
+    DiscordOAuth2TokenExchangeURL = "https://discord.com/api/oauth2/token"
+    DiscordOAuth2TokenExchangeHeaders = {
+        "Content-Type": "application/x-www-form-urlencoded"
+    }
+    DiscordOAuth2TokenExchangeData = {
+        "client_id": config.DISCORD_CLIENT_ID,
+        "client_secret": config.DISCORD_CLIENT_SECRET,
+        "grant_type": "authorization_code",
+        "code": AccessCode,
+        "redirect_uri": config.DISCORD_REDIRECT_URI,
+        "scope": "identify"
+    }
+    DiscordOAuth2TokenExchangeResponse = requests.post(DiscordOAuth2TokenExchangeURL, headers=DiscordOAuth2TokenExchangeHeaders, data=DiscordOAuth2TokenExchangeData)
+    DiscordOAuth2TokenExchangeResponseJSON = DiscordOAuth2TokenExchangeResponse.json()
+
+    if DiscordOAuth2TokenExchangeResponse.status_code != 200:
+        raise UnexpectedStatusCode(f"Unexpected status code {DiscordOAuth2TokenExchangeResponse.status_code} when exchanging code for token.")
+    # Make sure "identify" is in the scope
+    if "identify" not in DiscordOAuth2TokenExchangeResponseJSON["scope"]:
+        raise MissingScope("Missing scope 'identify' when exchanging code for token.")
+    if "guilds.join" not in DiscordOAuth2TokenExchangeResponseJSON["scope"]:
+        raise MissingScope("Missing scope 'guilds.join' when refreshing token.")
+    return DiscordOAuth2TokenExchangeResponseJSON
+
+def GetUserInfoFromToken( AccessToken : str ) -> DiscordUserInfo:
+    """
+    GetUserInfoFromToken gets a user's Discord user info from their access token.
+    """
+    DiscordOAuth2GetUserInfoURL = "https://discord.com/api/users/@me"
+    DiscordOAuth2GetUserInfoHeaders = {
+        "Authorization": f"Bearer {AccessToken}"
+    }
+    DiscordOAuth2GetUserInfoResponse = requests.get(DiscordOAuth2GetUserInfoURL, headers=DiscordOAuth2GetUserInfoHeaders)
+    DiscordOAuth2GetUserInfoResponseJSON = DiscordOAuth2GetUserInfoResponse.json()
+
+    if DiscordOAuth2GetUserInfoResponse.status_code != 200:
+        raise UnexpectedStatusCode(f"Unexpected status code {DiscordOAuth2GetUserInfoResponse.status_code} when getting user info from token.")
+    return DiscordUserInfo(
+        UserId = DiscordOAuth2GetUserInfoResponseJSON["id"],
+        Username = DiscordOAuth2GetUserInfoResponseJSON["username"],
+        Discriminator = DiscordOAuth2GetUserInfoResponseJSON["discriminator"],
+        AvatarHash = DiscordOAuth2GetUserInfoResponseJSON["avatar"],
+        GlobalName = DiscordOAuth2GetUserInfoResponseJSON["global_name"]
+    )
+
+def GenerateAuthorizationURL( State : str ) -> str:
+    """
+    GenerateAuthorizationURL generates a Discord OAuth2 authorization URL.
+    """
+    DiscordOAuth2AuthorizationURL = config.DISCORD_AUTHORIZATION_BASE_URL
+    DiscordOAuth2AuthorizationURLParams = {
+        "client_id": config.DISCORD_CLIENT_ID,
+        "redirect_uri": config.DISCORD_REDIRECT_URI,
+        "response_type": "code",
+        "scope": "identify guilds.join",
+        "state": State
+    }
+    # Format the URL
+    DiscordOAuth2AuthorizationURL = f"{DiscordOAuth2AuthorizationURL}?{urllib.parse.urlencode(DiscordOAuth2AuthorizationURLParams)}"
+    return DiscordOAuth2AuthorizationURL
+
+def RefreshAccessToken( RefreshToken : str ) -> dict:
+    """
+    RefreshAccessToken refreshes a Discord OAuth2 access token from a refresh token.
+    """
+    DiscordOAuth2TokenRefreshData = {
+        "client_id": config.DISCORD_CLIENT_ID,
+        "client_secret": config.DISCORD_CLIENT_SECRET,
+        "grant_type": "refresh_token",
+        "refresh_token": RefreshToken
+    }
+    DiscordOAuth2TokenRefreshURL = "https://discord.com/api/oauth2/token"
+    DiscordOAuth2TokenRefreshHeaders = {
+        "Content-Type": "application/x-www-form-urlencoded"
+    }
+    DiscordOAuth2TokenRefreshResponse = requests.post(DiscordOAuth2TokenRefreshURL, headers=DiscordOAuth2TokenRefreshHeaders, data=DiscordOAuth2TokenRefreshData)
+    DiscordOAuth2TokenRefreshResponseJSON = DiscordOAuth2TokenRefreshResponse.json()
+    if DiscordOAuth2TokenRefreshResponse.status_code != 200:
+        raise UnexpectedStatusCode(f"Unexpected status code {DiscordOAuth2TokenRefreshResponse.status_code} when refreshing token.")
+    if "identify" not in DiscordOAuth2TokenRefreshResponseJSON["scope"]:
+        raise MissingScope("Missing scope 'identify' when refreshing token.")
+    if "guilds.join" not in DiscordOAuth2TokenRefreshResponseJSON["scope"]:
+        raise MissingScope("Missing scope 'guilds.join' when refreshing token.")
+    return DiscordOAuth2TokenRefreshResponseJSON