113 lines
2.7 KiB
Python
113 lines
2.7 KiB
Python
import asyncio
|
|
import logging
|
|
|
|
from uuid import uuid4
|
|
|
|
from . import misc
|
|
|
|
|
|
async def handle_relay(request, actor, data, software):
|
|
if data.objectid in request.app.cache.objects:
|
|
logging.verbose(f'already relayed {data.objectid}')
|
|
return
|
|
|
|
logging.verbose(f'Relaying post from {data.actorid}')
|
|
|
|
message = misc.Message.new_announce(
|
|
host = request.app.config.host,
|
|
object = data.objectid
|
|
)
|
|
|
|
logging.debug(f'>> relay: {message}')
|
|
|
|
inboxes = misc.distill_inboxes(actor, data.objectid)
|
|
futures = [misc.request(inbox, data=message) for inbox in inboxes]
|
|
|
|
asyncio.ensure_future(asyncio.gather(*futures))
|
|
request.app.cache.objects[data.objectid] = message.id
|
|
|
|
|
|
async def handle_forward(request, actor, data, software):
|
|
if data.id in request.app.cache.objects:
|
|
logging.verbose(f'already forwarded {data.id}')
|
|
return
|
|
|
|
message = misc.Message.new_announce(
|
|
host = request.app.config.host,
|
|
object = data
|
|
)
|
|
|
|
logging.verbose(f'Forwarding post from {actor.id}')
|
|
logging.debug(f'>> Relay {data}')
|
|
|
|
inboxes = misc.distill_inboxes(actor, data.id)
|
|
futures = [misc.request(inbox, data=message) for inbox in inboxes]
|
|
|
|
asyncio.ensure_future(asyncio.gather(*futures))
|
|
request.app.cache.objects[data.id] = message.id
|
|
|
|
|
|
async def handle_follow(request, actor, data, software):
|
|
if request.app.database.add_inbox(inbox, data.id):
|
|
request.app.database.set_followid(actor.id, data.id)
|
|
|
|
request.app.database.save()
|
|
|
|
await misc.request(
|
|
actor.shared_inbox,
|
|
misc.Message.new_response(
|
|
host = request.app.config.host,
|
|
actor = actor.id,
|
|
followid = data.id,
|
|
accept = True
|
|
)
|
|
)
|
|
|
|
# Are Akkoma and Pleroma the only two that expect a follow back?
|
|
# Ignoring only Mastodon for now
|
|
if software != 'mastodon':
|
|
misc.request(
|
|
actor.shared_inbox,
|
|
misc.Message.new_follow(
|
|
host = request.app.config.host,
|
|
actor = actor.id
|
|
)
|
|
)
|
|
|
|
|
|
async def handle_undo(request, actor, data, software):
|
|
## If the object is not a Follow, forward it
|
|
if data['object']['type'] != 'Follow':
|
|
return await handle_forward(request, actor, data, software)
|
|
|
|
if not request.app.database.del_inbox(actor.domain, data.id):
|
|
return
|
|
|
|
request.app.database.save()
|
|
|
|
message = misc.Message.new_unfollow(
|
|
host = request.app.config.host,
|
|
actor = actor.id,
|
|
follow = data
|
|
)
|
|
|
|
await misc.request(actor.shared_inbox, message)
|
|
|
|
|
|
processors = {
|
|
'Announce': handle_relay,
|
|
'Create': handle_relay,
|
|
'Delete': handle_forward,
|
|
'Follow': handle_follow,
|
|
'Undo': handle_undo,
|
|
'Update': handle_forward,
|
|
}
|
|
|
|
|
|
async def run_processor(request, actor, data, software):
|
|
if data.type not in processors:
|
|
return
|
|
|
|
logging.verbose(f'New "{data.type}" from actor: {actor.id}')
|
|
return await processors[data.type](request, actor, data, software)
|