kopia lustrzana https://dev.funkwhale.audio/funkwhale/funkwhale
327 wiersze
12 KiB
Python
327 wiersze
12 KiB
Python
from django import forms
|
|
from django.core import paginator
|
|
from django.db.models import Prefetch
|
|
from django.http import HttpResponse
|
|
from django.urls import reverse
|
|
from rest_framework import exceptions, mixins, permissions, response, viewsets
|
|
from rest_framework.decorators import action
|
|
|
|
from funkwhale_api.common import preferences
|
|
from funkwhale_api.moderation import models as moderation_models
|
|
from funkwhale_api.music import models as music_models
|
|
from funkwhale_api.music import utils as music_utils
|
|
|
|
from . import activity, authentication, models, renderers, serializers, utils, webfinger
|
|
|
|
|
|
class AuthenticatedIfAllowListEnabled(permissions.BasePermission):
|
|
def has_permission(self, request, view):
|
|
allow_list_enabled = preferences.get("moderation__allow_list_enabled")
|
|
if not allow_list_enabled:
|
|
return True
|
|
return bool(request.actor)
|
|
|
|
|
|
class FederationMixin(object):
|
|
permission_classes = [AuthenticatedIfAllowListEnabled]
|
|
|
|
def dispatch(self, request, *args, **kwargs):
|
|
if not preferences.get("federation__enabled"):
|
|
return HttpResponse(status=405)
|
|
return super().dispatch(request, *args, **kwargs)
|
|
|
|
|
|
class SharedViewSet(FederationMixin, viewsets.GenericViewSet):
|
|
authentication_classes = [authentication.SignatureAuthentication]
|
|
renderer_classes = renderers.get_ap_renderers()
|
|
|
|
@action(methods=["post"], detail=False)
|
|
def inbox(self, request, *args, **kwargs):
|
|
if request.method.lower() == "post" and request.actor is None:
|
|
raise exceptions.AuthenticationFailed(
|
|
"You need a valid signature to send an activity"
|
|
)
|
|
if request.method.lower() == "post":
|
|
activity.receive(activity=request.data, on_behalf_of=request.actor)
|
|
return response.Response({}, status=200)
|
|
|
|
|
|
class ActorViewSet(FederationMixin, mixins.RetrieveModelMixin, viewsets.GenericViewSet):
|
|
lookup_field = "preferred_username"
|
|
authentication_classes = [authentication.SignatureAuthentication]
|
|
renderer_classes = renderers.get_ap_renderers()
|
|
queryset = models.Actor.objects.local().select_related("user")
|
|
serializer_class = serializers.ActorSerializer
|
|
|
|
@action(methods=["get", "post"], detail=True)
|
|
def inbox(self, request, *args, **kwargs):
|
|
inbox_actor = self.get_object()
|
|
if request.method.lower() == "post" and request.actor is None:
|
|
raise exceptions.AuthenticationFailed(
|
|
"You need a valid signature to send an activity"
|
|
)
|
|
if request.method.lower() == "post":
|
|
activity.receive(
|
|
activity=request.data,
|
|
on_behalf_of=request.actor,
|
|
inbox_actor=inbox_actor,
|
|
)
|
|
return response.Response({}, status=200)
|
|
|
|
@action(methods=["get", "post"], detail=True)
|
|
def outbox(self, request, *args, **kwargs):
|
|
actor = self.get_object()
|
|
channel = actor.channel
|
|
if channel:
|
|
return self.get_channel_outbox_response(request, channel)
|
|
return response.Response({}, status=200)
|
|
|
|
def get_channel_outbox_response(self, request, channel):
|
|
conf = {
|
|
"id": channel.actor.outbox_url,
|
|
"actor": channel.actor,
|
|
"items": channel.library.uploads.for_federation()
|
|
.order_by("-creation_date")
|
|
.prefetch_related("library__channel__actor", "track__artist"),
|
|
"item_serializer": serializers.ChannelCreateUploadSerializer,
|
|
}
|
|
page = request.GET.get("page")
|
|
if page is None:
|
|
serializer = serializers.ChannelOutboxSerializer(channel)
|
|
data = serializer.data
|
|
else:
|
|
try:
|
|
page_number = int(page)
|
|
except Exception:
|
|
return response.Response({"page": ["Invalid page number"]}, status=400)
|
|
conf["page_size"] = preferences.get("federation__collection_page_size")
|
|
p = paginator.Paginator(conf["items"], conf["page_size"])
|
|
try:
|
|
page = p.page(page_number)
|
|
conf["page"] = page
|
|
serializer = serializers.CollectionPageSerializer(conf)
|
|
data = serializer.data
|
|
except paginator.EmptyPage:
|
|
return response.Response(status=404)
|
|
|
|
return response.Response(data)
|
|
|
|
@action(methods=["get"], detail=True)
|
|
def followers(self, request, *args, **kwargs):
|
|
self.get_object()
|
|
# XXX to implement
|
|
return response.Response({})
|
|
|
|
@action(methods=["get"], detail=True)
|
|
def following(self, request, *args, **kwargs):
|
|
self.get_object()
|
|
# XXX to implement
|
|
return response.Response({})
|
|
|
|
|
|
class EditViewSet(FederationMixin, mixins.RetrieveModelMixin, viewsets.GenericViewSet):
|
|
lookup_field = "uuid"
|
|
authentication_classes = [authentication.SignatureAuthentication]
|
|
renderer_classes = renderers.get_ap_renderers()
|
|
# queryset = common_models.Mutation.objects.local().select_related()
|
|
# serializer_class = serializers.ActorSerializer
|
|
|
|
|
|
class ReportViewSet(
|
|
FederationMixin, mixins.RetrieveModelMixin, viewsets.GenericViewSet
|
|
):
|
|
lookup_field = "uuid"
|
|
authentication_classes = [authentication.SignatureAuthentication]
|
|
renderer_classes = renderers.get_ap_renderers()
|
|
queryset = moderation_models.Report.objects.none()
|
|
|
|
|
|
class WellKnownViewSet(viewsets.GenericViewSet):
|
|
authentication_classes = []
|
|
permission_classes = []
|
|
renderer_classes = [renderers.JSONRenderer, renderers.WebfingerRenderer]
|
|
|
|
@action(methods=["get"], detail=False)
|
|
def nodeinfo(self, request, *args, **kwargs):
|
|
data = {
|
|
"links": [
|
|
{
|
|
"rel": "http://nodeinfo.diaspora.software/ns/schema/2.0",
|
|
"href": utils.full_url(reverse("api:v1:instance:nodeinfo-2.0")),
|
|
}
|
|
]
|
|
}
|
|
return response.Response(data)
|
|
|
|
@action(methods=["get"], detail=False)
|
|
def webfinger(self, request, *args, **kwargs):
|
|
if not preferences.get("federation__enabled"):
|
|
return HttpResponse(status=405)
|
|
try:
|
|
resource_type, resource = webfinger.clean_resource(request.GET["resource"])
|
|
cleaner = getattr(webfinger, "clean_{}".format(resource_type))
|
|
result = cleaner(resource)
|
|
handler = getattr(self, "handler_{}".format(resource_type))
|
|
data = handler(result)
|
|
except forms.ValidationError as e:
|
|
return response.Response({"errors": {"resource": e.message}}, status=400)
|
|
except KeyError:
|
|
return response.Response(
|
|
{"errors": {"resource": "This field is required"}}, status=400
|
|
)
|
|
|
|
return response.Response(data)
|
|
|
|
def handler_acct(self, clean_result):
|
|
username, hostname = clean_result
|
|
|
|
try:
|
|
actor = models.Actor.objects.local().get(preferred_username=username)
|
|
except models.Actor.DoesNotExist:
|
|
raise forms.ValidationError("Invalid username")
|
|
|
|
return serializers.ActorWebfingerSerializer(actor).data
|
|
|
|
|
|
def has_library_access(request, library):
|
|
if library.privacy_level == "everyone":
|
|
return True
|
|
if request.user.is_authenticated and request.user.is_superuser:
|
|
return True
|
|
|
|
try:
|
|
actor = request.actor
|
|
except AttributeError:
|
|
return False
|
|
|
|
return library.received_follows.filter(actor=actor, approved=True).exists()
|
|
|
|
|
|
class MusicLibraryViewSet(
|
|
FederationMixin, mixins.RetrieveModelMixin, viewsets.GenericViewSet
|
|
):
|
|
authentication_classes = [authentication.SignatureAuthentication]
|
|
renderer_classes = renderers.get_ap_renderers()
|
|
serializer_class = serializers.LibrarySerializer
|
|
queryset = music_models.Library.objects.all().select_related("actor")
|
|
lookup_field = "uuid"
|
|
|
|
def retrieve(self, request, *args, **kwargs):
|
|
lb = self.get_object()
|
|
|
|
conf = {
|
|
"id": lb.get_federation_id(),
|
|
"actor": lb.actor,
|
|
"name": lb.name,
|
|
"summary": lb.description,
|
|
"items": lb.uploads.for_federation()
|
|
.order_by("-creation_date")
|
|
.prefetch_related(
|
|
Prefetch(
|
|
"track",
|
|
queryset=music_models.Track.objects.select_related(
|
|
"album__artist__attributed_to",
|
|
"artist__attributed_to",
|
|
"album__attributed_to",
|
|
"attributed_to",
|
|
"album__attachment_cover",
|
|
).prefetch_related(
|
|
"tagged_items__tag",
|
|
"album__tagged_items__tag",
|
|
"album__artist__tagged_items__tag",
|
|
"artist__tagged_items__tag",
|
|
),
|
|
)
|
|
),
|
|
"item_serializer": serializers.UploadSerializer,
|
|
}
|
|
page = request.GET.get("page")
|
|
if page is None:
|
|
serializer = serializers.LibrarySerializer(lb)
|
|
data = serializer.data
|
|
else:
|
|
# if actor is requesting a specific page, we ensure library is public
|
|
# or readable by the actor
|
|
if not has_library_access(request, lb):
|
|
raise exceptions.AuthenticationFailed(
|
|
"You do not have access to this library"
|
|
)
|
|
try:
|
|
page_number = int(page)
|
|
except Exception:
|
|
return response.Response({"page": ["Invalid page number"]}, status=400)
|
|
conf["page_size"] = preferences.get("federation__collection_page_size")
|
|
p = paginator.Paginator(conf["items"], conf["page_size"])
|
|
try:
|
|
page = p.page(page_number)
|
|
conf["page"] = page
|
|
serializer = serializers.CollectionPageSerializer(conf)
|
|
data = serializer.data
|
|
except paginator.EmptyPage:
|
|
return response.Response(status=404)
|
|
|
|
return response.Response(data)
|
|
|
|
@action(methods=["get"], detail=True)
|
|
def followers(self, request, *args, **kwargs):
|
|
self.get_object()
|
|
# XXX Implement this
|
|
return response.Response({})
|
|
|
|
|
|
class MusicUploadViewSet(
|
|
FederationMixin, mixins.RetrieveModelMixin, viewsets.GenericViewSet
|
|
):
|
|
authentication_classes = [authentication.SignatureAuthentication]
|
|
renderer_classes = renderers.get_ap_renderers()
|
|
queryset = music_models.Upload.objects.local().select_related(
|
|
"library__actor",
|
|
"track__artist",
|
|
"track__album__artist",
|
|
"track__album__attachment_cover",
|
|
)
|
|
serializer_class = serializers.UploadSerializer
|
|
lookup_field = "uuid"
|
|
|
|
def get_queryset(self):
|
|
queryset = super().get_queryset()
|
|
actor = music_utils.get_actor_from_request(self.request)
|
|
return queryset.playable_by(actor)
|
|
|
|
def get_serializer(self, obj):
|
|
if obj.library.get_channel():
|
|
return serializers.ChannelUploadSerializer(obj)
|
|
return super().get_serializer(obj)
|
|
|
|
|
|
class MusicArtistViewSet(
|
|
FederationMixin, mixins.RetrieveModelMixin, viewsets.GenericViewSet
|
|
):
|
|
authentication_classes = [authentication.SignatureAuthentication]
|
|
renderer_classes = renderers.get_ap_renderers()
|
|
queryset = music_models.Artist.objects.local()
|
|
serializer_class = serializers.ArtistSerializer
|
|
lookup_field = "uuid"
|
|
|
|
|
|
class MusicAlbumViewSet(
|
|
FederationMixin, mixins.RetrieveModelMixin, viewsets.GenericViewSet
|
|
):
|
|
authentication_classes = [authentication.SignatureAuthentication]
|
|
renderer_classes = renderers.get_ap_renderers()
|
|
queryset = music_models.Album.objects.local().select_related("artist")
|
|
serializer_class = serializers.AlbumSerializer
|
|
lookup_field = "uuid"
|
|
|
|
|
|
class MusicTrackViewSet(
|
|
FederationMixin, mixins.RetrieveModelMixin, viewsets.GenericViewSet
|
|
):
|
|
authentication_classes = [authentication.SignatureAuthentication]
|
|
renderer_classes = renderers.get_ap_renderers()
|
|
queryset = music_models.Track.objects.local().select_related(
|
|
"album__artist", "artist"
|
|
)
|
|
serializer_class = serializers.TrackSerializer
|
|
lookup_field = "uuid"
|