summaryrefslogtreecommitdiffstats
path: root/activities/views
diff options
context:
space:
mode:
authorAndrew Godwin2022-12-04 21:13:33 -0700
committerAndrew Godwin2022-12-04 21:13:33 -0700
commit93ccb5dd32f4e3942d08814e5a86fb30c90c9ff6 (patch)
tree001b4b4ab9270b007aa90bb4218836b1ef072d84 /activities/views
parent28ac03c42fb82db5f1dad1f711cffdde0a0cce6c (diff)
downloadtakahe-93ccb5dd32f4e3942d08814e5a86fb30c90c9ff6.tar.gz
takahe-93ccb5dd32f4e3942d08814e5a86fb30c90c9ff6.tar.bz2
takahe-93ccb5dd32f4e3942d08814e5a86fb30c90c9ff6.zip
Make search work with URLs
Diffstat (limited to 'activities/views')
-rw-r--r--activities/views/search.py86
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