import uuid from django.contrib.postgres.fields import JSONField from django.core.serializers.json import DjangoJSONEncoder from django.db import models from django.urls import reverse from django.utils import timezone from django.db.models.signals import post_delete from django.dispatch import receiver from funkwhale_api.federation import keys from funkwhale_api.federation import models as federation_models from funkwhale_api.federation import utils as federation_utils from funkwhale_api.users import models as user_models def empty_dict(): return {} class Channel(models.Model): uuid = models.UUIDField(default=uuid.uuid4, unique=True) artist = models.OneToOneField( "music.Artist", on_delete=models.CASCADE, related_name="channel" ) # the owner of the channel attributed_to = models.ForeignKey( "federation.Actor", on_delete=models.CASCADE, related_name="owned_channels" ) # the federation actor created for the channel # (the one people can follow to receive updates) actor = models.OneToOneField( "federation.Actor", on_delete=models.CASCADE, related_name="channel" ) library = models.OneToOneField( "music.Library", on_delete=models.CASCADE, related_name="channel" ) creation_date = models.DateTimeField(default=timezone.now) rss_url = models.URLField(max_length=500, null=True, blank=True) # metadata to enhance rss feed metadata = JSONField( default=empty_dict, max_length=50000, encoder=DjangoJSONEncoder, blank=True ) def get_absolute_url(self): suffix = self.uuid if self.actor.is_local: suffix = self.actor.preferred_username else: suffix = self.actor.full_username return federation_utils.full_url("/channels/{}".format(suffix)) def get_rss_url(self): if not self.artist.is_local: return self.rss_url return federation_utils.full_url( reverse( "api:v1:channels-rss", kwargs={"composite": self.actor.preferred_username}, ) ) @property def fid(self): return self.actor.fid def generate_actor(username, **kwargs): actor_data = user_models.get_actor_data(username, **kwargs) private, public = keys.get_key_pair() actor_data["private_key"] = private.decode("utf-8") actor_data["public_key"] = public.decode("utf-8") return federation_models.Actor.objects.create(**actor_data) @receiver(post_delete, sender=Channel) def delete_channel_related_objs(instance, **kwargs): instance.library.delete() instance.actor.delete() instance.artist.delete()