sedi-relay/relay/views.py

import aputils
import asyncio
import logging
import subprocess
import traceback

from pathlib import Path

from . import __version__, misc
from .misc import DotDict, Message, Response
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>SEDI中繼器</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>Misskey及Mastodon站長請訂閱這個地址：<a href="https://{host}/inbox">https://{host}/inbox</a></p>
<p>Pleroma及Friendica站長請訂閱這個地址：<a href="https://{host}/actor">https://{host}/actor</a></p>
<p>原始碼：<a href="https://git.seediqbale.xyz/pch_xyz/sedi-relay">https://git.seediqbale.xyz/pch_xyz/sedi-relay</a></p>
<p>activityrelay v0.2.4</p>
<br><p>有 {count} 個實例訂閱中：<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.signer.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 request.app.client.get(request.signature.keyid, sign_headers=True)

	## reject if actor is empty
	if not request.actor:
		## ld signatures aren't handled atm, so just ignore it
		if request['message'].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
	try:
		await request.actor.signer.validate_aiohttp_request(request)

	except aputils.SignatureValidationError as e:
		logging.verbose(f'signature validation failed for: {actor.id}')
		logging.debug(str(e))
		return Response.new_error(401, str(e), '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 = aputils.Webfinger.new(
		handle = 'relay',
		domain = request.config.host,
		actor = request.config.actor
	)

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


@register_route('GET', '/nodeinfo/{version:\d.\d\.json}')
async def nodeinfo(request):
	niversion = request.match_info['version'][:3]

	data = dict(
		name = 'activityrelay',
		version = version,
		protocols = ['activitypub'],
		open_regs = not request.config.whitelist_enabled,
		users = 1,
		metadata = {'peers': request.database.hostnames}
	)

	if niversion == '2.1':
		data['repo'] = 'https://git.pleroma.social/pleroma/relay'

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


@register_route('GET', '/.well-known/nodeinfo')
async def nodeinfo_wellknown(request):
	data = aputils.WellKnownNodeinfo.new_template(request.config.host)
	return Response.new(data, ctype='json')
-												add aputils module for hs2019 support

											
										
										
											2022-11-27 03:16:14 +00:00
+								import aputils
-												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
-												replace various classes with aputils classes

											
										
										
											2022-12-02 05:50:57 +00:00
+								from .misc import DotDict, Message, Response
-												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>
-												localization

											
										
										
											2023-02-10 00:51:40 +00:00
+								<title>SEDI中繼器</title>
-												Reorganize codebase

											
										
										
											2022-05-06 07:04:51 +00:00
+								<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>
-												localization

											
										
										
											2023-02-10 00:51:40 +00:00
+								<p>Misskey及Mastodon站長請訂閱這個地址：<a href="https://{host}/inbox">https://{host}/inbox</a></p>
 								<p>Pleroma及Friendica站長請訂閱這個地址：<a href="https://{host}/actor">https://{host}/actor</a></p>
 								<p>原始碼：<a href="https://git.seediqbale.xyz/pch_xyz/sedi-relay">https://git.seediqbale.xyz/pch_xyz/sedi-relay</a></p>
-												上傳檔案到「relay」

added version info

											
										
										
											2023-02-10 01:14:32 +00:00
+								<p>activityrelay v0.2.4</p>
-												localization

											
										
										
											2023-02-10 00:51:40 +00:00
+								<br><p>有 {count} 個實例訂閱中：<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,
-												add aputils module for hs2019 support

											
										
										
											2022-11-27 03:16:14 +00:00
+										pubkey = request.database.signer.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
-												create HttpClient class to avoid creating a new session every request

											
										
										
											2022-11-26 23:56:34 +00:00
+									request['actor'] = await request.app.client.get(request.signature.keyid, sign_headers=True)
-												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
-												fix NameError in inbox

											
										
										
											2022-11-22 23:09:25 +00:00
+										if request['message'].type == 'Delete':
-												ignore account Deletes

											
										
										
											2022-11-20 10:12:11 +00:00
+											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
-												add aputils module for hs2019 support

											
										
										
											2022-11-27 03:16:14 +00:00
+									try:
 										await request.actor.signer.validate_aiohttp_request(request)
 									except aputils.SignatureValidationError as e:
-												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}')
-												add aputils module for hs2019 support

											
										
										
											2022-11-27 03:16:14 +00:00
+										logging.debug(str(e))
 										return Response.new_error(401, str(e), '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
-												replace various classes with aputils classes

											
										
										
											2022-12-02 05:50:57 +00:00
+									data = aputils.Webfinger.new(
 										handle = 'relay',
 										domain = request.config.host,
 										actor = request.config.actor
 									)
-												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}')
-												replace various classes with aputils classes

											
										
										
											2022-12-02 05:50:57 +00:00
+								async def nodeinfo(request):
-												fix a couple nodeinfo values

											
										
										
											2022-11-09 11:11:16 +00:00
+									niversion = request.match_info['version'][:3]
-												create WKNodeinfo class and add nodeinfo 2.1 path

											
										
										
											2022-11-07 13:18:25 +00:00
-												replace various classes with aputils classes

											
										
										
											2022-12-02 05:50:57 +00:00
+									data = dict(
 										name = 'activityrelay',
 										version = version,
 										protocols = ['activitypub'],
 										open_regs = not request.config.whitelist_enabled,
 										users = 1,
 										metadata = {'peers': request.database.hostnames}
 									)
 									if niversion == '2.1':
 										data['repo'] = 'https://git.pleroma.social/pleroma/relay'
 									return Response.new(aputils.Nodeinfo.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):
-												replace various classes with aputils classes

											
										
										
											2022-12-02 05:50:57 +00:00
+									data = aputils.WellKnownNodeinfo.new_template(request.config.host)
-												create Response class

											
										
										
											2022-11-09 10:58:35 +00:00
+									return Response.new(data, ctype='json')