mirror of
https://github.com/cwinfo/powerdns-admin.git
synced 2024-09-19 14:59:35 +00:00
58 lines
2.0 KiB
Python
58 lines
2.0 KiB
Python
from flask import request, session, redirect, url_for, current_app
|
|
|
|
from .base import authlib_oauth_client
|
|
from ..models.setting import Setting
|
|
|
|
|
|
def google_oauth():
|
|
if not Setting().get('google_oauth_enabled'):
|
|
return None
|
|
|
|
def fetch_google_token():
|
|
return session.get('google_token')
|
|
|
|
def update_token(token):
|
|
session['google_token'] = token
|
|
return token
|
|
|
|
authlib_params = {
|
|
'client_id': Setting().get('google_oauth_client_id'),
|
|
'client_secret': Setting().get('google_oauth_client_secret'),
|
|
'api_base_url': Setting().get('google_base_url'),
|
|
'request_token_url': None,
|
|
'client_kwargs': {'scope': Setting().get('google_oauth_scope')},
|
|
'fetch_token': fetch_google_token,
|
|
'update_token': update_token
|
|
}
|
|
|
|
server_metadata_url = Setting().get('google_oauth_metadata_url')
|
|
|
|
if isinstance(server_metadata_url, str) and len(server_metadata_url.strip()) > 0:
|
|
authlib_params['server_metadata_url'] = server_metadata_url
|
|
else:
|
|
authlib_params['access_token_url'] = Setting().get('google_token_url')
|
|
authlib_params['authorize_url'] = Setting().get('google_authorize_url')
|
|
|
|
google = authlib_oauth_client.register(
|
|
'google',
|
|
**authlib_params
|
|
)
|
|
|
|
@current_app.route('/google/authorized')
|
|
def google_authorized():
|
|
use_ssl = current_app.config.get('SERVER_EXTERNAL_SSL')
|
|
params = {'_external': True}
|
|
if isinstance(use_ssl, bool):
|
|
params['_scheme'] = 'https' if use_ssl else 'http'
|
|
session['google_oauthredir'] = url_for('.google_authorized', **params)
|
|
token = google.authorize_access_token()
|
|
if token is None:
|
|
return 'Access denied: reason=%s error=%s' % (
|
|
request.args['error_reason'],
|
|
request.args['error_description']
|
|
)
|
|
session['google_token'] = token
|
|
return redirect(url_for('index.login', **params))
|
|
|
|
return google
|