mirror of
https://github.com/revng/revng
synced 2026-06-21 14:07:57 +00:00
Giacomo Vercesi
942cf50735
Package revng's python code in two wheels: `revng` and `revng_internal`.
The revng wheel contains the
`revng.{pipeline_description,model,tupletree}` modules, while the
`revng_internal` one everything under `revng.internal`.
79 lines
2.6 KiB
Python
79 lines
2.6 KiB
Python
#
|
|
# This file is distributed under the MIT License. See LICENSE.md for details.
|
|
#
|
|
|
|
import os
|
|
import sys
|
|
from importlib import import_module
|
|
from inspect import isfunction
|
|
from pathlib import Path
|
|
|
|
from .commands_registry import Options, commands_registry
|
|
from .support import collect_files, get_root, read_lines
|
|
|
|
|
|
def extend_list(paths, new_items):
|
|
return new_items + [path for path in paths if path not in new_items]
|
|
|
|
|
|
def revng_driver_init(arguments) -> Options:
|
|
options = Options(None, [], [], False, False, False, [])
|
|
modules = []
|
|
with os.scandir(Path(__file__).parent / "_commands") as scan:
|
|
for entry in scan:
|
|
entry_path = Path(entry.path)
|
|
if entry_path.name.startswith("__") or entry_path.name.startswith("."):
|
|
continue
|
|
if entry.is_file():
|
|
modules.append(import_module(f"._commands.{entry_path.stem}", __package__))
|
|
elif entry.is_dir():
|
|
modules.append(import_module(f"._commands.{entry_path.name}", __package__))
|
|
|
|
for module in modules:
|
|
setup = getattr(module, "setup", None)
|
|
if setup is not None and isfunction(setup):
|
|
setup(commands_registry)
|
|
|
|
# Add forced search prefixes
|
|
options.search_prefixes = extend_list(
|
|
options.search_prefixes, read_lines(get_root() / "additional-search-prefixes")
|
|
)
|
|
|
|
# Add root as search prefix
|
|
options.search_prefixes = extend_list(options.search_prefixes, [str(get_root())])
|
|
|
|
# Collect search prefixes
|
|
prefixes = []
|
|
for arg, next_arg in zip(arguments, arguments[1:] + [""]):
|
|
if arg.startswith("--prefix="):
|
|
prefixes += [arg[len("--prefix=") :]]
|
|
elif arg == "--prefix" and next_arg != "":
|
|
prefixes += [next_arg]
|
|
options.search_prefixes = extend_list(options.search_prefixes, prefixes)
|
|
|
|
# Register external commands
|
|
prefix = "revng-"
|
|
for executable in collect_files(options.search_prefixes, ["libexec", "revng"], f"{prefix}*"):
|
|
name = os.path.basename(executable)[len(prefix) :]
|
|
if not commands_registry.has_command(name):
|
|
commands_registry.register_external_command(name, executable)
|
|
|
|
commands_registry.post_init()
|
|
return options
|
|
|
|
|
|
def run_revng_command(arguments, options: Options):
|
|
if options.verbose:
|
|
sys.stderr.write("{}\n\n".format(" \\\n ".join([sys.argv[0]] + arguments)))
|
|
|
|
if options.dry_run:
|
|
return 0
|
|
|
|
return commands_registry.run(arguments, options)
|
|
|
|
|
|
def main():
|
|
arguments = sys.argv[1:]
|
|
options = revng_driver_init(arguments)
|
|
return run_revng_command(sys.argv[1:], options)
|