canaille-globuzma/canaille/oidc/clients.py

162 lines
5.2 KiB
Python
Raw Normal View History

2020-08-17 13:49:48 +00:00
import datetime
2021-12-20 22:57:27 +00:00
from canaille.app import models
2023-04-09 13:52:55 +00:00
from canaille.app.flask import permissions_needed
from canaille.app.flask import render_htmx_template
from canaille.app.forms import TableForm
2023-03-29 22:40:25 +00:00
from canaille.oidc.forms import ClientAddForm
2021-12-20 22:57:27 +00:00
from flask import abort
from flask import Blueprint
from flask import flash
from flask import redirect
from flask import request
from flask import url_for
from flask_babel import gettext as _
from flask_themer import render_template
2020-08-17 13:49:48 +00:00
from werkzeug.security import gen_salt
2022-01-11 18:49:06 +00:00
bp = Blueprint("clients", __name__, url_prefix="/admin/client")
2020-08-17 13:49:48 +00:00
@bp.route("/", methods=["GET", "POST"])
2021-12-02 17:23:14 +00:00
@permissions_needed("manage_oidc")
2020-10-29 10:09:31 +00:00
def index(user):
table_form = TableForm(models.Client, formdata=request.form)
if request.form and request.form.get("page") and not table_form.validate():
abort(404)
return render_htmx_template(
"oidc/admin/client_list.html", menuitem="admin", table_form=table_form
2022-01-11 18:49:06 +00:00
)
2020-08-17 13:49:48 +00:00
@bp.route("/add", methods=["GET", "POST"])
2021-12-02 17:23:14 +00:00
@permissions_needed("manage_oidc")
2020-10-29 10:09:31 +00:00
def add(user):
2023-03-29 22:40:25 +00:00
form = ClientAddForm(request.form or None)
2020-08-17 13:49:48 +00:00
2023-06-22 09:39:50 +00:00
if not request.form or form.form_control():
2022-01-11 18:49:06 +00:00
return render_template(
"oidc/admin/client_add.html", form=form, menuitem="admin"
)
2020-08-17 13:49:48 +00:00
if not form.validate():
flash(
2021-10-13 09:52:02 +00:00
_("The client has not been added. Please check your information."),
"error",
2020-08-17 13:49:48 +00:00
)
2022-01-11 18:49:06 +00:00
return render_template(
"oidc/admin/client_add.html", form=form, menuitem="admin"
)
2020-08-17 13:49:48 +00:00
client_id = gen_salt(24)
2023-03-17 23:38:56 +00:00
client_id_issued_at = datetime.datetime.now(datetime.timezone.utc)
client = models.Client(
client_id=client_id,
2022-10-17 15:49:52 +00:00
client_id_issued_at=client_id_issued_at,
client_name=form["client_name"].data,
2023-06-22 09:39:50 +00:00
contacts=form["contacts"].data,
2022-10-17 15:49:52 +00:00
client_uri=form["client_uri"].data,
grant_types=form["grant_types"].data,
2023-06-22 09:39:50 +00:00
redirect_uris=form["redirect_uris"].data,
post_logout_redirect_uris=form["post_logout_redirect_uris"].data,
2022-10-17 15:49:52 +00:00
response_types=form["response_types"].data,
scope=form["scope"].data.split(" "),
token_endpoint_auth_method=form["token_endpoint_auth_method"].data,
logo_uri=form["logo_uri"].data,
tos_uri=form["tos_uri"].data,
policy_uri=form["policy_uri"].data,
software_id=form["software_id"].data,
software_version=form["software_version"].data,
jwk=form["jwk"].data,
2022-10-17 15:49:52 +00:00
jwks_uri=form["jwks_uri"].data,
preconsent=form["preconsent"].data,
2022-10-17 15:49:52 +00:00
client_secret=""
if form["token_endpoint_auth_method"].data == "none"
2020-08-17 13:49:48 +00:00
else gen_salt(48),
)
client.audience = [client]
2020-08-17 13:49:48 +00:00
client.save()
flash(
2021-10-13 09:52:02 +00:00
_("The client has been created."),
"success",
2020-08-17 13:49:48 +00:00
)
2022-01-11 18:49:06 +00:00
return redirect(url_for("oidc.clients.edit", client_id=client_id))
2020-08-17 13:49:48 +00:00
@bp.route("/edit/<client_id>", methods=["GET", "POST"])
2021-12-02 17:23:14 +00:00
@permissions_needed("manage_oidc")
2020-10-29 10:09:31 +00:00
def edit(user, client_id):
2023-06-22 09:39:50 +00:00
if request.form and request.form.get("action") == "delete":
2020-11-23 16:32:40 +00:00
return client_delete(client_id)
2023-06-22 09:39:50 +00:00
return client_edit(client_id)
2020-11-23 16:32:40 +00:00
def client_edit(client_id):
client = models.Client.get(client_id=client_id)
if not client:
abort(404)
data = {attribute: getattr(client, attribute) for attribute in client.attributes}
data["scope"] = " ".join(data["scope"])
data["preconsent"] = client.preconsent
2023-03-29 22:40:25 +00:00
form = ClientAddForm(request.form or None, data=data, client=client)
2020-08-17 13:49:48 +00:00
2023-06-22 09:39:50 +00:00
if not request.form or form.form_control():
2020-10-21 10:14:35 +00:00
return render_template(
2022-01-11 18:49:06 +00:00
"oidc/admin/client_edit.html", form=form, client=client, menuitem="admin"
2020-10-21 10:14:35 +00:00
)
2020-08-17 13:49:48 +00:00
if not form.validate():
flash(
2020-10-28 14:49:14 +00:00
_("The client has not been edited. Please check your information."),
2020-08-17 13:49:48 +00:00
"error",
)
return render_template(
"oidc/admin/client_edit.html", form=form, client=client, menuitem="admin"
2020-08-17 13:49:48 +00:00
)
client.update(
client_name=form["client_name"].data,
2023-06-22 09:39:50 +00:00
contacts=form["contacts"].data,
client_uri=form["client_uri"].data,
grant_types=form["grant_types"].data,
2023-06-22 09:39:50 +00:00
redirect_uris=form["redirect_uris"].data,
post_logout_redirect_uris=form["post_logout_redirect_uris"].data,
response_types=form["response_types"].data,
scope=form["scope"].data.split(" "),
token_endpoint_auth_method=form["token_endpoint_auth_method"].data,
logo_uri=form["logo_uri"].data,
tos_uri=form["tos_uri"].data,
policy_uri=form["policy_uri"].data,
software_id=form["software_id"].data,
software_version=form["software_version"].data,
jwk=form["jwk"].data,
jwks_uri=form["jwks_uri"].data,
audience=[models.Client.get(id=id) for id in form["audience"].data],
preconsent=form["preconsent"].data,
2020-10-21 10:14:35 +00:00
)
client.save()
flash(
_("The client has been edited."),
"success",
)
return redirect(url_for("oidc.clients.edit", client_id=client_id))
2020-11-23 16:32:40 +00:00
def client_delete(client_id):
client = models.Client.get(client_id=client_id)
if not client:
abort(404)
2020-11-23 16:32:40 +00:00
flash(
2021-10-13 09:52:02 +00:00
_("The client has been deleted."),
"success",
2020-11-23 16:32:40 +00:00
)
client.delete()
2022-01-11 18:49:06 +00:00
return redirect(url_for("oidc.clients.index"))