mirror of
https://github.com/revng/revng
synced 2026-06-21 14:07:57 +00:00
Giacomo Vercesi
a13c3e5ac2
Some files had discordant executable bit presence and shebang presence. This commit fixes these occurrences by adding removing the required feautures where needed.
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)
|