import asyncio import logging from cachetools import LRUCache from uuid import uuid4 from .database import RELAY_SOFTWARE from .misc import Message cache = LRUCache(1024) def person_check(actor, software): ## pleroma and akkoma use Person for the actor type for some reason if software in {'akkoma', 'pleroma'} and actor.id != f'https://{actor.domain}/relay': return True ## make sure the actor is an application elif actor.type != 'Application': return True async def handle_relay(request, s): if request.message.objectid in cache: logging.verbose(f'already relayed {request.message.objectid}') return message = Message.new_announce( host = request.config.host, object = request.message.objectid ) cache[request.message.objectid] = message.id logging.debug(f'>> relay: {message}') inboxes = s.distill_inboxes(request.message) for inbox in inboxes: request.app.push_message(inbox, message) async def handle_forward(request, s): if request.message.id in cache: logging.verbose(f'already forwarded {request.message.id}') return message = Message.new_announce( host = request.config.host, object = request.message ) cache[request.message.id] = message.id logging.debug(f'>> forward: {message}') inboxes = s.distill_inboxes(request.message) for inbox in inboxes: request.app.push_message(inbox, message) async def handle_follow(request, s): approve = True nodeinfo = await request.app.client.fetch_nodeinfo(request.actor.domain) software = nodeinfo.sw_name if nodeinfo else None ## reject if the actor isn't whitelisted while the whiltelist is enabled if s.get_config('whitelist') and not s.get_whitelist(request.actor.domain): logging.verbose(f'Rejected actor for not being in the whitelist: {request.actor.id}') accept = False ## reject if software used by actor is banned if s.get_banned_software(software): logging.verbose(f'Rejected follow from actor for using specific software: actor={request.actor.id}, software={software}') accept = False ## reject if the actor is not an instance actor if person_check(request.actor, software): logging.verbose(f'Non-application actor tried to follow: {request.actor.id}') accept = False if approve: if not request.instance: s.put_instance( domain = request.actor.domain, actor = request.actor.id, inbox = request.actor.shared_inbox, actor_data = request.actor, software = software, followid = request.message.id, accept = s.get_config('require_approval') ) if s.get_config('require_approval'): return else: s.put_instance( domain = request.actor.domain, followid = request.message.id ) request.app.push_message( request.actor.shared_inbox, Message.new_response( host = request.config.host, actor = request.actor.id, followid = request.message.id, accept = approve ) ) ## Don't send a follow if the the follow has been rejected if not approve: return ## Make sure two relays aren't continuously following each other if software in RELAY_SOFTWARE and not request.instance: return # Are Akkoma and Pleroma the only two that expect a follow back? # Ignoring only Mastodon for now if software != 'mastodon': request.app.push_message( request.actor.shared_inbox, Message.new_follow( host = request.config.host, actor = request.actor.id ) ) async def handle_undo(request, s): ## If the object is not a Follow, forward it if request.message.object.type != 'Follow': return await handle_forward(request) s.delete('instances', id=request.instance.id) request.app.push_message( request.actor.shared_inbox, Message.new_unfollow( host = request.config.host, actor = request.actor.id, follow = request.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): if request.message.type not in processors: return with request.database.session as s: new_data = {} if request.instance and not request.instance.software: nodeinfo = await request.app.client.fetch_nodeinfo(request.instance.domain) if nodeinfo: new_data['software'] = nodeinfo.sw_name if not request.instance.actor: new_data['actor'] = request.signature.keyid.split('#', 1)[0] if not request.instance.actor_data: new_data['actor_data'] = request.actor if new_data: s.put_instance(request.actor.domain, **new_data) logging.verbose(f'New "{request.message.type}" from actor: {request.actor.id}') return await processors[request.message.type](request, s)