mirror of
https://github.com/revng/revng
synced 2026-06-21 14:07:57 +00:00
Giacomo Vercesi
e23d238e85
Commands now require a `setup` function, where the will register the command instance manually
64 lines
2.1 KiB
Python
Executable File
64 lines
2.1 KiB
Python
Executable File
#
|
|
# 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 Command, Options, commands_registry
|
|
from .support import collect_files
|
|
|
|
|
|
def extend_list(paths, new_items):
|
|
return new_items + [path for path in paths if path not in new_items]
|
|
|
|
|
|
# flake8: noqa: F401
|
|
def run_revng_command(arguments, options: Options):
|
|
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}", "revng.cli"))
|
|
elif entry.is_dir():
|
|
modules.append(import_module(f"._commands.{entry_path.name}", "revng.cli"))
|
|
|
|
for module in modules:
|
|
setup = getattr(module, "setup", None)
|
|
if setup is not None and isfunction(setup):
|
|
setup(commands_registry)
|
|
|
|
if options.verbose:
|
|
sys.stderr.write("{}\n\n".format(" \\\n ".join([sys.argv[0]] + arguments)))
|
|
|
|
if options.dry_run:
|
|
return 0
|
|
|
|
# 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)
|
|
|
|
return commands_registry.run(arguments, options)
|
|
|
|
|
|
def main():
|
|
return run_revng_command(sys.argv[1:], Options(None, [], [], False, False, False))
|