youngkingdom/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-05-02 16:38:04 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: add invitation email judgment

Browse Source
This commit is contained in:
Joe
2024-09-02 18:04:11 +08:00
parent 7136714cfa
commit 7e48aa6402
2 changed files with 8 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
api/controllers/console/auth/oauth.py
View File

@ -77,6 +77,11 @@ class OAuthCallback(Resource):
return {"error": "OAuth process failed"}, 400
if invite_token:
invitation = RegisterService._get_invitation_by_token(token=invite_token)
if invitation:
invitation_email = invitation.get("email", None)
if invitation_email != user_info.email:
return redirect(f"{dify_config.CONSOLE_WEB_URL}/signin?message=InvalidToken")
return redirect(f"{dify_config.CONSOLE_WEB_URL}/signin/invite-settings?invite_token={invite_token}")
try: