134 lines
3.5 KiB
Python
134 lines
3.5 KiB
Python
import aiohttp
|
|
import aiohttp.web
|
|
import logging
|
|
import uuid
|
|
import urllib.parse
|
|
import simplejson as json
|
|
import re
|
|
import cgi
|
|
from Crypto.PublicKey import RSA
|
|
from .database import DATABASE
|
|
|
|
|
|
# generate actor keys if not present
|
|
if "actorKeys" not in DATABASE:
|
|
logging.info("No actor keys present, generating 4096-bit RSA keypair.")
|
|
|
|
privkey = RSA.generate(4096)
|
|
pubkey = privkey.publickey()
|
|
|
|
DATABASE["actorKeys"] = {
|
|
"publicKey": pubkey.exportKey('PEM'),
|
|
"privateKey": privkey.exportKey('PEM')
|
|
}
|
|
|
|
|
|
PRIVKEY = RSA.importKey(DATABASE["actorKeys"]["privateKey"])
|
|
PUBKEY = PRIVKEY.publickey()
|
|
|
|
|
|
from . import app
|
|
from .remote_actor import fetch_actor
|
|
|
|
|
|
|
|
async def actor(request):
|
|
data = {
|
|
"@context": "https://www.w3.org/ns/activitystreams",
|
|
"endpoints": {
|
|
"sharedInbox": "https://{}/inbox".format(request.host)
|
|
},
|
|
"followers": "https://{}/followers".format(request.host),
|
|
"inbox": "https://{}/inbox".format(request.host),
|
|
"name": "Viera",
|
|
"type": "Application",
|
|
"id": "https://{}/actor".format(request.host),
|
|
"publicKey": {
|
|
"id": "https://{}/actor#main-key".format(request.host),
|
|
"owner": "https://{}/actor".format(request.host),
|
|
"publicKeyPem": DATABASE["actorKeys"]["publicKey"]
|
|
},
|
|
"summary": "Viera, the bot",
|
|
"preferredUsername": "viera"
|
|
}
|
|
return aiohttp.web.json_response(data)
|
|
|
|
|
|
app.router.add_get('/actor', actor)
|
|
|
|
|
|
from .http_signatures import sign_headers
|
|
|
|
|
|
async def push_message_to_actor(actor, message, our_key_id):
|
|
url = urllib.parse.urlsplit(actor['inbox'])
|
|
|
|
# XXX: Digest
|
|
data = json.dumps(message)
|
|
headers = {
|
|
'(request-target)': 'post {}'.format(url.path),
|
|
'Content-Length': str(len(data)),
|
|
'Content-Type': 'application/activity+json',
|
|
'User-Agent': 'Viera'
|
|
}
|
|
headers['signature'] = sign_headers(headers, PRIVKEY, our_key_id)
|
|
|
|
async with aiohttp.ClientSession() as session:
|
|
async with session.post(actor['inbox'], data=data, headers=headers) as resp:
|
|
pass
|
|
|
|
|
|
tag_re = re.compile(r'(<!--.*?-->|<[^>]*>)')
|
|
def strip_html(data):
|
|
no_tags = tag_re.sub('', data)
|
|
return cgi.escape(no_tags)
|
|
|
|
|
|
from .authreqs import check_reqs
|
|
|
|
|
|
async def handle_create(actor, data, request):
|
|
content = strip_html(data['object']['content']).split()
|
|
check_reqs(content, actor)
|
|
|
|
|
|
async def handle_follow(actor, data, request):
|
|
message = {
|
|
"@context": "https://www.w3.org/ns/activitystreams",
|
|
"type": "Accept",
|
|
"to": [actor["id"]],
|
|
|
|
# this is wrong per litepub, but mastodon < 2.4 is not compliant with that profile.
|
|
"object": {
|
|
"type": "Follow",
|
|
"id": data["id"],
|
|
"object": "https://{}/actor".format(request.host),
|
|
"actor": actor["id"]
|
|
},
|
|
|
|
"id": "https://{}/activities/{}".format(request.host, uuid.uuid4()),
|
|
}
|
|
await push_message_to_actor(actor, message, 'https://{}/actor#main-key'.format(request.host))
|
|
|
|
|
|
processors = {
|
|
'Create': handle_create,
|
|
'Follow': handle_follow
|
|
}
|
|
|
|
|
|
async def inbox(request):
|
|
data = await request.json()
|
|
|
|
actor = await fetch_actor(data["actor"])
|
|
actor_uri = 'https://{}/actor'.format(request.host)
|
|
|
|
processor = processors.get(data['type'], None)
|
|
if processor:
|
|
await processor(actor, data, request)
|
|
|
|
return aiohttp.web.Response(body=b'{}', content_type='application/activity+json')
|
|
|
|
|
|
app.router.add_post('/inbox', inbox)
|