bookwyrm/bookwyrm/activitypub/verbs.py

''' undo wrapper activity '''
from dataclasses import dataclass
from typing import List

from .base_activity import ActivityObject, Signature, resolve_remote_id
from .book import Edition


@dataclass(init=False)
class Verb(ActivityObject):
    ''' generic fields for activities - maybe an unecessary level of
        abstraction but w/e '''
    actor: str
    object: ActivityObject

    def action(self):
        ''' usually we just want to save, this can be overridden as needed '''
        self.object.to_model()


@dataclass(init=False)
class Create(Verb):
    ''' Create activity '''
    to: List
    cc: List
    signature: Signature = None
    type: str = 'Create'


@dataclass(init=False)
class Delete(Verb):
    ''' Create activity '''
    to: List
    cc: List
    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
    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 '''
        obj = self.object.to_model(save=False, allow_create=False)
        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: str
    object: Edition
    type: str = 'Add'

    def action(self):
        ''' add obj to collection '''
        target = resolve_remote_id(self.target, refresh=False)
        # we want to related field that isn't the book, this is janky af sorry
        model = [t for t in type(target)._meta.related_objects \
                if t.name != 'edition'][0].related_model
        self.to_model(model=model)


@dataclass(init=False)
class AddBook(Add):
    '''Add activity that's aware of the book obj '''
    object: Edition


@dataclass(init=False)
class AddListItem(AddBook):
    '''Add activity that's aware of the book obj '''
    notes: str = None
    order: int = 0
    approved: bool = True


@dataclass(init=False)
class Remove(Verb):
    '''Remove activity '''
    target: ActivityObject
    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 '''
    object: str
    type: str = 'Announce'

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