1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
|
"""Authentication endpoints."""
from uuid import UUID
from urllib.parse import urljoin, urlparse, urlunparse
from flask import (
flash, request, Blueprint, url_for, redirect, render_template,
current_app as app)
from . import session
from .client import SCOPE, no_token_post
from .checks import require_oauth2, user_logged_in
from .request_utils import user_details, process_error
toplevel = Blueprint("toplevel", __name__)
@toplevel.route("/register-client", methods=["GET", "POST"])
@require_oauth2
def register_client():
"""Register an OAuth2 client."""
return "USER IS LOGGED IN AND SUCCESSFULLY ACCESSED THIS ENDPOINT!"
@toplevel.route("/code", methods=["GET"])
def authorisation_code():
"""Use authorisation code to get token."""
def __error__(error):
flash(f"{error['error']}: {error['error_description']}",
"alert-danger")
return redirect("/")
def __success__(token):
session.set_user_token(token)
udets = user_details()
session.set_user_details({
"user_id": UUID(udets["user_id"]),
"name": udets["name"],
"email": udets["email"],
"token": session.user_token(),
"logged_in": True
})
return redirect("/")
code = request.args.get("code", "")
if bool(code):
base_url = urlparse(request.base_url, scheme=request.scheme)
request_data = {
"grant_type": "authorization_code",
"code": code,
"scope": SCOPE,
"redirect_uri": urljoin(
urlunparse(base_url),
url_for("oauth2.toplevel.authorisation_code")),
"client_id": app.config["OAUTH2_CLIENT_ID"]
}
return no_token_post(
"auth/token", data=request_data).either(
lambda err: __error__(process_error(err)), __success__)
flash("AuthorisationError: No code was provided.", "alert-danger")
return redirect("/")
|
