nikdoof/simple-webfinger
1
0
Fork 0
mirror of https://github.com/nikdoof/simple-webfinger.git synced 2025-12-13 10:22:15 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add Pydantic support for JRD (resolves #1)

Browse Source
This commit is contained in:
Andrew Williams
2024-08-16 16:12:47 +01:00
parent 82ba2af669
commit 18ab52e123
6 changed files with 221 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files

66
simple_webfinger/app.py
View File

@@ -1,41 +1,14 @@
from urllib.parse import urlparse
import yaml
import json
from flask import Flask, abort, request
from simple_webfinger.models.webfinger import JSONResourceDefinition
def get_account_links(user: str, data: dict) -> list:
links = []
account_data = data["accounts"][user]
# Append custom links
if "links" in account_data:
links.extend(account_data["links"])
if "mastodon" in account_data:
account, domain = account_data["mastodon"].split("@")
links.extend(
[
{
"rel": "http://webfinger.net/rel/profile-page",
"type": "text/html",
"href": "https://{0}/@{1}".format(domain, account),
},
{
"rel": "self",
"type": "application/activity+json",
"href": "https://{0}/users/{1}".format(domain, account),
},
{
"rel": "http://ostatus.org/schema/1.0/subscribe",
"template": "https://{0}/authorize_interaction?uri={{uri}}".format(
domain
),
},
]
)
# Append the OIDC link
if "oidc_href" in data:
links.append(
@@ -45,6 +18,34 @@ def get_account_links(user: str, data: dict) -> list:
}
)
# Append custom links
if account_data:
if "links" in account_data:
links.extend(account_data["links"])
if "mastodon" in account_data:
account, domain = account_data["mastodon"].split("@")
links.extend(
[
{
"rel": "http://webfinger.net/rel/profile-page",
"type": "text/html",
"href": "https://{0}/@{1}".format(domain, account),
},
{
"rel": "self",
"type": "application/activity+json",
"href": "https://{0}/users/{1}".format(domain, account),
},
{
"rel": "http://ostatus.org/schema/1.0/subscribe",
"template": "https://{0}/authorize_interaction?uri={{uri}}".format(
domain
),
},
]
)
return links
@@ -103,17 +104,20 @@ def create_app(config={}):
# If we have a 'rel' value on the request, filter down to the requested rel
# https://datatracker.ietf.org/doc/html/rfc7033#section-4.3
print(request.args)
rel = request.args.getlist("rel")
if rel:
links = filter_links(links, rel)
response = {"subject": resource, "links": links}
if "properties" in account_data and len(account_data['properties']):
# Add properties if defined in the config
if account_data and "properties" in account_data and len(account_data["properties"]):
response.update({"properties": account_data["properties"]})
return app.response_class(
response=json.dumps(response),
response=JSONResourceDefinition(**response).model_dump_json(
exclude_none=True
),
status=200,
mimetype="application/jrd+json",
)