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

feat: add oauth invite redict

Browse Source
This commit is contained in:
Joe
2024-09-02 14:50:45 +08:00
parent cbdbfb844d
commit c84f004035
2 changed files with 19 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -43,7 +43,7 @@ def get_oauth_providers():
class OAuthLogin(Resource):
def get(self, provider: str):
def get(self, provider: str, invite_toke: Optional[str] = None):
OAUTH_PROVIDERS = get_oauth_providers()
with current_app.app_context():
oauth_provider = OAUTH_PROVIDERS.get(provider)
@ -51,7 +51,7 @@ class OAuthLogin(Resource):
if not oauth_provider:
return {"error": "Invalid provider"}, 400
auth_url = oauth_provider.get_authorization_url()
auth_url = oauth_provider.get_authorization_url(invite_toke)
return redirect(auth_url)
@ -64,6 +64,11 @@ class OAuthCallback(Resource):
return {"error": "Invalid provider"}, 400
code = request.args.get("code")
state = request.args.get("state")
invite_token = None
if state:
invite_token = state
try:
token = oauth_provider.get_access_token(code)
user_info = oauth_provider.get_user_info(token)
@ -71,6 +76,11 @@ class OAuthCallback(Resource):
logging.exception(f"An error occurred during the OAuth process with {provider}: {e.response.text}")
return {"error": "OAuth process failed"}, 400
if invite_token:
return redirect(
f"{dify_config.CONSOLE_WEB_URL}/invite-settings?invite_token={invite_token}"
)
try:
account = _generate_account(provider, user_info)
except services.errors.account.AccountNotFound as e: