mirror of
https://github.com/jointakahe/takahe.git
synced 2024-12-01 11:41:00 +00:00
130 lines
4.4 KiB
Python
130 lines
4.4 KiB
Python
from asgiref.sync import sync_to_async
|
|
from django.db import models
|
|
|
|
from activities.models.timeline_event import TimelineEvent
|
|
from core.ld import canonicalise
|
|
from core.signatures import HttpSignature
|
|
from stator.models import State, StateField, StateGraph, StatorModel
|
|
|
|
|
|
class FanOutStates(StateGraph):
|
|
new = State(try_interval=300)
|
|
sent = State()
|
|
|
|
new.transitions_to(sent)
|
|
|
|
@classmethod
|
|
async def handle_new(cls, instance: "FanOut"):
|
|
"""
|
|
Sends the fan-out to the right inbox.
|
|
"""
|
|
fan_out = await instance.afetch_full()
|
|
# Handle Posts
|
|
if fan_out.type == FanOut.Types.post:
|
|
if fan_out.identity.local:
|
|
# Make a timeline event directly
|
|
await sync_to_async(TimelineEvent.add_post)(
|
|
identity=fan_out.identity,
|
|
post=fan_out.subject_post,
|
|
)
|
|
else:
|
|
# Send it to the remote inbox
|
|
post = await fan_out.subject_post.afetch_full()
|
|
# Sign it and send it
|
|
await HttpSignature.signed_request(
|
|
uri=fan_out.identity.inbox_uri,
|
|
body=canonicalise(post.to_create_ap()),
|
|
private_key=post.author.private_key,
|
|
key_id=post.author.public_key_id,
|
|
)
|
|
# Handle boosts/likes
|
|
elif fan_out.type == FanOut.Types.interaction:
|
|
interaction = await fan_out.subject_post_interaction.afetch_full()
|
|
if fan_out.identity.local:
|
|
# Make a timeline event directly
|
|
await sync_to_async(TimelineEvent.add_post_interaction)(
|
|
identity=fan_out.identity,
|
|
interaction=interaction,
|
|
)
|
|
else:
|
|
# Send it to the remote inbox
|
|
await HttpSignature.signed_request(
|
|
uri=fan_out.identity.inbox_uri,
|
|
body=canonicalise(interaction.to_ap()),
|
|
private_key=interaction.identity.private_key,
|
|
key_id=interaction.identity.public_key_id,
|
|
)
|
|
# Handle undoing boosts/likes
|
|
elif fan_out.type == FanOut.Types.undo_interaction:
|
|
interaction = await fan_out.subject_post_interaction.afetch_full()
|
|
if fan_out.identity.local:
|
|
# Delete any local timeline events
|
|
await sync_to_async(TimelineEvent.delete_post_interaction)(
|
|
identity=fan_out.identity,
|
|
interaction=interaction,
|
|
)
|
|
else:
|
|
# Send an undo to the remote inbox
|
|
await HttpSignature.signed_request(
|
|
uri=fan_out.identity.inbox_uri,
|
|
body=canonicalise(interaction.to_undo_ap()),
|
|
private_key=interaction.identity.private_key,
|
|
key_id=interaction.identity.public_key_id,
|
|
)
|
|
else:
|
|
raise ValueError(f"Cannot fan out with type {fan_out.type}")
|
|
return cls.sent
|
|
|
|
|
|
class FanOut(StatorModel):
|
|
"""
|
|
An activity that needs to get to an inbox somewhere.
|
|
"""
|
|
|
|
class Types(models.TextChoices):
|
|
post = "post"
|
|
interaction = "interaction"
|
|
undo_interaction = "undo_interaction"
|
|
|
|
state = StateField(FanOutStates)
|
|
|
|
# The user this event is targeted at
|
|
identity = models.ForeignKey(
|
|
"users.Identity",
|
|
on_delete=models.CASCADE,
|
|
related_name="fan_outs",
|
|
)
|
|
|
|
# What type of activity it is
|
|
type = models.CharField(max_length=100, choices=Types.choices)
|
|
|
|
# Links to the appropriate objects
|
|
subject_post = models.ForeignKey(
|
|
"activities.Post",
|
|
on_delete=models.CASCADE,
|
|
blank=True,
|
|
null=True,
|
|
related_name="fan_outs",
|
|
)
|
|
subject_post_interaction = models.ForeignKey(
|
|
"activities.PostInteraction",
|
|
on_delete=models.CASCADE,
|
|
blank=True,
|
|
null=True,
|
|
related_name="fan_outs",
|
|
)
|
|
|
|
created = models.DateTimeField(auto_now_add=True)
|
|
updated = models.DateTimeField(auto_now=True)
|
|
|
|
### Async helpers ###
|
|
|
|
async def afetch_full(self):
|
|
"""
|
|
Returns a version of the object with all relations pre-loaded
|
|
"""
|
|
return await FanOut.objects.select_related(
|
|
"identity",
|
|
"subject_post",
|
|
"subject_post_interaction",
|
|
).aget(pk=self.pk)
|