bookwyrm/fedireads/activitypub/status.py

''' status serializers '''
from uuid import uuid4

from fedireads.settings import DOMAIN


def get_review(review):
    ''' fedireads json for book reviews '''
    status = get_status(review)
    status['inReplyToBook'] = review.book.absolute_id
    status['fedireadsType'] = review.status_type
    status['name'] = review.name
    status['rating'] = review.rating
    return status


def get_comment(comment):
    ''' fedireads json for book reviews '''
    status = get_status(comment)
    status['inReplyToBook'] = comment.book.absolute_id
    status['fedireadsType'] = comment.status_type
    return status


def get_review_article(review):
    ''' a book review formatted for a non-fedireads isntance (mastodon) '''
    status = get_status(review)
    if review.rating:
        name = 'Review of "%s" (%d stars): %s' % (
            review.book.title,
            review.rating,
            review.name
        )
    else:
        name = 'Review of "%s": %s' % (
            review.book.title,
            review.name
        )

    status['name'] = name
    return status


def get_comment_article(comment):
    ''' a book comment formatted for a non-fedireads isntance (mastodon) '''
    status = get_status(comment)
    status['content'] += '<br><br>(comment on <a href="%s">"%s"</a>)' % \
        (comment.book.absolute_id, comment.book.title)
    return status


def get_status(status):
    ''' create activitypub json for a status '''
    user = status.user
    uri = status.absolute_id
    reply_parent_id = status.reply_parent.absolute_id \
        if status.reply_parent else None

    image_attachments = []
    books = list(status.mention_books.all()[:3])
    if hasattr(status, 'book'):
        books.append(status.book)
    for book in books:
        if book and book.cover:
            image_path = book.cover.url
            image_type = image_path.split('.')[-1]
            image_attachments.append({
                'type': 'Document',
                'mediaType': 'image/%s' % image_type,
                'url': 'https://%s%s' % (DOMAIN, image_path),
                'name': 'Cover of "%s"' % book.title,
            })
    status_json = {
        'id': uri,
        'url': uri,
        'inReplyTo': reply_parent_id,
        'published': status.published_date.isoformat(),
        'attributedTo': user.actor,
        # TODO: assuming all posts are public -- should check privacy db field
        'to': ['https://www.w3.org/ns/activitystreams#Public'],
        'cc': ['%s/followers' % user.absolute_id],
        'sensitive': status.sensitive,
        'content': status.content,
        'type': status.activity_type,
        'attachment': image_attachments,
        'replies': {
            'id': '%s/replies' % uri,
            'type': 'Collection',
            'first': {
                'type': 'CollectionPage',
                'next': '%s/replies?only_other_accounts=true&page=true' % uri,
                'partOf': '%s/replies' % uri,
                'items': [], # TODO: populate with replies
            }
        }
    }

    return status_json


def get_replies(status, replies):
    ''' collection of replies '''
    id_slug = status.absolute_id + '/replies'
    return {
        '@context': 'https://www.w3.org/ns/activitystreams',
        'id': id_slug,
        'type': 'Collection',
        'first': {
            'id': '%s?page=true' % id_slug,
            'type': 'CollectionPage',
            'next': '%s?only_other_accounts=true&page=true' % id_slug,
            'partOf': id_slug,
            'items': [get_status(r) for r in replies],
        }
    }


def get_replies_page(status, replies):
    ''' actual reply list content '''
    id_slug = status.absolute_id + '/replies?page=true&only_other_accounts=true'
    items = []
    for reply in replies:
        if reply.user.local:
            items.append(get_status(reply))
        else:
            items.append(reply.remote_id)
    return {
        '@context': 'https://www.w3.org/ns/activitystreams',
        'id': id_slug,
        'type': 'CollectionPage',
        'next': '%s&min_id=%d' % (id_slug, replies[len(replies) - 1].id),
        'partOf': status.absolute_id + '/replies',
        'items': [items]
    }


def get_favorite(favorite):
    ''' like a post '''
    return {
        '@context': 'https://www.w3.org/ns/activitystreams',
        'id': favorite.absolute_id,
        'type': 'Like',
        'actor': favorite.user.actor,
        'object': favorite.status.absolute_id,
    }


def get_unfavorite(favorite):
    ''' like a post '''
    return {
        '@context': 'https://www.w3.org/ns/activitystreams',
        'id': '%s/undo' % favorite.absolute_id,
        'type': 'Undo',
        'actor': favorite.user.actor,
        'object': {
            'id': favorite.absolute_id,
            'type': 'Like',
            'actor': favorite.user.actor,
            'object': favorite.status.absolute_id,
        }
    }


def get_boost(boost):
    ''' boost/announce a post '''
    return {
        '@context': 'https://www.w3.org/ns/activitystreams',
        'id': boost.absolute_id,
        'type': 'Announce',
        'actor': boost.user.actor,
        'object': boost.boosted_status.absolute_id,
    }


def get_add_tag(tag):
    ''' add activity for tagging a book '''
    uuid = uuid4()
    return {
        '@context': 'https://www.w3.org/ns/activitystreams',
        'id': str(uuid),
        'type': 'Add',
        'actor': tag.user.actor,
        'object': {
            'type': 'Tag',
            'id': tag.absolute_id,
            'name': tag.name,
        },
        'target': {
            'type': 'Book',
            'id': tag.book.absolute_id,
        }
    }


def get_remove_tag(tag):
    ''' add activity for tagging a book '''
    uuid = uuid4()
    return {
        '@context': 'https://www.w3.org/ns/activitystreams',
        'id': str(uuid),
        'type': 'Remove',
        'actor': tag.user.actor,
        'object': {
            'type': 'Tag',
            'id': tag.absolute_id,
            'name': tag.name,
        },
        'target': {
            'type': 'Book',
            'id': tag.book.absolute_id,
        }
    }