canaille-globuzma/canaille/oidc/endpoints/clients.py
2024-03-15 19:58:37 +01:00

153 lines
5 KiB
Python

import datetime
from flask import Blueprint
from flask import abort
from flask import flash
from flask import redirect
from flask import request
from flask import url_for
from werkzeug.security import gen_salt
from canaille.app import models
from canaille.app.flask import permissions_needed
from canaille.app.flask import render_htmx_template
from canaille.app.forms import TableForm
from canaille.app.i18n import gettext as _
from canaille.app.themes import render_template
from .forms import ClientAddForm
bp = Blueprint("clients", __name__, url_prefix="/admin/client")
@bp.route("/", methods=["GET", "POST"])
@permissions_needed("manage_oidc")
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(
"client_list.html", menuitem="admin", table_form=table_form
)
@bp.route("/add", methods=["GET", "POST"])
@permissions_needed("manage_oidc")
def add(user):
form = ClientAddForm(request.form or None)
if not request.form or form.form_control():
return render_template("client_add.html", form=form, menuitem="admin")
if not form.validate():
flash(
_("The client has not been added. Please check your information."),
"error",
)
return render_template("client_add.html", form=form, menuitem="admin")
client_id = gen_salt(24)
client_id_issued_at = datetime.datetime.now(datetime.timezone.utc)
client = models.Client(
client_id=client_id,
client_id_issued_at=client_id_issued_at,
client_name=form["client_name"].data,
contacts=form["contacts"].data,
client_uri=form["client_uri"].data,
grant_types=form["grant_types"].data,
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,
preconsent=form["preconsent"].data,
client_secret=""
if form["token_endpoint_auth_method"].data == "none"
else gen_salt(48),
)
client.audience = [client]
client.save()
flash(
_("The client has been created."),
"success",
)
return redirect(url_for("oidc.clients.edit", client=client))
@bp.route("/edit/<client:client>", methods=["GET", "POST"])
@permissions_needed("manage_oidc")
def edit(user, client):
if request.form.get("action") == "confirm-delete":
return render_template("modals/delete-client.html", client=client)
if request.form and request.form.get("action") == "delete":
return client_delete(client)
return client_edit(client)
def client_edit(client):
data = {attribute: getattr(client, attribute) for attribute in client.attributes}
if data["scope"]:
data["scope"] = " ".join(data["scope"])
data["preconsent"] = client.preconsent
form = ClientAddForm(request.form or None, data=data, client=client)
if not request.form or form.form_control():
return render_template(
"client_edit.html", form=form, client=client, menuitem="admin"
)
if not form.validate():
flash(
_("The client has not been edited. Please check your information."),
"error",
)
return render_template(
"client_edit.html", form=form, client=client, menuitem="admin"
)
client.update(
client_name=form["client_name"].data,
contacts=form["contacts"].data,
client_uri=form["client_uri"].data,
grant_types=form["grant_types"].data,
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=form["audience"].data,
preconsent=form["preconsent"].data,
)
client.save()
flash(
_("The client has been edited."),
"success",
)
return redirect(url_for("oidc.clients.edit", client=client))
def client_delete(client):
flash(
_("The client has been deleted."),
"success",
)
client.delete()
return redirect(url_for("oidc.clients.index"))