mirror of
https://github.com/revng/revng
synced 2026-06-21 14:07:57 +00:00
Giacomo Vercesi
9287c51d27
Instead of executing `chdir` when using the `-C` option, which is fragile when other paths are involved, store the option in the click context and propagate it to the required code paths that require knowing what the base directory is.
75 lines
2.2 KiB
Python
75 lines
2.2 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"],
|
|
base_directory=ctx.obj["base_directory"],
|
|
)
|
|
|
|
# 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)
|