bitbot-3.11-fork/modules/fediverse/ap_actor.py

import email.utils, urllib.parse
from src import utils
from . import ap_security, ap_utils

class Actor(object):
    def __init__(self, url):
        self.url = url

        self.username = None
        self.inbox = None
        self.outbox = None

    def load(self):
        response = ap_utils.activity_request(self.url)
        if response.code == 200:
            self.username = response.data["preferredUsername"]
            self.inbox = Inbox(response.data["inbox"])
            self.outbox = Outbox(response.data["outbox"])
            return True
        return False

class Outbox(object):
    def __init__(self, url):
        self._url = url

    def load(self):
        outbox = ap_utils.activity_request(self._url)

        items = None
        if "first" in outbox.data:
            if type(outbox.data["first"]) == dict:
                # pleroma
                items = outbox.data["first"]["orderedItems"]
            else:
                # mastodon
                first = ap_utils.activity_request(outbox.data["first"])
                items = first["orderedItems"]
        else:
            items = outbox.data["orderedItems"]
        return items

class Inbox(object):
    def __init__(self, url):
        self._url = url
    def send(self, sender, activity, private_key):
        now = email.utils.formatdate(timeval=None, localtime=False, usegmt=True)
        parts = urllib.parse.urlparse(self._url)
        headers = [
            ["host", parts.netloc],
            ["date", now]
        ]
        sign_headers = headers[:]
        sign_headers.insert(0, ["(request-target)", "post %s" % parts.path])
        signature = ap_security.signature(private_key, sign_headers)

        headers.append(["signature", signature])

        return ap_utils.activity_request(self._url, activity.format(sender),
            method="POST", headers=dict(headers)).data