feat: Invite endpoint

nexus/core/routes/auth.py

@@ -1,9 +1,13 @@
+from typing import Annotated, Union
+
+from fastapi import Header
 from fastapi.requests import Request
 from fastapi.responses import JSONResponse, RedirectResponse
 from fastapi.routing import APIRouter
 
 from nexus.core.oauth import OAuth2Session
 from nexus.core.oauth.decorators import requireValidAuth
+from nexus.utils.discord_utils import generateInviteLink
 
 
 router = APIRouter()
@@ -27,3 +31,13 @@ async def logout(request: Request):
     if request.cookies.get("loggedIn"):
         resp.delete_cookie("loggedIn")
     return resp
+
+
+@router.get("/invite")
+async def invite(request: Request):
+    inviteLink = generateInviteLink(request)
+    resp = RedirectResponse(inviteLink)
+    contentType: Union[str, None] = request.headers.get("content-type")
+    if contentType == "application/json":
+        resp = JSONResponse({"invite": inviteLink})
+    return resp

nexus/core/routes/ng/meta.py

@@ -15,6 +15,7 @@
 from nexus.core.oauth import Guild, User
 from nexus.core.oauth.decorators import requireValidAuth
 from nexus.core.oauth.utils import validateAuth
+from nexus.utils.discord_utils import generateInviteLink
 
 
 if TYPE_CHECKING:
@@ -123,12 +124,7 @@ async def managedGuilds(request: Request):
         guildJson = guild.json()
         # for some reason 'guild.id in botGuilds' always returns False
         guildJson["bot"] = int(guild.id) in [int(i) for i in botGuilds]
-        guildJson["invite"] = discord.utils.oauth_url(
-            request.app.clientId,
-            permissions=discord.Permissions(4260883702),
-            guild=guild,
-            # redirect_uri=os.getenv("DISCORD_GUILD_REDIRECT_URI"),
-        )
+        guildJson["invite"] = generateInviteLink(request, guild)
         ret.append(guildJson)
 
     # show guilds that has the bot in it first, while also sort them by name

nexus/utils/discord_utils.py

@@ -0,0 +1,15 @@
+from typing import Optional
+import discord
+from discord.abc import MISSING
+from starlette.requests import Request
+
+from nexus.core.oauth.models import Guild
+
+
+def generateInviteLink(request: Request, guild: Optional[Guild] = None):
+    return discord.utils.oauth_url(
+        request.app.clientId,
+        permissions=discord.Permissions(4260883702),
+        guild=guild or MISSING,
+        # redirect_uri=os.getenv("DISCORD_GUILD_REDIRECT_URI"),
+    )