128 lines
4.6 KiB
Python
128 lines
4.6 KiB
Python
from authlib.integrations.base_client.errors import MissingTokenError
|
|
from flask import Blueprint, redirect, request, url_for, render_template
|
|
from flask import current_app, session
|
|
from flask import jsonify
|
|
from flask.typing import ResponseReturnValue
|
|
from flask_login import current_user, logout_user
|
|
from oauthlib.oauth2.rfc6749.errors import TokenExpiredError
|
|
from authlib.integrations.base_client.errors import InvalidTokenError
|
|
from ory_hydra_client.api.o_auth_2 import list_o_auth_2_clients, get_o_auth_2_client, set_o_auth_2_client, create_o_auth_2_client
|
|
from ory_hydra_client.models import OAuth20Client, GenericError
|
|
from typing import Optional, List
|
|
from collections.abc import Iterable
|
|
from http import HTTPStatus
|
|
import httpx
|
|
import logging
|
|
|
|
from ..model import db, User
|
|
from .oauth2 import redirect_login, oauth2
|
|
from ..form.admin import OAuth2ClientForm
|
|
from ..hydra import hydra_service
|
|
|
|
|
|
admin_views = Blueprint('admin', __name__, url_prefix='/admin')
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
def before_request() -> Optional[ResponseReturnValue]:
|
|
try:
|
|
resp = oauth2.custom.get('/userinfo')
|
|
data = resp.json()
|
|
if not current_user.is_authenticated or resp.status_code != 200:
|
|
return redirect_login()
|
|
if 'groups' not in data or 'admin' not in data['groups']:
|
|
return 'Not an admin', 403
|
|
except (MissingTokenError, InvalidTokenError):
|
|
return redirect_login()
|
|
return None
|
|
|
|
|
|
admin_views.before_request(before_request)
|
|
|
|
|
|
@admin_views.route('/', methods=['GET', 'POST'])
|
|
async def index() -> ResponseReturnValue:
|
|
return render_template('admin/index.html.j2')
|
|
|
|
|
|
@admin_views.route('/user', methods=['GET'])
|
|
async def users() -> ResponseReturnValue:
|
|
users = User.query.all() # type: Iterable[User]
|
|
return render_template('admin/users.html.j2', users=users)
|
|
|
|
|
|
@admin_views.route('/registrations', methods=['GET'])
|
|
def registrations() -> ResponseReturnValue:
|
|
users = User.query.filter_by(enabled=False).all() # type: Iterable[User]
|
|
return render_template('admin/registrations.html.j2', users=users)
|
|
|
|
|
|
@admin_views.route('/registration/<registration_id>', methods=['DELETE'])
|
|
def registration_delete(registration_id) -> ResponseReturnValue:
|
|
user = User.query.get(registration_id) # type: Optional[User]
|
|
if user is None:
|
|
return jsonify({}), 404
|
|
db.session.delete(user)
|
|
db.session.commit()
|
|
return jsonify({})
|
|
|
|
|
|
@admin_views.route('/registration/<registration_id>', methods=['PUT'])
|
|
def registration_accept(registration_id) -> ResponseReturnValue:
|
|
user = User.query.get(registration_id) # type: Optional[User]
|
|
if user is None:
|
|
return jsonify({'message':'user not found'}), 404
|
|
user.enabled = True
|
|
db.session.commit()
|
|
return jsonify({})
|
|
|
|
|
|
@admin_views.route('/clients')
|
|
async def clients() -> ResponseReturnValue:
|
|
response = await list_o_auth_2_clients.asyncio_detailed(_client=hydra_service.hydra_client)
|
|
clients = response.parsed
|
|
if clients is None:
|
|
logger.error(f"could not fetch client list response {response}")
|
|
return 'internal error', 500
|
|
return render_template('admin/clients.html.j2', clients=clients)
|
|
|
|
@admin_views.route('/client/<client_id>', methods=['GET', 'POST'])
|
|
async def client(client_id: str) -> ResponseReturnValue:
|
|
|
|
client = await get_o_auth_2_client.asyncio(client_id, _client=hydra_service.hydra_client)
|
|
if client is None or isinstance( client, GenericError):
|
|
logger.error(f"oauth2 client not found with id: '{client_id}'")
|
|
return 'client not found', 404
|
|
|
|
form = OAuth2ClientForm(obj=client)
|
|
if form.validate_on_submit():
|
|
form.populate_obj(client)
|
|
|
|
client = await set_o_auth_2_client.asyncio(id=client_id ,json_body=client, _client=hydra_service.hydra_client)
|
|
if client is None or isinstance(client, GenericError):
|
|
logger.error(f"oauth2 client update failed: '{client_id}'")
|
|
return 'client update failed', 500
|
|
|
|
|
|
|
|
return render_template('admin/client.html.j2', client=client, form=form)
|
|
|
|
|
|
@admin_views.route('/client_new', methods=['GET','POST'])
|
|
async def client_new() -> ResponseReturnValue:
|
|
|
|
client = OAuth20Client()
|
|
|
|
form = OAuth2ClientForm()
|
|
if form.validate_on_submit():
|
|
form.populate_obj(client)
|
|
|
|
resp_client = await create_o_auth_2_client.asyncio(json_body=client, _client=hydra_service.hydra_client)
|
|
if resp_client is None:
|
|
logger.error(f"oauth2 client created failed: '{client.client_id}'")
|
|
return 'internal error', 500
|
|
return redirect(url_for('.client', client_id=client.client_id))
|
|
|
|
|
|
return render_template('admin/client.html.j2', client=client, form=form)
|