microblog/app/lookup.py

47 lines
1.6 KiB
Python
Raw Normal View History

2022-06-22 18:11:22 +00:00
import mf2py # type: ignore
from app import activitypub as ap
from app import webfinger
from app.actor import Actor
2022-08-15 20:22:15 +00:00
from app.actor import RemoteActor
2022-06-22 18:11:22 +00:00
from app.ap_object import RemoteObject
2022-06-29 18:43:17 +00:00
from app.database import AsyncSession
2022-08-21 07:52:48 +00:00
from app.source import _MENTION_REGEX
2022-06-22 18:11:22 +00:00
2022-06-29 18:43:17 +00:00
async def lookup(db_session: AsyncSession, query: str) -> Actor | RemoteObject:
query = query.strip()
2022-08-21 07:52:48 +00:00
if query.startswith("@") or _MENTION_REGEX.match("@" + query):
2022-06-29 22:28:07 +00:00
query = await webfinger.get_actor_url(query) # type: ignore # None check below
2022-06-22 18:11:22 +00:00
if not query:
raise ap.NotAnObjectError(query)
try:
2022-06-29 22:28:07 +00:00
ap_obj = await ap.fetch(query)
2022-06-22 18:11:22 +00:00
except ap.NotAnObjectError as not_an_object_error:
resp = not_an_object_error.resp
if not resp:
raise ap.NotAnObjectError(query)
alternate_obj = None
if resp.headers.get("content-type", "").startswith("text/html"):
for alternate in mf2py.parse(doc=resp.text).get("alternates", []):
if alternate.get("type") == "application/activity+json":
2022-06-29 22:28:07 +00:00
alternate_obj = await ap.fetch(alternate["url"])
2022-06-22 18:11:22 +00:00
if alternate_obj:
ap_obj = alternate_obj
else:
raise
2022-08-13 20:37:44 +00:00
if ap.as_list(ap_obj["type"])[0] in ap.ACTOR_TYPES:
2022-08-15 20:22:15 +00:00
return RemoteActor(ap_obj)
2022-06-22 18:11:22 +00:00
else:
2022-09-13 05:59:35 +00:00
# Some software return objects wrapped in a Create activity (like
# python-federation)
if ap.as_list(ap_obj["type"])[0] == "Create":
ap_obj = await ap.get_object(ap_obj)
2022-06-29 22:28:07 +00:00
return await RemoteObject.from_raw_object(ap_obj)