github/OpenHands
1
0
Fork 0
mirror of https://github.com/OpenHands/OpenHands.git synced 2025-12-26 13:52:43 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update openhands/server/listen.py

Browse Source
This commit is contained in:
Robert Brennan 2024-11-08 15:25:24 -05:00 committed by GitHub
parent 3356753f79
commit 8a0620b21e
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
openhands/server/listen.py
View File

@ -328,17 +328,7 @@ async def websocket_endpoint(websocket: WebSocket):
jwt_token = protocols[1] if protocols[1] != 'NO_JWT' else ''
github_token = protocols[2] if protocols[2] != 'NO_GITHUB' else ''
# First check for auth cookie
cookie_header = websocket.headers.get('cookie', '')
github_auth_cookie = None
for cookie in cookie_header.split(';'):
name, _, value = cookie.strip().partition('=')
if name == 'github_auth':
github_auth_cookie = value
break
# If cookie exists, use it, otherwise verify with GitHub
if not github_auth_cookie and not await authenticate_github_user(github_token):
if not await authenticate_github_user(github_token):
await websocket.close(code=status.WS_1008_POLICY_VIOLATION)
return