89 lines
2.2 KiB
Python
89 lines
2.2 KiB
Python
|
import argparse
|
||
|
|
from collections.abc import Callable
|
||
|
|
from pathlib import Path
|
||
|
|
import json
|
||
|
|
|
||
|
|
from .options import EffectsOptions
|
||
|
|
from . import list_effects, instantiate_effects, parse_derivation, run_effects
|
||
|
|
|
||
|
|
|
||
|
|
def list_command(options: EffectsOptions) -> None:
|
||
|
|
print(list_effects(options))
|
||
|
|
|
||
|
|
|
||
|
|
def run_command(options: EffectsOptions) -> None:
|
||
|
|
drv_path = instantiate_effects(options)
|
||
|
|
drvs = parse_derivation(drv_path)
|
||
|
|
drv = next(iter(drvs.values()))
|
||
|
|
|
||
|
|
if options.secrets:
|
||
|
|
secrets = json.loads(options.secrets.read_text())
|
||
|
|
else:
|
||
|
|
secrets = {}
|
||
|
|
run_effects(drv_path, drv, secrets=secrets)
|
||
|
|
|
||
|
|
|
||
|
|
def run_all_command(options: EffectsOptions) -> None:
|
||
|
|
print("TODO")
|
||
|
|
|
||
|
|
|
||
|
|
def parse_args() -> tuple[Callable[[EffectsOptions], None], EffectsOptions]:
|
||
|
|
parser = argparse.ArgumentParser(description="Run effects from a hercules-ci flake")
|
||
|
|
parser.add_argument(
|
||
|
|
"--secrets",
|
||
|
|
type=Path,
|
||
|
|
help="Path to a json file with secrets",
|
||
|
|
)
|
||
|
|
parser.add_argument(
|
||
|
|
"--rev",
|
||
|
|
type=str,
|
||
|
|
help="Git revision to use",
|
||
|
|
)
|
||
|
|
parser.add_argument(
|
||
|
|
"--branch",
|
||
|
|
type=str,
|
||
|
|
help="Git branch to use",
|
||
|
|
)
|
||
|
|
parser.add_argument(
|
||
|
|
"--repo",
|
||
|
|
type=str,
|
||
|
|
help="Git repo to prepend to be",
|
||
|
|
)
|
||
|
|
parser.add_argument(
|
||
|
|
"--path",
|
||
|
|
type=str,
|
||
|
|
help="Path to the repository",
|
||
|
|
)
|
||
|
|
subparser = parser.add_subparsers(
|
||
|
|
dest="command",
|
||
|
|
required=True,
|
||
|
|
help="Command to run",
|
||
|
|
)
|
||
|
|
list_parser = subparser.add_parser(
|
||
|
|
"list",
|
||
|
|
help="List available effects",
|
||
|
|
)
|
||
|
|
list_parser.set_defaults(command=list_command)
|
||
|
|
run_parser = subparser.add_parser(
|
||
|
|
"run",
|
||
|
|
help="Run an effect",
|
||
|
|
)
|
||
|
|
run_parser.set_defaults(command=run_command)
|
||
|
|
run_parser.add_argument(
|
||
|
|
"effect",
|
||
|
|
help="Effect to run",
|
||
|
|
)
|
||
|
|
run_all_parser = subparser.add_parser(
|
||
|
|
"run-all",
|
||
|
|
help="Run all effects",
|
||
|
|
)
|
||
|
|
run_all_parser.set_defaults(command=run_all_command)
|
||
|
|
|
||
|
|
args = parser.parse_args()
|
||
|
|
return args.command, EffectsOptions(secrets=args.secrets)
|
||
|
|
|
||
|
|
|
||
|
|
def main() -> None:
|
||
|
|
command, options = parse_args()
|
||
|
|
command(options)
|