sedi-relay/relay/views.py

import asyncio
import logging
import subprocess
import traceback

from pathlib import Path

from . import __version__, misc
from .http_debug import STATS
from .misc import DotDict, Message, Response, WKNodeinfo
from .processors import run_processor


routes = []
version = __version__


if Path(__file__).parent.parent.joinpath('.git').exists():
	try:
		commit_label = subprocess.check_output(["git", "rev-parse", "HEAD"]).strip().decode('ascii')
		version = f'{__version__} {commit_label}'

	except:
		pass


def register_route(method, path):
	def wrapper(func):
		routes.append([method, path, func])
		return func

	return wrapper


@register_route('GET', '/')
async def home(request):
	targets = '<br>'.join(request.database.hostnames)
	note = request.config.note
	count = len(request.database.hostnames)
	host = request.config.host

	text = f"""
<html><head>
<title>ActivityPub Relay at {host}</title>
<style>
p {{ color: #FFFFFF; font-family: monospace, arial; font-size: 100%; }}
body {{ background-color: #000000; }}
a {{ color: #26F; }}
a:visited {{ color: #46C; }}
a:hover {{ color: #8AF; }}
</style>
</head>
<body>
<p>This is an Activity Relay for fediverse instances.</p>
<p>{note}</p>
<p>You may subscribe to this relay with the address: <a href="https://{host}/actor">https://{host}/actor</a></p>
<p>To host your own relay, you may download the code at this address: <a href="https://git.pleroma.social/pleroma/relay">https://git.pleroma.social/pleroma/relay</a></p>
<br><p>List of {count} registered instances:<br>{targets}</p>
</body></html>"""

	return Response.new(text, ctype='html')


@register_route('GET', '/inbox')
@register_route('GET', '/actor')
async def actor(request):
	data = Message.new_actor(
		host = request.config.host, 
		pubkey = request.database.pubkey
	)

	return Response.new(data, ctype='activity')


@register_route('POST', '/inbox')
@register_route('POST', '/actor')
async def inbox(request):
	config = request.config
	database = request.database

	## reject if missing signature header
	if not request.signature:
		logging.verbose('Actor missing signature header')
		raise HTTPUnauthorized(body='missing signature')

	try:
		request['message'] = await request.json(loads=Message.new_from_json)

		## reject if there is no message
		if not request.message:
			logging.verbose('empty message')
			return Response.new_error(400, 'missing message', 'json')

		## reject if there is no actor in the message
		if 'actor' not in request.message:
			logging.verbose('actor not in message')
			return Response.new_error(400, 'no actor in message', 'json')

	except:
		## this code should hopefully never get called
		traceback.print_exc()
		logging.verbose('Failed to parse inbox message')
		return Response.new_error(400, 'failed to parse message', 'json')

	request['actor'] = await misc.request(request.signature.keyid)

	## reject if actor is empty
	if not request.actor:
		## ld signatures aren't handled atm, so just ignore it
		if data.type == 'Delete':
			logging.verbose(f'Instance sent a delete which cannot be handled')
			return Response.new(status=202)

		logging.verbose(f'Failed to fetch actor: {request.signature.keyid}')
		return Response.new_error(400, 'failed to fetch actor', 'json')

	request['instance'] = request.database.get_inbox(request['actor'].inbox)

	## reject if the actor isn't whitelisted while the whiltelist is enabled
	if config.whitelist_enabled and not config.is_whitelisted(request.actor.domain):
		logging.verbose(f'Rejected actor for not being in the whitelist: {request.actor.id}')
		return Response.new_error(403, 'access denied', 'json')

	## reject if actor is banned
	if request.config.is_banned(request.actor.domain):
		logging.verbose(f'Ignored request from banned actor: {actor.id}')
		return Response.new_error(403, 'access denied', 'json')

	## reject if the signature is invalid
	if not (await misc.validate_signature(request.actor, request.signature, request)):
		logging.verbose(f'signature validation failed for: {actor.id}')
		return Response.new_error(401, 'signature check failed', 'json')

	## reject if activity type isn't 'Follow' and the actor isn't following
	if request.message.type != 'Follow' and not database.get_inbox(request.actor.domain):
		logging.verbose(f'Rejected actor for trying to post while not following: {request.actor.id}')
		return Response.new_error(401, 'access denied', 'json')

	logging.debug(f">> payload {request.message.to_json(4)}")

	asyncio.ensure_future(run_processor(request))
	return Response.new(status=202)


@register_route('GET', '/.well-known/webfinger')
async def webfinger(request):
	try:
		subject = request.query['resource']

	except KeyError:
		return Response.new_error(400, 'missing \'resource\' query key', 'json')

	if subject != f'acct:relay@{request.config.host}':
		return Response.new_error(404, 'user not found', 'json')

	data = {
		'subject': subject,
		'aliases': [request.config.actor],
		'links': [
			{'href': request.config.actor, 'rel': 'self', 'type': 'application/activity+json'},
			{'href': request.config.actor, 'rel': 'self', 'type': 'application/ld+json; profile=\"https://www.w3.org/ns/activitystreams\"'}
		]
	}

	return Response.new(data, ctype='json')


@register_route('GET', '/nodeinfo/{version:\d.\d\.json}')
async def nodeinfo_2_0(request):
	niversion = request.match_info['version'][:3]
	data = {
		'openRegistrations': not request.config.whitelist_enabled,
		'protocols': ['activitypub'],
		'services': {
			'inbound': [],
			'outbound': []
		},
		'software': {
			'name': 'activityrelay',
			'version': version
		},
		'usage': {
			'localPosts': 0,
			'users': {
				'total': 1
			}
		},
		'metadata': {
			'peers': request.database.hostnames
		},
		'version': niversion
	}

	if version == '2.1':
		data['software']['repository'] = 'https://git.pleroma.social/pleroma/relay'

	return Response.new(data, ctype='json')


@register_route('GET', '/.well-known/nodeinfo')
async def nodeinfo_wellknown(request):
	data = WKNodeinfo.new(
		v20 = f'https://{request.config.host}/nodeinfo/2.0.json',
		v21 = f'https://{request.config.host}/nodeinfo/2.1.json'
	)

	return Response.new(data, ctype='json')


@register_route('GET', '/stats')
async def stats(request):
    return Response.new(STATS, ctype='json')
add optional push worker threads 2022-11-20 10:50:14 +00:00			`import asyncio`
Reorganize codebase 2022-05-06 07:04:51 +00:00			`import logging`
			`import subprocess`
			`import traceback`

only fetch commit hash if in running from git repo 2022-11-16 15:41:00 +00:00			`from pathlib import Path`

create Application class 2022-11-07 12:54:32 +00:00			`from . import __version__, misc`
Reorganize codebase 2022-05-06 07:04:51 +00:00			`from .http_debug import STATS`
use signature keyid instead of object actor to fetch actor 2022-11-17 21:30:56 +00:00			`from .misc import DotDict, Message, Response, WKNodeinfo`
Reorganize codebase 2022-05-06 07:04:51 +00:00			`from .processors import run_processor`


create Application class 2022-11-07 12:54:32 +00:00			`routes = []`
only fetch commit hash if in running from git repo 2022-11-16 15:41:00 +00:00			`version = __version__`

create Application class 2022-11-07 12:54:32 +00:00
only fetch commit hash if in running from git repo 2022-11-16 15:41:00 +00:00			`if Path(__file__).parent.parent.joinpath('.git').exists():`
			`try:`
			`commit_label = subprocess.check_output(["git", "rev-parse", "HEAD"]).strip().decode('ascii')`
			`version = f'{__version__} {commit_label}'`
create Response class 2022-11-09 10:58:35 +00:00
only fetch commit hash if in running from git repo 2022-11-16 15:41:00 +00:00			`except:`
			`pass`
create Response class 2022-11-09 10:58:35 +00:00
create Application class 2022-11-07 12:54:32 +00:00
			`def register_route(method, path):`
			`def wrapper(func):`
			`routes.append([method, path, func])`
			`return func`

			`return wrapper`


			`@register_route('GET', '/')`
Reorganize codebase 2022-05-06 07:04:51 +00:00			`async def home(request):`
use new request properties and only fetch nodeinfo on follow 2022-11-18 21:41:14 +00:00			`targets = '<br>'.join(request.database.hostnames)`
			`note = request.config.note`
			`count = len(request.database.hostnames)`
			`host = request.config.host`
create Application class 2022-11-07 12:54:32 +00:00
			`text = f"""`
Reorganize codebase 2022-05-06 07:04:51 +00:00			`<html><head>`
			`<title>ActivityPub Relay at {host}</title>`
			`<style>`
			`p {{ color: #FFFFFF; font-family: monospace, arial; font-size: 100%; }}`
			`body {{ background-color: #000000; }}`
			`a {{ color: #26F; }}`
			`a:visited {{ color: #46C; }}`
			`a:hover {{ color: #8AF; }}`
			`</style>`
			`</head>`
			`<body>`
			`<p>This is an Activity Relay for fediverse instances.</p>`
			`<p>{note}</p>`
			`<p>You may subscribe to this relay with the address: <a href="https://{host}/actor">https://{host}/actor</a></p>`
			`<p>To host your own relay, you may download the code at this address: <a href="https://git.pleroma.social/pleroma/relay">https://git.pleroma.social/pleroma/relay</a></p>`
			`<br><p>List of {count} registered instances:<br>{targets}</p>`
create Application class 2022-11-07 12:54:32 +00:00			`</body></html>"""`
Reorganize codebase 2022-05-06 07:04:51 +00:00
create Response class 2022-11-09 10:58:35 +00:00			`return Response.new(text, ctype='html')`
Reorganize codebase 2022-05-06 07:04:51 +00:00

create Application class 2022-11-07 12:54:32 +00:00			`@register_route('GET', '/inbox')`
			`@register_route('GET', '/actor')`
Reorganize codebase 2022-05-06 07:04:51 +00:00			`async def actor(request):`
create Message class 2022-11-07 10:30:13 +00:00			`data = Message.new_actor(`
use new request properties and only fetch nodeinfo on follow 2022-11-18 21:41:14 +00:00			`host = request.config.host,`
			`pubkey = request.database.pubkey`
create Message class 2022-11-07 10:30:13 +00:00			`)`
Reorganize codebase 2022-05-06 07:04:51 +00:00
create Response class 2022-11-09 10:58:35 +00:00			`return Response.new(data, ctype='activity')`
Reorganize codebase 2022-05-06 07:04:51 +00:00

create Application class 2022-11-07 12:54:32 +00:00			`@register_route('POST', '/inbox')`
			`@register_route('POST', '/actor')`
Reorganize codebase 2022-05-06 07:04:51 +00:00			`async def inbox(request):`
use new request properties and only fetch nodeinfo on follow 2022-11-18 21:41:14 +00:00			`config = request.config`
			`database = request.database`
Reorganize codebase 2022-05-06 07:04:51 +00:00
			`## reject if missing signature header`
use new request properties and only fetch nodeinfo on follow 2022-11-18 21:41:14 +00:00			`if not request.signature:`
Reorganize codebase 2022-05-06 07:04:51 +00:00			`logging.verbose('Actor missing signature header')`
			`raise HTTPUnauthorized(body='missing signature')`

			`try:`
use new request properties and only fetch nodeinfo on follow 2022-11-18 21:41:14 +00:00			`request['message'] = await request.json(loads=Message.new_from_json)`

			`## reject if there is no message`
			`if not request.message:`
			`logging.verbose('empty message')`
			`return Response.new_error(400, 'missing message', 'json')`
create Message class 2022-11-07 10:30:13 +00:00
skip raising a KeyError on missing actor 2022-11-16 14:12:23 +00:00			`## reject if there is no actor in the message`
use new request properties and only fetch nodeinfo on follow 2022-11-18 21:41:14 +00:00			`if 'actor' not in request.message:`
			`logging.verbose('actor not in message')`
skip raising a KeyError on missing actor 2022-11-16 14:12:23 +00:00			`return Response.new_error(400, 'no actor in message', 'json')`
Reorganize codebase 2022-05-06 07:04:51 +00:00
			`except:`
use new request properties and only fetch nodeinfo on follow 2022-11-18 21:41:14 +00:00			`## this code should hopefully never get called`
Reorganize codebase 2022-05-06 07:04:51 +00:00			`traceback.print_exc()`
			`logging.verbose('Failed to parse inbox message')`
create Response class 2022-11-09 10:58:35 +00:00			`return Response.new_error(400, 'failed to parse message', 'json')`
Reorganize codebase 2022-05-06 07:04:51 +00:00
use new request properties and only fetch nodeinfo on follow 2022-11-18 21:41:14 +00:00			`request['actor'] = await misc.request(request.signature.keyid)`
Reorganize codebase 2022-05-06 07:04:51 +00:00
			`## reject if actor is empty`
use new request properties and only fetch nodeinfo on follow 2022-11-18 21:41:14 +00:00			`if not request.actor:`
ignore account Deletes 2022-11-20 10:12:11 +00:00			`## ld signatures aren't handled atm, so just ignore it`
			`if data.type == 'Delete':`
			`logging.verbose(f'Instance sent a delete which cannot be handled')`
			`return Response.new(status=202)`

			`logging.verbose(f'Failed to fetch actor: {request.signature.keyid}')`
create Response class 2022-11-09 10:58:35 +00:00			`return Response.new_error(400, 'failed to fetch actor', 'json')`
Reorganize codebase 2022-05-06 07:04:51 +00:00
fetch nodeinfo software name on inbox request instead of startup 2022-11-20 10:22:57 +00:00			`request['instance'] = request.database.get_inbox(request['actor'].inbox)`

Reorganize codebase 2022-05-06 07:04:51 +00:00			`## reject if the actor isn't whitelisted while the whiltelist is enabled`
use new request properties and only fetch nodeinfo on follow 2022-11-18 21:41:14 +00:00			`if config.whitelist_enabled and not config.is_whitelisted(request.actor.domain):`
			`logging.verbose(f'Rejected actor for not being in the whitelist: {request.actor.id}')`
create Response class 2022-11-09 10:58:35 +00:00			`return Response.new_error(403, 'access denied', 'json')`
Reorganize codebase 2022-05-06 07:04:51 +00:00
			`## reject if actor is banned`
use new request properties and only fetch nodeinfo on follow 2022-11-18 21:41:14 +00:00			`if request.config.is_banned(request.actor.domain):`
use signature keyid instead of object actor to fetch actor 2022-11-17 21:30:56 +00:00			`logging.verbose(f'Ignored request from banned actor: {actor.id}')`
create Response class 2022-11-09 10:58:35 +00:00			`return Response.new_error(403, 'access denied', 'json')`
Reorganize codebase 2022-05-06 07:04:51 +00:00
			`## reject if the signature is invalid`
use new request properties and only fetch nodeinfo on follow 2022-11-18 21:41:14 +00:00			`if not (await misc.validate_signature(request.actor, request.signature, request)):`
use signature keyid instead of object actor to fetch actor 2022-11-17 21:30:56 +00:00			`logging.verbose(f'signature validation failed for: {actor.id}')`
create Response class 2022-11-09 10:58:35 +00:00			`return Response.new_error(401, 'signature check failed', 'json')`
Reorganize codebase 2022-05-06 07:04:51 +00:00
			`## reject if activity type isn't 'Follow' and the actor isn't following`
use new request properties and only fetch nodeinfo on follow 2022-11-18 21:41:14 +00:00			`if request.message.type != 'Follow' and not database.get_inbox(request.actor.domain):`
			`logging.verbose(f'Rejected actor for trying to post while not following: {request.actor.id}')`
create Response class 2022-11-09 10:58:35 +00:00			`return Response.new_error(401, 'access denied', 'json')`
Reorganize codebase 2022-05-06 07:04:51 +00:00
use new request properties and only fetch nodeinfo on follow 2022-11-18 21:41:14 +00:00			`logging.debug(f">> payload {request.message.to_json(4)}")`
Reorganize codebase 2022-05-06 07:04:51 +00:00
add optional push worker threads 2022-11-20 10:50:14 +00:00			`asyncio.ensure_future(run_processor(request))`
create Response class 2022-11-09 10:58:35 +00:00			`return Response.new(status=202)`
Reorganize codebase 2022-05-06 07:04:51 +00:00

create Application class 2022-11-07 12:54:32 +00:00			`@register_route('GET', '/.well-known/webfinger')`
Reorganize codebase 2022-05-06 07:04:51 +00:00			`async def webfinger(request):`
views.webfinger: return 400 error on missing resource 2022-11-18 16:50:12 +00:00			`try:`
			`subject = request.query['resource']`

			`except KeyError:`
			`return Response.new_error(400, 'missing \'resource\' query key', 'json')`
Reorganize codebase 2022-05-06 07:04:51 +00:00
use new request properties and only fetch nodeinfo on follow 2022-11-18 21:41:14 +00:00			`if subject != f'acct:relay@{request.config.host}':`
create Response class 2022-11-09 10:58:35 +00:00			`return Response.new_error(404, 'user not found', 'json')`
Reorganize codebase 2022-05-06 07:04:51 +00:00
			`data = {`
			`'subject': subject,`
use new request properties and only fetch nodeinfo on follow 2022-11-18 21:41:14 +00:00			`'aliases': [request.config.actor],`
Reorganize codebase 2022-05-06 07:04:51 +00:00			`'links': [`
use new request properties and only fetch nodeinfo on follow 2022-11-18 21:41:14 +00:00			`{'href': request.config.actor, 'rel': 'self', 'type': 'application/activity+json'},`
			`{'href': request.config.actor, 'rel': 'self', 'type': 'application/ld+json; profile=\"https://www.w3.org/ns/activitystreams\"'}`
Reorganize codebase 2022-05-06 07:04:51 +00:00			`]`
			`}`

create Response class 2022-11-09 10:58:35 +00:00			`return Response.new(data, ctype='json')`
Reorganize codebase 2022-05-06 07:04:51 +00:00

create WKNodeinfo class and add nodeinfo 2.1 path 2022-11-07 13:18:25 +00:00			`@register_route('GET', '/nodeinfo/{version:\d.\d\.json}')`
Reorganize codebase 2022-05-06 07:04:51 +00:00			`async def nodeinfo_2_0(request):`
fix a couple nodeinfo values 2022-11-09 11:11:16 +00:00			`niversion = request.match_info['version'][:3]`
Reorganize codebase 2022-05-06 07:04:51 +00:00			`data = {`
use new request properties and only fetch nodeinfo on follow 2022-11-18 21:41:14 +00:00			`'openRegistrations': not request.config.whitelist_enabled,`
Reorganize codebase 2022-05-06 07:04:51 +00:00			`'protocols': ['activitypub'],`
			`'services': {`
			`'inbound': [],`
			`'outbound': []`
			`},`
			`'software': {`
			`'name': 'activityrelay',`
exclude git hash if unavailable 2022-06-06 12:37:08 +00:00			`'version': version`
Reorganize codebase 2022-05-06 07:04:51 +00:00			`},`
			`'usage': {`
			`'localPosts': 0,`
			`'users': {`
			`'total': 1`
			`}`
			`},`
			`'metadata': {`
use new request properties and only fetch nodeinfo on follow 2022-11-18 21:41:14 +00:00			`'peers': request.database.hostnames`
Reorganize codebase 2022-05-06 07:04:51 +00:00			`},`
fix a couple nodeinfo values 2022-11-09 11:11:16 +00:00			`'version': niversion`
Reorganize codebase 2022-05-06 07:04:51 +00:00			`}`

create WKNodeinfo class and add nodeinfo 2.1 path 2022-11-07 13:18:25 +00:00			`if version == '2.1':`
			`data['software']['repository'] = 'https://git.pleroma.social/pleroma/relay'`

create Response class 2022-11-09 10:58:35 +00:00			`return Response.new(data, ctype='json')`
Reorganize codebase 2022-05-06 07:04:51 +00:00

create Application class 2022-11-07 12:54:32 +00:00			`@register_route('GET', '/.well-known/nodeinfo')`
Reorganize codebase 2022-05-06 07:04:51 +00:00			`async def nodeinfo_wellknown(request):`
create WKNodeinfo class and add nodeinfo 2.1 path 2022-11-07 13:18:25 +00:00			`data = WKNodeinfo.new(`
use new request properties and only fetch nodeinfo on follow 2022-11-18 21:41:14 +00:00			`v20 = f'https://{request.config.host}/nodeinfo/2.0.json',`
			`v21 = f'https://{request.config.host}/nodeinfo/2.1.json'`
create WKNodeinfo class and add nodeinfo 2.1 path 2022-11-07 13:18:25 +00:00			`)`

create Response class 2022-11-09 10:58:35 +00:00			`return Response.new(data, ctype='json')`
Reorganize codebase 2022-05-06 07:04:51 +00:00

create Application class 2022-11-07 12:54:32 +00:00			`@register_route('GET', '/stats')`
Reorganize codebase 2022-05-06 07:04:51 +00:00			`async def stats(request):`
create Response class 2022-11-09 10:58:35 +00:00			`return Response.new(STATS, ctype='json')`