mirror of
https://github.com/revng/revng
synced 2026-06-21 14:07:57 +00:00
Tommaso Fontana
4e73bc05f3
The usage strings didn't always show '-- [PYPEBOX ARGS...]' or didn't show it at the end of the usage string. This commit fixes it.
74 lines
2.1 KiB
Python
74 lines
2.1 KiB
Python
#
|
|
# This file is distributed under the MIT License. See LICENSE.md for details.
|
|
#
|
|
|
|
import os
|
|
|
|
import click
|
|
import uvicorn
|
|
import yaml
|
|
|
|
import revng
|
|
from revng.pypeline.cli.utils import PypeCommand
|
|
from revng.pypeline.daemon.app import make_starlette
|
|
from revng.pypeline.daemon.daemon import Daemon
|
|
|
|
|
|
@click.command(cls=PypeCommand)
|
|
@click.option(
|
|
"--production",
|
|
is_flag=True,
|
|
help="Enable production settings.",
|
|
)
|
|
@click.pass_context
|
|
def run_daemon(ctx, production, **kwargs):
|
|
"""Start the HTTP daemon."""
|
|
|
|
with open(ctx.obj["pipeline_path"], "r", encoding="utf-8") as f:
|
|
pipeline_yaml = yaml.safe_load(f.read())
|
|
|
|
# Configure uvicorn logging
|
|
log_config = uvicorn.config.LOGGING_CONFIG
|
|
|
|
# Setup formatting
|
|
log_format = "%(asctime)s - %(name)s - %(levelname)s - %(message)s"
|
|
log_config["formatters"]["access"]["fmt"] = log_format
|
|
log_config["formatters"]["default"]["fmt"] = log_format
|
|
|
|
for uvlogger in log_config["loggers"].values():
|
|
uvlogger["level"] = "INFO"
|
|
|
|
if not production:
|
|
os.environ["STARLETTE_DEBUG"] = "1"
|
|
os.environ["REVNG_ORIGINS"] = "*"
|
|
kwargs.setdefault("host", "127.0.0.1")
|
|
else:
|
|
kwargs.setdefault("host", "0.0.0.0")
|
|
|
|
daemon = Daemon(
|
|
version=revng.__version__,
|
|
pipeline_yaml=pipeline_yaml,
|
|
pipeline=ctx.obj["pipeline"],
|
|
debug=not production,
|
|
storage_provider_url=ctx.obj["storage_provider"],
|
|
cache_dir=ctx.obj["cache_dir"],
|
|
)
|
|
|
|
# Start the uvicorn server
|
|
uvicorn.run(app=make_starlette(daemon), **kwargs)
|
|
|
|
|
|
# Inherit all params from the uvicorn cli
|
|
for param in uvicorn.main.params:
|
|
# ignore the app param because we will pass it in `run_daemon`
|
|
if param.name == "app":
|
|
continue
|
|
# Add help text that explains production changes
|
|
if param.name == "host":
|
|
param.default = None # type: ignore [attr-defined]
|
|
param.show_default = False # type: ignore [attr-defined]
|
|
param.help += ( # type: ignore [attr-defined]
|
|
" Defaults to 0.0.0.0 in production and 127.0.0.1 otherwise."
|
|
)
|
|
run_daemon.params.append(param)
|