takahe/api/views/notifications.py

101 lines
3 KiB
Python
Raw Normal View History

from django.http import HttpRequest
from django.shortcuts import get_object_or_404
from hatchway import ApiResponse, api_view
from activities.models import PostInteraction, TimelineEvent
2022-12-21 19:47:48 +00:00
from activities.services import TimelineService
2022-12-11 19:37:28 +00:00
from api import schemas
2023-02-19 18:37:02 +00:00
from api.decorators import scope_required
from api.pagination import MastodonPaginator, PaginatingApiResponse, PaginationResult
2022-12-11 18:22:06 +00:00
# Types/exclude_types use weird syntax so we have to handle them manually
NOTIFICATION_TYPES = {
"favourite": TimelineEvent.Types.liked,
"reblog": TimelineEvent.Types.boosted,
"mention": TimelineEvent.Types.mentioned,
"follow": TimelineEvent.Types.followed,
"admin.sign_up": TimelineEvent.Types.identity_created,
}
2022-12-11 18:22:06 +00:00
2023-02-19 18:37:02 +00:00
@scope_required("read:notifications")
@api_view.get
2022-12-11 18:22:06 +00:00
def notifications(
request: HttpRequest,
2022-12-11 18:22:06 +00:00
max_id: str | None = None,
since_id: str | None = None,
min_id: str | None = None,
limit: int = 20,
account_id: str | None = None,
) -> ApiResponse[list[schemas.Notification]]:
2022-12-11 18:22:06 +00:00
requested_types = set(request.GET.getlist("types[]"))
excluded_types = set(request.GET.getlist("exclude_types[]"))
if not requested_types:
requested_types = set(NOTIFICATION_TYPES.keys())
2022-12-11 18:22:06 +00:00
requested_types.difference_update(excluded_types)
# Use that to pull relevant events
2022-12-21 19:47:48 +00:00
queryset = TimelineService(request.identity).notifications(
[NOTIFICATION_TYPES[r] for r in requested_types if r in NOTIFICATION_TYPES]
2022-12-11 18:22:06 +00:00
)
2023-01-09 06:06:09 +00:00
paginator = MastodonPaginator()
pager: PaginationResult[TimelineEvent] = paginator.paginate(
2022-12-12 07:38:02 +00:00
queryset,
min_id=min_id,
max_id=max_id,
since_id=since_id,
limit=limit,
)
interactions = PostInteraction.get_event_interactions(
pager.results,
request.identity,
)
return PaginatingApiResponse(
[
schemas.Notification.from_timeline_event(event, interactions=interactions)
for event in pager.results
],
request=request,
include_params=["limit", "account_id"],
)
@scope_required("read:notifications")
@api_view.get
def get_notification(
request: HttpRequest,
id: str,
) -> schemas.Notification:
notification = get_object_or_404(
TimelineService(request.identity).notifications(
list(NOTIFICATION_TYPES.values())
),
id=id,
)
return schemas.Notification.from_timeline_event(notification)
@scope_required("write:notifications")
@api_view.post
def dismiss_notifications(request: HttpRequest) -> dict:
TimelineService(request.identity).notifications(
list(NOTIFICATION_TYPES.values())
).update(dismissed=True)
return {}
@scope_required("write:notifications")
@api_view.post
def dismiss_notification(request: HttpRequest, id: str) -> dict:
notification = get_object_or_404(
TimelineService(request.identity).notifications(
list(NOTIFICATION_TYPES.values())
),
id=id,
)
notification.dismissed = True
notification.save()
return {}