bookwyrm-mastodon/bookwyrm/activitypub/verbs.py

""" activities that do things """
from dataclasses import dataclass, field
from typing import List
from django.apps import apps

from .base_activity import ActivityObject, Signature, resolve_remote_id
from .ordered_collection import CollectionItem


@dataclass(init=False)
class Verb(ActivityObject):
    """generic fields for activities """

    actor: str
    object: ActivityObject

    def action(self):
        """ usually we just want to update and save """
        self.object.to_model()


@dataclass(init=False)
class Create(Verb):
    """ Create activity """

    to: List[str]
    cc: List[str] = field(default_factory=lambda: [])
    signature: Signature = None
    type: str = "Create"


@dataclass(init=False)
class Delete(Verb):
    """ Create activity """

    to: List[str]
    cc: List[str] = field(default_factory=lambda: [])
    type: str = "Delete"

    def action(self):
        """ find and delete the activity object """
        obj = self.object.to_model(save=False, allow_create=False)
        obj.delete()


@dataclass(init=False)
class Update(Verb):
    """ Update activity """

    to: List[str]
    type: str = "Update"

    def action(self):
        """ update a model instance from the dataclass """
        self.object.to_model(allow_create=False)


@dataclass(init=False)
class Undo(Verb):
    """ Undo an activity """

    type: str = "Undo"

    def action(self):
        """ find and remove the activity object """
        if isinstance(self.object, str):
            # it may be that sometihng should be done with these, but idk what
            # this seems just to be coming from pleroma
            return

        # this is so hacky but it does make it work....
        # (because you Reject a request and Undo a follow
        model = None
        if self.object.type == "Follow":
            model = apps.get_model("bookwyrm.UserFollows")
            obj = self.object.to_model(model=model, save=False, allow_create=False)
            if not obj:
                # this could be a folloq request not a follow proper
                model = apps.get_model("bookwyrm.UserFollowRequest")
                obj = self.object.to_model(model=model, save=False, allow_create=False)
        else:
            obj = self.object.to_model(model=model, save=False, allow_create=False)
        if not obj:
            # if we don't have the object, we can't undo it. happens a lot with boosts
            return
        obj.delete()


@dataclass(init=False)
class Follow(Verb):
    """ Follow activity """

    object: str
    type: str = "Follow"

    def action(self):
        """ relationship save """
        self.to_model()


@dataclass(init=False)
class Block(Verb):
    """ Block activity """

    object: str
    type: str = "Block"

    def action(self):
        """ relationship save """
        self.to_model()


@dataclass(init=False)
class Accept(Verb):
    """ Accept activity """

    object: Follow
    type: str = "Accept"

    def action(self):
        """ find and remove the activity object """
        obj = self.object.to_model(save=False, allow_create=False)
        obj.accept()


@dataclass(init=False)
class Reject(Verb):
    """ Reject activity """

    object: Follow
    type: str = "Reject"

    def action(self):
        """ find and remove the activity object """
        obj = self.object.to_model(save=False, allow_create=False)
        obj.reject()


@dataclass(init=False)
class Add(Verb):
    """Add activity """

    target: ActivityObject
    object: CollectionItem
    type: str = "Add"

    def action(self):
        """ figure out the target to assign the item to a collection  """
        target = resolve_remote_id(self.target)
        item = self.object.to_model(save=False)
        setattr(item, item.collection_field, target)
        item.save()


@dataclass(init=False)
class Remove(Add):
    """Remove activity """

    type: str = "Remove"

    def action(self):
        """ find and remove the activity object """
        obj = self.object.to_model(save=False, allow_create=False)
        obj.delete()


@dataclass(init=False)
class Like(Verb):
    """ a user faving an object """

    object: str
    type: str = "Like"

    def action(self):
        """ like """
        self.to_model()


@dataclass(init=False)
class Announce(Verb):
    """ boosting a status """

    published: str
    to: List[str] = field(default_factory=lambda: [])
    cc: List[str] = field(default_factory=lambda: [])
    object: str
    type: str = "Announce"

    def action(self):
        """ boost """
        self.to_model()