mirror of
https://github.com/jointakahe/takahe.git
synced 2024-11-28 18:21:00 +00:00
More API read coverage
This commit is contained in:
parent
3e062aed36
commit
fc8a21fc5c
15 changed files with 536 additions and 147 deletions
|
@ -185,3 +185,10 @@ class Hashtag(StatorModel):
|
||||||
return f'<a class="hashtag" href="/tags/{hashtag.lower()}/">#{hashtag}</a>'
|
return f'<a class="hashtag" href="/tags/{hashtag.lower()}/">#{hashtag}</a>'
|
||||||
|
|
||||||
return mark_safe(Hashtag.hashtag_regex.sub(replacer, content))
|
return mark_safe(Hashtag.hashtag_regex.sub(replacer, content))
|
||||||
|
|
||||||
|
def to_mastodon_json(self):
|
||||||
|
return {
|
||||||
|
"name": self.hashtag,
|
||||||
|
"url": self.urls.view.full(),
|
||||||
|
"history": [],
|
||||||
|
}
|
||||||
|
|
|
@ -262,7 +262,7 @@ class Post(StatorModel):
|
||||||
r"(^|[^\w\d\-_])@([\w\d\-_]+(?:@[\w\d\-_]+\.[\w\d\-_\.]+)?)"
|
r"(^|[^\w\d\-_])@([\w\d\-_]+(?:@[\w\d\-_]+\.[\w\d\-_\.]+)?)"
|
||||||
)
|
)
|
||||||
|
|
||||||
def linkify_mentions(self, content, local=False):
|
def linkify_mentions(self, content: str, local: bool = False) -> str:
|
||||||
"""
|
"""
|
||||||
Links mentions _in the context of the post_ - as in, using the mentions
|
Links mentions _in the context of the post_ - as in, using the mentions
|
||||||
property as the only source (as we might be doing this without other
|
property as the only source (as we might be doing this without other
|
||||||
|
|
|
@ -1,5 +1,7 @@
|
||||||
from django.db import models
|
from django.db import models
|
||||||
|
|
||||||
|
from core.ld import format_ld_date
|
||||||
|
|
||||||
|
|
||||||
class TimelineEvent(models.Model):
|
class TimelineEvent(models.Model):
|
||||||
"""
|
"""
|
||||||
|
@ -143,3 +145,26 @@ class TimelineEvent(models.Model):
|
||||||
subject_post_id=interaction.post_id,
|
subject_post_id=interaction.post_id,
|
||||||
subject_identity_id=interaction.identity_id,
|
subject_identity_id=interaction.identity_id,
|
||||||
).delete()
|
).delete()
|
||||||
|
|
||||||
|
### Mastodon Client API ###
|
||||||
|
|
||||||
|
def to_mastodon_notification_json(self):
|
||||||
|
result = {
|
||||||
|
"id": self.pk,
|
||||||
|
"created_at": format_ld_date(self.created),
|
||||||
|
"account": self.subject_identity.to_mastodon_json(),
|
||||||
|
}
|
||||||
|
if self.type == self.Types.liked:
|
||||||
|
result["type"] = "favourite"
|
||||||
|
result["status"] = self.subject_post.to_mastodon_json()
|
||||||
|
elif self.type == self.Types.boosted:
|
||||||
|
result["type"] = "reblog"
|
||||||
|
result["status"] = self.subject_post.to_mastodon_json()
|
||||||
|
elif self.type == self.Types.mentioned:
|
||||||
|
result["type"] = "mention"
|
||||||
|
result["status"] = self.subject_post.to_mastodon_json()
|
||||||
|
elif self.type == self.Types.followed:
|
||||||
|
result["type"] = "follow"
|
||||||
|
else:
|
||||||
|
raise ValueError(f"Cannot convert {self.type} to notification JSON")
|
||||||
|
return result
|
||||||
|
|
136
activities/search.py
Normal file
136
activities/search.py
Normal file
|
@ -0,0 +1,136 @@
|
||||||
|
import httpx
|
||||||
|
from asgiref.sync import async_to_sync
|
||||||
|
|
||||||
|
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 Searcher:
|
||||||
|
"""
|
||||||
|
Captures the logic needed to search - reused in the UI and API
|
||||||
|
"""
|
||||||
|
|
||||||
|
def __init__(self, query: str, identity: Identity | None):
|
||||||
|
self.query = query.strip().lower()
|
||||||
|
self.identity = identity
|
||||||
|
|
||||||
|
def search_identities_handle(self) -> set[Identity]:
|
||||||
|
"""
|
||||||
|
Searches for identities by their handles
|
||||||
|
"""
|
||||||
|
|
||||||
|
# Short circuit if it's obviously not for us
|
||||||
|
if "://" in self.query:
|
||||||
|
return set()
|
||||||
|
|
||||||
|
# Try to fetch the user by handle
|
||||||
|
handle = self.query.lstrip("@")
|
||||||
|
results: set[Identity] = set()
|
||||||
|
if "@" in handle:
|
||||||
|
username, domain = handle.split("@", 1)
|
||||||
|
|
||||||
|
# Resolve the domain to the display domain
|
||||||
|
domain_instance = Domain.get_domain(domain)
|
||||||
|
try:
|
||||||
|
if domain_instance is None:
|
||||||
|
raise Identity.DoesNotExist()
|
||||||
|
identity = Identity.objects.get(
|
||||||
|
domain=domain_instance, username=username
|
||||||
|
)
|
||||||
|
except Identity.DoesNotExist:
|
||||||
|
if self.identity is not None:
|
||||||
|
# Allow authenticated users to fetch remote
|
||||||
|
identity = Identity.by_username_and_domain(
|
||||||
|
username, domain, fetch=True
|
||||||
|
)
|
||||||
|
if identity and identity.state == IdentityStates.outdated:
|
||||||
|
async_to_sync(identity.fetch_actor)()
|
||||||
|
else:
|
||||||
|
identity = None
|
||||||
|
if identity:
|
||||||
|
results.add(identity)
|
||||||
|
|
||||||
|
else:
|
||||||
|
for identity in Identity.objects.filter(username=handle)[:20]:
|
||||||
|
results.add(identity)
|
||||||
|
for identity in Identity.objects.filter(username__startswith=handle)[:20]:
|
||||||
|
results.add(identity)
|
||||||
|
return results
|
||||||
|
|
||||||
|
def search_url(self) -> Post | Identity | None:
|
||||||
|
"""
|
||||||
|
Searches for an identity or post by URL.
|
||||||
|
"""
|
||||||
|
|
||||||
|
# Short circuit if it's obviously not for us
|
||||||
|
if "://" not in self.query:
|
||||||
|
return None
|
||||||
|
|
||||||
|
# Fetch the provided URL as the system actor to retrieve the AP JSON
|
||||||
|
try:
|
||||||
|
response = async_to_sync(SystemActor().signed_request)(
|
||||||
|
method="get",
|
||||||
|
uri=self.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:
|
||||||
|
return Post.by_object_uri(document["id"], fetch=True)
|
||||||
|
except Post.DoesNotExist:
|
||||||
|
return None
|
||||||
|
|
||||||
|
# Dunno what it is
|
||||||
|
else:
|
||||||
|
return None
|
||||||
|
|
||||||
|
def search_hashtags(self) -> set[Hashtag]:
|
||||||
|
"""
|
||||||
|
Searches for hashtags by their name
|
||||||
|
"""
|
||||||
|
|
||||||
|
# Short circuit out if it's obviously not a hashtag
|
||||||
|
if "@" in self.query or "://" in self.query:
|
||||||
|
return set()
|
||||||
|
|
||||||
|
results: set[Hashtag] = set()
|
||||||
|
name = self.query.lstrip("#")
|
||||||
|
for hashtag in Hashtag.objects.public().hashtag_or_alias(name)[:10]:
|
||||||
|
results.add(hashtag)
|
||||||
|
for hashtag in Hashtag.objects.public().filter(hashtag__startswith=name)[:10]:
|
||||||
|
results.add(hashtag)
|
||||||
|
return results
|
||||||
|
|
||||||
|
def search_all(self):
|
||||||
|
"""
|
||||||
|
Returns all possible results for a search
|
||||||
|
"""
|
||||||
|
results = {
|
||||||
|
"identities": self.search_identities_handle(),
|
||||||
|
"hashtags": self.search_hashtags(),
|
||||||
|
"posts": set(),
|
||||||
|
}
|
||||||
|
url_result = self.search_url()
|
||||||
|
if isinstance(url_result, Identity):
|
||||||
|
results["identities"].add(url_result)
|
||||||
|
if isinstance(url_result, Post):
|
||||||
|
results["posts"].add(url_result)
|
||||||
|
return results
|
|
@ -1,12 +1,7 @@
|
||||||
import httpx
|
|
||||||
from asgiref.sync import async_to_sync
|
|
||||||
from django import forms
|
from django import forms
|
||||||
from django.views.generic import FormView
|
from django.views.generic import FormView
|
||||||
|
|
||||||
from activities.models import Hashtag, Post
|
from activities.search import Searcher
|
||||||
from core.ld import canonicalise
|
|
||||||
from users.models import Domain, Identity, IdentityStates
|
|
||||||
from users.models.system_actor import SystemActor
|
|
||||||
|
|
||||||
|
|
||||||
class Search(FormView):
|
class Search(FormView):
|
||||||
|
@ -19,126 +14,9 @@ class Search(FormView):
|
||||||
widget=forms.TextInput(attrs={"type": "search", "autofocus": "autofocus"}),
|
widget=forms.TextInput(attrs={"type": "search", "autofocus": "autofocus"}),
|
||||||
)
|
)
|
||||||
|
|
||||||
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:
|
|
||||||
username, domain = query.split("@", 1)
|
|
||||||
|
|
||||||
# Resolve the domain to the display domain
|
|
||||||
domain_instance = Domain.get_domain(domain)
|
|
||||||
try:
|
|
||||||
if domain_instance is None:
|
|
||||||
raise Identity.DoesNotExist()
|
|
||||||
identity = Identity.objects.get(
|
|
||||||
domain=domain_instance, username=username
|
|
||||||
)
|
|
||||||
except Identity.DoesNotExist:
|
|
||||||
if self.request.identity is not None:
|
|
||||||
# Allow authenticated users to fetch remote
|
|
||||||
identity = Identity.by_username_and_domain(
|
|
||||||
username, domain, fetch=True
|
|
||||||
)
|
|
||||||
if identity and identity.state == IdentityStates.outdated:
|
|
||||||
async_to_sync(identity.fetch_actor)()
|
|
||||||
else:
|
|
||||||
identity = None
|
|
||||||
if identity:
|
|
||||||
results.add(identity)
|
|
||||||
|
|
||||||
else:
|
|
||||||
for identity in Identity.objects.filter(username=query)[:20]:
|
|
||||||
results.add(identity)
|
|
||||||
for identity in Identity.objects.filter(username__startswith=query)[:20]:
|
|
||||||
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:
|
|
||||||
return Post.by_object_uri(document["id"], fetch=True)
|
|
||||||
except Post.DoesNotExist:
|
|
||||||
return None
|
|
||||||
|
|
||||||
# Dunno what it is
|
|
||||||
else:
|
|
||||||
return None
|
|
||||||
|
|
||||||
def search_hashtags(self, query: str):
|
|
||||||
"""
|
|
||||||
Searches for hashtags by their name
|
|
||||||
"""
|
|
||||||
|
|
||||||
# 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)
|
|
||||||
for hashtag in Hashtag.objects.public().filter(hashtag__startswith=query)[:10]:
|
|
||||||
results.add(hashtag)
|
|
||||||
return results
|
|
||||||
|
|
||||||
def form_valid(self, form):
|
def form_valid(self, form):
|
||||||
query = form.cleaned_data["query"].lower()
|
searcher = Searcher(form.cleaned_data["query"], self.request.identity)
|
||||||
results = {
|
|
||||||
"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
|
# Render results
|
||||||
context = self.get_context_data(form=form)
|
context = self.get_context_data(form=form)
|
||||||
context["results"] = results
|
context["results"] = searcher.search_all()
|
||||||
return self.render_to_response(context)
|
return self.render_to_response(context)
|
||||||
|
|
|
@ -106,3 +106,57 @@ class Status(Schema):
|
||||||
muted: bool | None
|
muted: bool | None
|
||||||
bookmarked: bool | None
|
bookmarked: bool | None
|
||||||
pinned: bool | None
|
pinned: bool | None
|
||||||
|
|
||||||
|
|
||||||
|
class Conversation(Schema):
|
||||||
|
id: str
|
||||||
|
unread: bool
|
||||||
|
accounts: list[Account]
|
||||||
|
last_status: Status | None = Field(...)
|
||||||
|
|
||||||
|
|
||||||
|
class Notification(Schema):
|
||||||
|
id: str
|
||||||
|
type: Literal[
|
||||||
|
"mention",
|
||||||
|
"status",
|
||||||
|
"reblog",
|
||||||
|
"follow",
|
||||||
|
"follow_request",
|
||||||
|
"favourite",
|
||||||
|
"poll",
|
||||||
|
"update",
|
||||||
|
"admin.sign_up",
|
||||||
|
"admin.report",
|
||||||
|
]
|
||||||
|
created_at: str
|
||||||
|
account: Account
|
||||||
|
status: Status | None
|
||||||
|
|
||||||
|
|
||||||
|
class Tag(Schema):
|
||||||
|
name: str
|
||||||
|
url: str
|
||||||
|
history: dict
|
||||||
|
|
||||||
|
|
||||||
|
class Search(Schema):
|
||||||
|
accounts: list[Account]
|
||||||
|
statuses: list[Status]
|
||||||
|
hashtags: list[Tag]
|
||||||
|
|
||||||
|
|
||||||
|
class Relationship(Schema):
|
||||||
|
id: str
|
||||||
|
following: bool
|
||||||
|
followed_by: bool
|
||||||
|
showing_reblogs: bool
|
||||||
|
notifying: bool
|
||||||
|
blocking: bool
|
||||||
|
blocked_by: bool
|
||||||
|
muting: bool
|
||||||
|
muting_notifications: bool
|
||||||
|
requested: bool
|
||||||
|
domain_blocking: bool
|
||||||
|
endorsed: bool
|
||||||
|
note: str
|
|
@ -1,6 +1,8 @@
|
||||||
from .accounts import * # noqa
|
from .accounts import * # noqa
|
||||||
from .apps import * # noqa
|
from .apps import * # noqa
|
||||||
from .base import api # noqa
|
from .base import api_router # noqa
|
||||||
from .instance import * # noqa
|
from .instance import * # noqa
|
||||||
|
from .notifications import * # noqa
|
||||||
from .oauth import * # noqa
|
from .oauth import * # noqa
|
||||||
|
from .search import * # noqa
|
||||||
from .timelines import * # noqa
|
from .timelines import * # noqa
|
||||||
|
|
|
@ -1,9 +1,94 @@
|
||||||
from .. import schemas
|
from django.shortcuts import get_object_or_404
|
||||||
|
|
||||||
|
from activities.models import Post
|
||||||
|
from api import schemas
|
||||||
|
from api.views.base import api_router
|
||||||
|
from users.models import Identity
|
||||||
|
|
||||||
from ..decorators import identity_required
|
from ..decorators import identity_required
|
||||||
from .base import api
|
|
||||||
|
|
||||||
|
|
||||||
@api.get("/v1/accounts/verify_credentials", response=schemas.Account)
|
@api_router.get("/v1/accounts/verify_credentials", response=schemas.Account)
|
||||||
@identity_required
|
@identity_required
|
||||||
def verify_credentials(request):
|
def verify_credentials(request):
|
||||||
return request.identity.to_mastodon_json()
|
return request.identity.to_mastodon_json()
|
||||||
|
|
||||||
|
|
||||||
|
@api_router.get("/v1/accounts/relationships", response=list[schemas.Relationship])
|
||||||
|
@identity_required
|
||||||
|
def account_relationships(request):
|
||||||
|
ids = request.GET.getlist("id[]")
|
||||||
|
result = []
|
||||||
|
for id in ids:
|
||||||
|
identity = get_object_or_404(Identity, pk=id)
|
||||||
|
result.append(
|
||||||
|
{
|
||||||
|
"id": identity.pk,
|
||||||
|
"following": identity.inbound_follows.filter(
|
||||||
|
source=request.identity
|
||||||
|
).exists(),
|
||||||
|
"followed_by": identity.outbound_follows.filter(
|
||||||
|
target=request.identity
|
||||||
|
).exists(),
|
||||||
|
"showing_reblogs": True,
|
||||||
|
"notifying": False,
|
||||||
|
"blocking": False,
|
||||||
|
"blocked_by": False,
|
||||||
|
"muting": False,
|
||||||
|
"muting_notifications": False,
|
||||||
|
"requested": False,
|
||||||
|
"domain_blocking": False,
|
||||||
|
"endorsed": False,
|
||||||
|
"note": "",
|
||||||
|
}
|
||||||
|
)
|
||||||
|
return result
|
||||||
|
|
||||||
|
|
||||||
|
@api_router.get("/v1/accounts/{id}", response=schemas.Account)
|
||||||
|
@identity_required
|
||||||
|
def account(request, id: str):
|
||||||
|
identity = get_object_or_404(Identity, pk=id)
|
||||||
|
return identity.to_mastodon_json()
|
||||||
|
|
||||||
|
|
||||||
|
@api_router.get("/v1/accounts/{id}/statuses", response=list[schemas.Status])
|
||||||
|
@identity_required
|
||||||
|
def account_statuses(
|
||||||
|
request,
|
||||||
|
id: str,
|
||||||
|
exclude_reblogs: bool = False,
|
||||||
|
exclude_replies: bool = False,
|
||||||
|
only_media: bool = False,
|
||||||
|
pinned: bool = False,
|
||||||
|
tagged: str | None = None,
|
||||||
|
max_id: str | None = None,
|
||||||
|
since_id: str | None = None,
|
||||||
|
min_id: str | None = None,
|
||||||
|
limit: int = 20,
|
||||||
|
):
|
||||||
|
identity = get_object_or_404(Identity, pk=id)
|
||||||
|
posts = (
|
||||||
|
identity.posts.public()
|
||||||
|
.select_related("author")
|
||||||
|
.prefetch_related("attachments")
|
||||||
|
.order_by("-created")
|
||||||
|
)
|
||||||
|
if pinned:
|
||||||
|
return []
|
||||||
|
if only_media:
|
||||||
|
posts = posts.filter(attachments__pk__isnull=False)
|
||||||
|
if tagged:
|
||||||
|
posts = posts.tagged_with(tagged)
|
||||||
|
if max_id:
|
||||||
|
anchor_post = Post.objects.get(pk=max_id)
|
||||||
|
posts = posts.filter(created__lt=anchor_post.created)
|
||||||
|
if since_id:
|
||||||
|
anchor_post = Post.objects.get(pk=since_id)
|
||||||
|
posts = posts.filter(created__gt=anchor_post.created)
|
||||||
|
if min_id:
|
||||||
|
# Min ID requires LIMIT posts _immediately_ newer than specified, so we
|
||||||
|
# invert the ordering to accomodate
|
||||||
|
anchor_post = Post.objects.get(pk=min_id)
|
||||||
|
posts = posts.filter(created__gt=anchor_post.created).order_by("created")
|
||||||
|
return [post.to_mastodon_json() for post in posts[:limit]]
|
||||||
|
|
|
@ -4,7 +4,7 @@ from ninja import Schema
|
||||||
|
|
||||||
from .. import schemas
|
from .. import schemas
|
||||||
from ..models import Application
|
from ..models import Application
|
||||||
from .base import api
|
from .base import api_router
|
||||||
|
|
||||||
|
|
||||||
class CreateApplicationSchema(Schema):
|
class CreateApplicationSchema(Schema):
|
||||||
|
@ -14,7 +14,7 @@ class CreateApplicationSchema(Schema):
|
||||||
website: None | str = None
|
website: None | str = None
|
||||||
|
|
||||||
|
|
||||||
@api.post("/v1/apps", response=schemas.Application)
|
@api_router.post("/v1/apps", response=schemas.Application)
|
||||||
def add_app(request, details: CreateApplicationSchema):
|
def add_app(request, details: CreateApplicationSchema):
|
||||||
client_id = "tk-" + secrets.token_urlsafe(16)
|
client_id = "tk-" + secrets.token_urlsafe(16)
|
||||||
client_secret = secrets.token_urlsafe(40)
|
client_secret = secrets.token_urlsafe(40)
|
||||||
|
|
|
@ -2,4 +2,4 @@ from ninja import NinjaAPI
|
||||||
|
|
||||||
from api.parser import FormOrJsonParser
|
from api.parser import FormOrJsonParser
|
||||||
|
|
||||||
api = NinjaAPI(parser=FormOrJsonParser())
|
api_router = NinjaAPI(parser=FormOrJsonParser())
|
||||||
|
|
|
@ -5,10 +5,10 @@ from core.models import Config
|
||||||
from takahe import __version__
|
from takahe import __version__
|
||||||
from users.models import Domain, Identity
|
from users.models import Domain, Identity
|
||||||
|
|
||||||
from .base import api
|
from .base import api_router
|
||||||
|
|
||||||
|
|
||||||
@api.get("/v1/instance")
|
@api_router.get("/v1/instance")
|
||||||
def instance_info(request):
|
def instance_info(request):
|
||||||
return {
|
return {
|
||||||
"uri": request.headers.get("host", settings.SETUP.MAIN_DOMAIN),
|
"uri": request.headers.get("host", settings.SETUP.MAIN_DOMAIN),
|
||||||
|
@ -16,7 +16,7 @@ def instance_info(request):
|
||||||
"short_description": "",
|
"short_description": "",
|
||||||
"description": "",
|
"description": "",
|
||||||
"email": "",
|
"email": "",
|
||||||
"version": __version__,
|
"version": f"takahe/{__version__}",
|
||||||
"urls": {},
|
"urls": {},
|
||||||
"stats": {
|
"stats": {
|
||||||
"user_count": Identity.objects.filter(local=True).count(),
|
"user_count": Identity.objects.filter(local=True).count(),
|
||||||
|
|
52
api/views/notifications.py
Normal file
52
api/views/notifications.py
Normal file
|
@ -0,0 +1,52 @@
|
||||||
|
from activities.models import TimelineEvent
|
||||||
|
|
||||||
|
from .. import schemas
|
||||||
|
from ..decorators import identity_required
|
||||||
|
from .base import api_router
|
||||||
|
|
||||||
|
|
||||||
|
@api_router.get("/v1/notifications", response=list[schemas.Notification])
|
||||||
|
@identity_required
|
||||||
|
def notifications(
|
||||||
|
request,
|
||||||
|
max_id: str | None = None,
|
||||||
|
since_id: str | None = None,
|
||||||
|
min_id: str | None = None,
|
||||||
|
limit: int = 20,
|
||||||
|
account_id: str | None = None,
|
||||||
|
):
|
||||||
|
if limit > 40:
|
||||||
|
limit = 40
|
||||||
|
# Types/exclude_types use weird syntax so we have to handle them manually
|
||||||
|
base_types = {
|
||||||
|
"favourite": TimelineEvent.Types.liked,
|
||||||
|
"reblog": TimelineEvent.Types.boosted,
|
||||||
|
"mention": TimelineEvent.Types.mentioned,
|
||||||
|
"follow": TimelineEvent.Types.followed,
|
||||||
|
}
|
||||||
|
requested_types = set(request.GET.getlist("types[]"))
|
||||||
|
excluded_types = set(request.GET.getlist("exclude_types[]"))
|
||||||
|
if not requested_types:
|
||||||
|
requested_types = set(base_types.keys())
|
||||||
|
requested_types.difference_update(excluded_types)
|
||||||
|
# Use that to pull relevant events
|
||||||
|
events = (
|
||||||
|
TimelineEvent.objects.filter(
|
||||||
|
identity=request.identity,
|
||||||
|
type__in=[base_types[r] for r in requested_types],
|
||||||
|
)
|
||||||
|
.order_by("-created")
|
||||||
|
.select_related("subject_post", "subject_post__author", "subject_identity")
|
||||||
|
)
|
||||||
|
if max_id:
|
||||||
|
anchor_event = TimelineEvent.objects.get(pk=max_id)
|
||||||
|
events = events.filter(created__lt=anchor_event.created)
|
||||||
|
if since_id:
|
||||||
|
anchor_event = TimelineEvent.objects.get(pk=since_id)
|
||||||
|
events = events.filter(created__gt=anchor_event.created)
|
||||||
|
if min_id:
|
||||||
|
# Min ID requires LIMIT events _immediately_ newer than specified, so we
|
||||||
|
# invert the ordering to accomodate
|
||||||
|
anchor_event = TimelineEvent.objects.get(pk=min_id)
|
||||||
|
events = events.filter(created__gt=anchor_event.created).order_by("created")
|
||||||
|
return [event.to_mastodon_notification_json() for event in events[:limit]]
|
42
api/views/search.py
Normal file
42
api/views/search.py
Normal file
|
@ -0,0 +1,42 @@
|
||||||
|
from typing import Literal
|
||||||
|
|
||||||
|
from ninja import Field
|
||||||
|
|
||||||
|
from activities.search import Searcher
|
||||||
|
from api import schemas
|
||||||
|
from api.decorators import identity_required
|
||||||
|
from api.views.base import api_router
|
||||||
|
|
||||||
|
|
||||||
|
@api_router.get("/v2/search", response=schemas.Search)
|
||||||
|
@identity_required
|
||||||
|
def search(
|
||||||
|
request,
|
||||||
|
q: str,
|
||||||
|
type: Literal["accounts", "hashtags", "statuses"] | None = None,
|
||||||
|
fetch_identities: bool = Field(False, alias="resolve"),
|
||||||
|
following: bool = False,
|
||||||
|
exclude_unreviewed: bool = False,
|
||||||
|
account_id: str | None = None,
|
||||||
|
max_id: str | None = None,
|
||||||
|
since_id: str | None = None,
|
||||||
|
min_id: str | None = None,
|
||||||
|
limit: int = 20,
|
||||||
|
offset: int = 0,
|
||||||
|
):
|
||||||
|
if limit > 40:
|
||||||
|
limit = 40
|
||||||
|
result: dict[str, list] = {"accounts": [], "statuses": [], "hashtags": []}
|
||||||
|
# We don't support pagination for searches yet
|
||||||
|
if max_id or since_id or min_id or offset:
|
||||||
|
return result
|
||||||
|
# Run search
|
||||||
|
searcher = Searcher(q, request.identity)
|
||||||
|
search_result = searcher.search_all()
|
||||||
|
if type is None or type == "accounts":
|
||||||
|
result["accounts"] = [i.to_mastodon_json() for i in search_result["identities"]]
|
||||||
|
if type is None or type == "hashtag":
|
||||||
|
result["hashtag"] = [h.to_mastodon_json() for h in search_result["hashtags"]]
|
||||||
|
if type is None or type == "statuses":
|
||||||
|
result["statuses"] = [p.to_mastodon_json() for p in search_result["posts"]]
|
||||||
|
return result
|
|
@ -1,16 +1,21 @@
|
||||||
from activities.models import TimelineEvent
|
from activities.models import Post, TimelineEvent
|
||||||
|
|
||||||
from .. import schemas
|
from .. import schemas
|
||||||
from ..decorators import identity_required
|
from ..decorators import identity_required
|
||||||
from .base import api
|
from .base import api_router
|
||||||
|
|
||||||
|
|
||||||
@api.get("/v1/timelines/home", response=list[schemas.Status])
|
@api_router.get("/v1/timelines/home", response=list[schemas.Status])
|
||||||
@identity_required
|
@identity_required
|
||||||
def home(request):
|
def home(
|
||||||
if request.GET.get("max_id"):
|
request,
|
||||||
return []
|
max_id: str | None = None,
|
||||||
limit = int(request.GET.get("limit", "20"))
|
since_id: str | None = None,
|
||||||
|
min_id: str | None = None,
|
||||||
|
limit: int = 20,
|
||||||
|
):
|
||||||
|
if limit > 40:
|
||||||
|
limit = 40
|
||||||
events = (
|
events = (
|
||||||
TimelineEvent.objects.filter(
|
TimelineEvent.objects.filter(
|
||||||
identity=request.identity,
|
identity=request.identity,
|
||||||
|
@ -18,6 +23,109 @@ def home(request):
|
||||||
)
|
)
|
||||||
.select_related("subject_post", "subject_post__author")
|
.select_related("subject_post", "subject_post__author")
|
||||||
.prefetch_related("subject_post__attachments")
|
.prefetch_related("subject_post__attachments")
|
||||||
.order_by("-created")[:limit]
|
.order_by("-created")
|
||||||
)
|
)
|
||||||
return [event.subject_post.to_mastodon_json() for event in events]
|
if max_id:
|
||||||
|
anchor_post = Post.objects.get(pk=max_id)
|
||||||
|
events = events.filter(created__lt=anchor_post.created)
|
||||||
|
if since_id:
|
||||||
|
anchor_post = Post.objects.get(pk=since_id)
|
||||||
|
events = events.filter(created__gt=anchor_post.created)
|
||||||
|
if min_id:
|
||||||
|
# Min ID requires LIMIT events _immediately_ newer than specified, so we
|
||||||
|
# invert the ordering to accomodate
|
||||||
|
anchor_post = Post.objects.get(pk=min_id)
|
||||||
|
events = events.filter(created__gt=anchor_post.created).order_by("created")
|
||||||
|
return [event.subject_post.to_mastodon_json() for event in events[:limit]]
|
||||||
|
|
||||||
|
|
||||||
|
@api_router.get("/v1/timelines/public", response=list[schemas.Status])
|
||||||
|
@identity_required
|
||||||
|
def public(
|
||||||
|
request,
|
||||||
|
local: bool = False,
|
||||||
|
remote: bool = False,
|
||||||
|
only_media: bool = False,
|
||||||
|
max_id: str | None = None,
|
||||||
|
since_id: str | None = None,
|
||||||
|
min_id: str | None = None,
|
||||||
|
limit: int = 20,
|
||||||
|
):
|
||||||
|
if limit > 40:
|
||||||
|
limit = 40
|
||||||
|
posts = (
|
||||||
|
Post.objects.public()
|
||||||
|
.select_related("author")
|
||||||
|
.prefetch_related("attachments")
|
||||||
|
.order_by("-created")
|
||||||
|
)
|
||||||
|
if local:
|
||||||
|
posts = posts.filter(local=True)
|
||||||
|
elif remote:
|
||||||
|
posts = posts.filter(local=False)
|
||||||
|
if only_media:
|
||||||
|
posts = posts.filter(attachments__id__isnull=True)
|
||||||
|
if max_id:
|
||||||
|
anchor_post = Post.objects.get(pk=max_id)
|
||||||
|
posts = posts.filter(created__lt=anchor_post.created)
|
||||||
|
if since_id:
|
||||||
|
anchor_post = Post.objects.get(pk=since_id)
|
||||||
|
posts = posts.filter(created__gt=anchor_post.created)
|
||||||
|
if min_id:
|
||||||
|
# Min ID requires LIMIT posts _immediately_ newer than specified, so we
|
||||||
|
# invert the ordering to accomodate
|
||||||
|
anchor_post = Post.objects.get(pk=min_id)
|
||||||
|
posts = posts.filter(created__gt=anchor_post.created).order_by("created")
|
||||||
|
return [post.to_mastodon_json() for post in posts[:limit]]
|
||||||
|
|
||||||
|
|
||||||
|
@api_router.get("/v1/timelines/tag/{hashtag}", response=list[schemas.Status])
|
||||||
|
@identity_required
|
||||||
|
def hashtag(
|
||||||
|
request,
|
||||||
|
hashtag: str,
|
||||||
|
local: bool = False,
|
||||||
|
only_media: bool = False,
|
||||||
|
max_id: str | None = None,
|
||||||
|
since_id: str | None = None,
|
||||||
|
min_id: str | None = None,
|
||||||
|
limit: int = 20,
|
||||||
|
):
|
||||||
|
if limit > 40:
|
||||||
|
limit = 40
|
||||||
|
posts = (
|
||||||
|
Post.objects.public()
|
||||||
|
.tagged_with(hashtag)
|
||||||
|
.select_related("author")
|
||||||
|
.prefetch_related("attachments")
|
||||||
|
.order_by("-created")
|
||||||
|
)
|
||||||
|
if local:
|
||||||
|
posts = posts.filter(local=True)
|
||||||
|
if only_media:
|
||||||
|
posts = posts.filter(attachments__id__isnull=True)
|
||||||
|
if max_id:
|
||||||
|
anchor_post = Post.objects.get(pk=max_id)
|
||||||
|
posts = posts.filter(created__lt=anchor_post.created)
|
||||||
|
if since_id:
|
||||||
|
anchor_post = Post.objects.get(pk=since_id)
|
||||||
|
posts = posts.filter(created__gt=anchor_post.created)
|
||||||
|
if min_id:
|
||||||
|
# Min ID requires LIMIT posts _immediately_ newer than specified, so we
|
||||||
|
# invert the ordering to accomodate
|
||||||
|
anchor_post = Post.objects.get(pk=min_id)
|
||||||
|
posts = posts.filter(created__gt=anchor_post.created).order_by("created")
|
||||||
|
return [post.to_mastodon_json() for post in posts[:limit]]
|
||||||
|
|
||||||
|
|
||||||
|
@api_router.get("/v1/conversations", response=list[schemas.Status])
|
||||||
|
@identity_required
|
||||||
|
def conversations(
|
||||||
|
request,
|
||||||
|
max_id: str | None = None,
|
||||||
|
since_id: str | None = None,
|
||||||
|
min_id: str | None = None,
|
||||||
|
limit: int = 20,
|
||||||
|
):
|
||||||
|
# We don't implement this yet
|
||||||
|
return []
|
||||||
|
|
|
@ -4,7 +4,7 @@ from django.urls import path, re_path
|
||||||
from django.views.static import serve
|
from django.views.static import serve
|
||||||
|
|
||||||
from activities.views import compose, explore, follows, posts, search, timelines
|
from activities.views import compose, explore, follows, posts, search, timelines
|
||||||
from api.views import api, oauth
|
from api.views import api_router, oauth
|
||||||
from core import views as core
|
from core import views as core
|
||||||
from mediaproxy import views as mediaproxy
|
from mediaproxy import views as mediaproxy
|
||||||
from stator import views as stator
|
from stator import views as stator
|
||||||
|
@ -203,7 +203,7 @@ urlpatterns = [
|
||||||
path("actor/inbox/", activitypub.Inbox.as_view()),
|
path("actor/inbox/", activitypub.Inbox.as_view()),
|
||||||
path("inbox/", activitypub.Inbox.as_view(), name="shared_inbox"),
|
path("inbox/", activitypub.Inbox.as_view(), name="shared_inbox"),
|
||||||
# API/Oauth
|
# API/Oauth
|
||||||
path("api/", api.urls),
|
path("api/", api_router.urls),
|
||||||
path("oauth/authorize", oauth.AuthorizationView.as_view()),
|
path("oauth/authorize", oauth.AuthorizationView.as_view()),
|
||||||
path("oauth/token", oauth.TokenView.as_view()),
|
path("oauth/token", oauth.TokenView.as_view()),
|
||||||
path("oauth/revoke_token", oauth.RevokeTokenView.as_view()),
|
path("oauth/revoke_token", oauth.RevokeTokenView.as_view()),
|
||||||
|
|
Loading…
Reference in a new issue