nex/college-bot-v1
1
0
Fork 0
mirror of https://github.com/nexy7574/LCC-bot.git synced 2024-09-19 18:16:34 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions

Automatically free up old states

Browse source
This commit is contained in:
Nexus 2023-04-09 21:13:07 +01:00
parent b842ef2acb
commit e8f08398a2
Signed by: nex
GPG key ID: 0FA334385D0B689F
1 changed files with 12 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
web/server.py
View file

@ -20,7 +20,7 @@ OAUTH_ENABLED = OAUTH_ID and OAUTH_SECRET and OAUTH_REDIRECT_URI
app = FastAPI(root_path="/jimmy") app = FastAPI(root_path="/jimmy")
app.state.bot = None app.state.bot = None
app.state.states = set() app.state.states = {}
app.state.http = httpx.Client() app.state.http = httpx.Client()
try: try:
@ -64,9 +64,16 @@ async def authenticate(req: Request, code: str = None, state: str = None):
) )
if not (code and state) or state not in app.state.states: if not (code and state) or state not in app.state.states:
value = os.urandom(8).hex() value = os.urandom(4).hex()
assert value not in app.state.states, "Generated a state that already exists." if value in app.state.states:
app.state.states.add(value) # remove any states older than 5 minutes
for _value in list(app.state.states):
if (datetime.now() - app.state.states[_value]).total_seconds() > 300:
del app.state.states[_value]
if value in app.state.states:
assert value not in app.state.states, "Generated a state that already exists and could not free any slots."
app.state.states[value] = datetime.now()
return RedirectResponse( return RedirectResponse(
discord.utils.oauth_url( discord.utils.oauth_url(
OAUTH_ID, OAUTH_ID,
@ -79,7 +86,7 @@ async def authenticate(req: Request, code: str = None, state: str = None):
} }
) )
else: else:
app.state.states.discard(state) app.state.states.pop(state)
# First, we need to do the auth code flow # First, we need to do the auth code flow
response = app.state.http.post( response = app.state.http.post(
"https://discord.com/api/oauth2/token", "https://discord.com/api/oauth2/token",