diff options
Diffstat (limited to 'activities/views/search.py')
| -rw-r--r-- | activities/views/search.py | 86 |
1 files changed, 79 insertions, 7 deletions
diff --git a/activities/views/search.py b/activities/views/search.py index 8bdef78..ab37e17 100644 --- a/activities/views/search.py +++ b/activities/views/search.py @@ -1,11 +1,14 @@ from typing import Set +import httpx from asgiref.sync import async_to_sync from django import forms from django.views.generic import FormView -from activities.models import Hashtag +from activities.models import Hashtag, Post +from core.ld import canonicalise from users.models import Domain, Identity, IdentityStates +from users.models.system_actor import SystemActor class Search(FormView): @@ -14,11 +17,20 @@ class Search(FormView): class form_class(forms.Form): query = forms.CharField( - help_text="Search for a user by @username@domain or hashtag by #tagname", + help_text="Search for:\nA user by @username@domain or their profile URL\nA hashtag by #tagname\nA post by its URL", widget=forms.TextInput(attrs={"type": "search", "autofocus": "autofocus"}), ) - def search_identities(self, query: str): + def search_identities_handle(self, query: str): + """ + Searches for identities by their handles + """ + + # Short circuit if it's obviously not for us + if "://" in query: + return set() + + # Try to fetch the user by handle query = query.lstrip("@") results: Set[Identity] = set() if "@" in query: @@ -52,12 +64,65 @@ class Search(FormView): results.add(identity) return results + def search_url(self, query: str) -> Post | Identity | None: + """ + Searches for an identity or post by URL. + """ + + # Short circuit if it's obviously not for us + if "://" not in query: + return None + + # Clean up query + query = query.strip() + + # Fetch the provided URL as the system actor to retrieve the AP JSON + try: + response = async_to_sync(SystemActor().signed_request)( + method="get", uri=query + ) + except (httpx.RequestError, httpx.ConnectError): + return None + if response.status_code >= 400: + return None + document = canonicalise(response.json(), include_security=True) + type = document.get("type", "unknown").lower() + + # Is it an identity? + if type == "person": + # Try and retrieve the profile by actor URI + identity = Identity.by_actor_uri(document["id"], create=True) + if identity and identity.state == IdentityStates.outdated: + async_to_sync(identity.fetch_actor)() + return identity + + # Is it a post? + elif type == "note": + # Try and retrieve the post by URI + # (we do not trust the JSON we just got - fetch from source!) + try: + post = Post.by_object_uri(document["id"], fetch=True) + # We may need to live-fetch the identity too + if post.author.state == IdentityStates.outdated: + async_to_sync(post.author.fetch_actor)() + return post + except Post.DoesNotExist: + return None + + # Dunno what it is + else: + return None + def search_hashtags(self, query: str): - results: Set[Hashtag] = set() + """ + Searches for hashtags by their name + """ - if "@" in query: - return results + # Short circuit out if it's obviously not a hashtag + if "@" in query or "://" in query: + return set() + results: Set[Hashtag] = set() query = query.lstrip("#") for hashtag in Hashtag.objects.public().hashtag_or_alias(query)[:10]: results.add(hashtag) @@ -68,10 +133,17 @@ class Search(FormView): def form_valid(self, form): query = form.cleaned_data["query"].lower() results = { - "identities": self.search_identities(query), + "identities": self.search_identities_handle(query), "hashtags": self.search_hashtags(query), + "posts": set(), } + url_result = self.search_url(query) + if isinstance(url_result, Identity): + results["identities"].add(url_result) + if isinstance(url_result, Post): + results["posts"].add(url_result) + # Render results context = self.get_context_data(form=form) context["results"] = results |