mirror of
https://github.com/revng/revng
synced 2026-06-21 14:07:57 +00:00
Giacomo Vercesi
828bd9c49f
Add the PRSS (Pypeline Remote Storage Server) and the relay companion server.
48 lines
1.2 KiB
Python
48 lines
1.2 KiB
Python
#
|
|
# This file is distributed under the MIT License. See LICENSE.md for details.
|
|
#
|
|
|
|
import click
|
|
|
|
import revng.pypeline.rss_server.relay as app
|
|
from revng.pypeline.cli.context import ClickContext, pass_context
|
|
from revng.pypeline.cli.hypercorn import hypercorn_command, run_hypercorn
|
|
|
|
|
|
@click.command(help="Start the notification relay HTTP server")
|
|
@hypercorn_command(exclude={"bind"})
|
|
@click.option(
|
|
"--psk",
|
|
help="The pre-shared key for the /publish endpoint",
|
|
envvar="PYPELINE_RSS_RELAY_PSK",
|
|
)
|
|
@click.option(
|
|
"--notifications-bind",
|
|
help="Address where notifications will be listened on",
|
|
default="127.0.0.1:8002",
|
|
)
|
|
@click.option(
|
|
"--publish-bind",
|
|
help="Address where publications will be listened on",
|
|
default="127.0.0.1:8003",
|
|
)
|
|
@pass_context
|
|
def relay(
|
|
ctx: ClickContext,
|
|
production: bool,
|
|
psk: str,
|
|
notifications_bind: str,
|
|
publish_bind: str,
|
|
):
|
|
def on_shutdown():
|
|
assert app.shutdown_begun is not None
|
|
app.shutdown_begun.set()
|
|
|
|
ctx.obj.hypercorn_configuration.bind = [*{notifications_bind, publish_bind}]
|
|
|
|
return run_hypercorn(
|
|
lambda: app.make_starlette(production, notifications_bind, publish_bind, psk),
|
|
ctx.obj.hypercorn_configuration,
|
|
on_shutdown,
|
|
)
|