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 not request.app.database.add_inbox(actor.shared_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': await 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)