mirror of
https://github.com/bookwyrm-social/bookwyrm.git
synced 2024-11-29 04:51:11 +00:00
373 lines
12 KiB
Python
373 lines
12 KiB
Python
''' handles all the activity coming out of the server '''
|
|
from datetime import datetime
|
|
from urllib.parse import urlencode
|
|
|
|
from django.db import IntegrityError, transaction
|
|
from django.http import HttpResponseNotFound, JsonResponse
|
|
from django.views.decorators.csrf import csrf_exempt
|
|
import requests
|
|
|
|
from fedireads import activitypub, books_manager
|
|
from fedireads import models
|
|
from fedireads.broadcast import broadcast
|
|
from fedireads.status import create_review, create_status
|
|
from fedireads.status import create_quotation, create_comment
|
|
from fedireads.status import create_tag, create_notification, create_rating
|
|
from fedireads.remote_user import get_or_create_remote_user
|
|
|
|
|
|
@csrf_exempt
|
|
def outbox(request, username):
|
|
''' outbox for the requested user '''
|
|
if request.method != 'GET':
|
|
return HttpResponseNotFound()
|
|
|
|
try:
|
|
user = models.User.objects.get(localname=username)
|
|
except models.User.DoesNotExist:
|
|
return HttpResponseNotFound()
|
|
|
|
# paginated list of messages
|
|
if request.GET.get('page'):
|
|
limit = 20
|
|
min_id = request.GET.get('min_id')
|
|
max_id = request.GET.get('max_id')
|
|
|
|
# filters for use in the django queryset min/max
|
|
filters = {}
|
|
# params for the outbox page id
|
|
params = {'page': 'true'}
|
|
if min_id is not None:
|
|
params['min_id'] = min_id
|
|
filters['id__gt'] = min_id
|
|
if max_id is not None:
|
|
params['max_id'] = max_id
|
|
filters['id__lte'] = max_id
|
|
|
|
page_id = user.outbox + '?' + urlencode(params)
|
|
statuses = models.Status.objects.filter(
|
|
user=user,
|
|
**filters
|
|
).select_subclasses().all()[:limit]
|
|
|
|
return JsonResponse(
|
|
activitypub.get_outbox_page(user, page_id, statuses, max_id, min_id)
|
|
)
|
|
|
|
# collection overview
|
|
size = models.Status.objects.filter(user=user).count()
|
|
return JsonResponse(activitypub.get_outbox(user, size))
|
|
|
|
|
|
def handle_account_search(query):
|
|
''' webfingerin' other servers '''
|
|
user = None
|
|
domain = query.split('@')[1]
|
|
try:
|
|
user = models.User.objects.get(username=query)
|
|
except models.User.DoesNotExist:
|
|
url = 'https://%s/.well-known/webfinger?resource=acct:%s' % \
|
|
(domain, query)
|
|
response = requests.get(url)
|
|
if not response.ok:
|
|
response.raise_for_status()
|
|
data = response.json()
|
|
for link in data['links']:
|
|
if link['rel'] == 'self':
|
|
try:
|
|
user = get_or_create_remote_user(link['href'])
|
|
except KeyError:
|
|
return HttpResponseNotFound()
|
|
return user
|
|
|
|
|
|
def handle_follow(user, to_follow):
|
|
''' someone local wants to follow someone '''
|
|
activity = activitypub.get_follow_request(user, to_follow)
|
|
broadcast(user, activity, direct_recipients=[to_follow])
|
|
|
|
|
|
def handle_unfollow(user, to_unfollow):
|
|
''' someone local wants to follow someone '''
|
|
relationship = models.UserFollows.objects.get(
|
|
user_subject=user,
|
|
user_object=to_unfollow
|
|
)
|
|
activity = activitypub.get_unfollow(relationship)
|
|
broadcast(user, activity, direct_recipients=[to_unfollow])
|
|
to_unfollow.followers.remove(user)
|
|
|
|
|
|
def handle_accept(user, to_follow, follow_request):
|
|
''' send an acceptance message to a follow request '''
|
|
with transaction.atomic():
|
|
relationship = models.UserFollows.from_request(follow_request)
|
|
follow_request.delete()
|
|
relationship.save()
|
|
|
|
activity = activitypub.get_accept(to_follow, follow_request)
|
|
broadcast(to_follow, activity, privacy='direct', direct_recipients=[user])
|
|
|
|
|
|
def handle_reject(user, to_follow, relationship):
|
|
''' a local user who managed follows rejects a follow request '''
|
|
relationship.delete()
|
|
|
|
activity = activitypub.get_reject(to_follow, relationship)
|
|
broadcast(to_follow, activity, privacy='direct', direct_recipients=[user])
|
|
|
|
|
|
def handle_shelve(user, book, shelf):
|
|
''' a local user is getting a book put on their shelf '''
|
|
# update the database
|
|
models.ShelfBook(book=book, shelf=shelf, added_by=user).save()
|
|
|
|
activity = activitypub.get_add(user, book, shelf)
|
|
broadcast(user, activity)
|
|
|
|
# tell the world about this cool thing that happened
|
|
verb = {
|
|
'to-read': 'wants to read',
|
|
'reading': 'started reading',
|
|
'read': 'finished reading'
|
|
}[shelf.identifier]
|
|
message = '%s "%s"' % (verb, book.title)
|
|
status = create_status(user, message, mention_books=[book])
|
|
status.status_type = 'Update'
|
|
status.save()
|
|
|
|
if shelf.identifier == 'reading':
|
|
read = models.ReadThrough(
|
|
user=user,
|
|
book=book,
|
|
start_date=datetime.now())
|
|
read.save()
|
|
elif shelf.identifier == 'read':
|
|
read = models.ReadThrough.objects.filter(
|
|
user=user,
|
|
book=book,
|
|
finish_date=None).order_by('-created_date').first()
|
|
if not read:
|
|
read = models.ReadThrough(
|
|
user=user,
|
|
book=book,
|
|
start_date=datetime.now())
|
|
read.finish_date = datetime.now()
|
|
read.save()
|
|
|
|
activity = activitypub.get_status(status)
|
|
create_activity = activitypub.get_create(user, activity)
|
|
|
|
broadcast(user, create_activity)
|
|
|
|
|
|
def handle_unshelve(user, book, shelf):
|
|
''' a local user is getting a book put on their shelf '''
|
|
# update the database
|
|
row = models.ShelfBook.objects.get(book=book, shelf=shelf)
|
|
row.delete()
|
|
|
|
activity = activitypub.get_remove(user, book, shelf)
|
|
|
|
broadcast(user, activity)
|
|
|
|
|
|
def handle_import_books(user, items):
|
|
''' process a goodreads csv and then post about it '''
|
|
new_books = []
|
|
for item in items:
|
|
if item.shelf:
|
|
desired_shelf = models.Shelf.objects.get(
|
|
identifier=item.shelf,
|
|
user=user
|
|
)
|
|
if isinstance(item.book, models.Work):
|
|
item.book = item.book.default_edition
|
|
if not item.book:
|
|
continue
|
|
_, created = models.ShelfBook.objects.get_or_create(
|
|
book=item.book, shelf=desired_shelf, added_by=user)
|
|
if created:
|
|
new_books.append(item.book)
|
|
activity = activitypub.get_add(user, item.book, desired_shelf)
|
|
broadcast(user, activity)
|
|
|
|
if item.rating or item.review:
|
|
review_title = "Review of {!r} on Goodreads".format(
|
|
item.book.title,
|
|
) if item.review else ""
|
|
handle_review(
|
|
user,
|
|
item.book,
|
|
review_title,
|
|
item.review,
|
|
item.rating,
|
|
)
|
|
for read in item.reads:
|
|
read.book = item.book
|
|
read.user = user
|
|
read.save()
|
|
|
|
if new_books:
|
|
message = 'imported {} books'.format(len(new_books))
|
|
status = create_status(user, message, mention_books=new_books)
|
|
status.status_type = 'Update'
|
|
status.save()
|
|
|
|
create_activity = activitypub.get_create(
|
|
user, activitypub.get_status(status))
|
|
broadcast(user, create_activity)
|
|
return status
|
|
|
|
|
|
def handle_rate(user, book, rating):
|
|
''' a review that's just a rating '''
|
|
builder = create_rating
|
|
fr_serializer = activitypub.get_rating
|
|
ap_serializer = activitypub.get_rating_note
|
|
|
|
handle_status(user, book, builder, fr_serializer, ap_serializer, rating)
|
|
|
|
|
|
def handle_review(user, book, name, content, rating):
|
|
''' post a review '''
|
|
# validated and saves the review in the database so it has an id
|
|
builder = create_review
|
|
fr_serializer = activitypub.get_review
|
|
ap_serializer = activitypub.get_review_article
|
|
handle_status(
|
|
user, book, builder, fr_serializer,
|
|
ap_serializer, name, content, rating)
|
|
|
|
|
|
def handle_quotation(user, book, content, quote):
|
|
''' post a review '''
|
|
# validated and saves the review in the database so it has an id
|
|
builder = create_quotation
|
|
fr_serializer = activitypub.get_quotation
|
|
ap_serializer = activitypub.get_quotation_article
|
|
handle_status(
|
|
user, book, builder, fr_serializer, ap_serializer, content, quote)
|
|
|
|
|
|
def handle_comment(user, book, content):
|
|
''' post a comment '''
|
|
# validated and saves the review in the database so it has an id
|
|
builder = create_comment
|
|
fr_serializer = activitypub.get_comment
|
|
ap_serializer = activitypub.get_comment_article
|
|
handle_status(
|
|
user, book, builder, fr_serializer, ap_serializer, content)
|
|
|
|
|
|
def handle_status(user, book_id, \
|
|
builder, fr_serializer, ap_serializer, *args):
|
|
''' generic handler for statuses '''
|
|
book = books_manager.get_or_create_book(book_id)
|
|
status = builder(user, book, *args)
|
|
|
|
activity = fr_serializer(status)
|
|
create_activity = activitypub.get_create(user, activity)
|
|
broadcast(user, create_activity, software='fedireads')
|
|
|
|
# re-format the activity for non-fedireads servers
|
|
remote_activity = ap_serializer(status)
|
|
remote_create_activity = activitypub.get_create(user, remote_activity)
|
|
|
|
broadcast(user, remote_create_activity, software='other')
|
|
|
|
|
|
def handle_tag(user, book, name):
|
|
''' tag a book '''
|
|
tag = create_tag(user, book, name)
|
|
tag_activity = activitypub.get_add_tag(tag)
|
|
|
|
broadcast(user, tag_activity)
|
|
|
|
|
|
def handle_untag(user, book, name):
|
|
''' tag a book '''
|
|
book = models.Book.objects.get(id=book)
|
|
tag = models.Tag.objects.get(name=name, book=book, user=user)
|
|
tag_activity = activitypub.get_remove_tag(tag)
|
|
tag.delete()
|
|
|
|
broadcast(user, tag_activity)
|
|
|
|
|
|
def handle_reply(user, review, content):
|
|
''' respond to a review or status '''
|
|
# validated and saves the comment in the database so it has an id
|
|
reply = create_status(user, content, reply_parent=review)
|
|
if reply.reply_parent:
|
|
create_notification(
|
|
reply.reply_parent.user,
|
|
'REPLY',
|
|
related_user=user,
|
|
related_status=reply,
|
|
)
|
|
reply_activity = activitypub.get_status(reply)
|
|
create_activity = activitypub.get_create(user, reply_activity)
|
|
|
|
broadcast(user, create_activity)
|
|
|
|
|
|
def handle_favorite(user, status):
|
|
''' a user likes a status '''
|
|
try:
|
|
favorite = models.Favorite.objects.create(
|
|
status=status,
|
|
user=user
|
|
)
|
|
except IntegrityError:
|
|
# you already fav'ed that
|
|
return
|
|
|
|
fav_activity = activitypub.get_favorite(favorite)
|
|
broadcast(
|
|
user, fav_activity, privacy='direct', direct_recipients=[status.user])
|
|
|
|
|
|
def handle_unfavorite(user, status):
|
|
''' a user likes a status '''
|
|
try:
|
|
favorite = models.Favorite.objects.get(
|
|
status=status,
|
|
user=user
|
|
)
|
|
except models.Favorite.DoesNotExist:
|
|
# can't find that status, idk
|
|
return
|
|
|
|
fav_activity = activitypub.get_unfavorite(favorite)
|
|
broadcast(user, fav_activity, direct_recipients=[status.user])
|
|
|
|
|
|
def handle_boost(user, status):
|
|
''' a user wishes to boost a status '''
|
|
if models.Boost.objects.filter(
|
|
boosted_status=status, user=user).exists():
|
|
# you already boosted that.
|
|
return
|
|
boost = models.Boost.objects.create(
|
|
boosted_status=status,
|
|
user=user,
|
|
)
|
|
boost.save()
|
|
|
|
boost_activity = activitypub.get_boost(boost)
|
|
broadcast(user, boost_activity)
|
|
|
|
|
|
def handle_update_book(user, book):
|
|
''' broadcast the news about our book '''
|
|
book_activity = activitypub.get_book(book)
|
|
update_activity = activitypub.get_update(user, book_activity)
|
|
broadcast(user, update_activity)
|
|
|
|
|
|
def handle_update_user(user):
|
|
''' broadcast editing a user's profile '''
|
|
actor = activitypub.get_actor(user)
|
|
update_activity = activitypub.get_update(user, actor)
|
|
broadcast(user, update_activity)
|