diff --git a/cli/src/ioc.py b/cli/src/ioc.py index c3707b6..19e08f6 100644 --- a/cli/src/ioc.py +++ b/cli/src/ioc.py @@ -20,9 +20,12 @@ IOC_PLAYBOOK_START = ANSIBLE_PLAYBOOKS_DIR / "ioc-start.yml" def install( env: str = typer.Option(..., "--env"), service_name_lower: str = typer.Option(..., "--service-name", "-s"), + branch_name: str | None = typer.Option(None, "--branch-name", "-b"), ): env_var = f"agebd_env={env}" - branch_var = f"branch_name=feature/add-service-{service_name_lower}" + if branch_name is None: + branch_name = f"feature/add-service-{service_name_lower}" + branch_var = f"branch_name={branch_name}" service_name_var = f"service_name_lower={service_name_lower}" run_playbook(playbook=str(IOC_PLAYBOOK_INSTALL), vars=[env_var, branch_var, service_name_var]) @@ -32,9 +35,12 @@ def install( def restart( env: str = typer.Option(..., "--env"), service_name_lower: str = typer.Option(..., "--service-name", "-s"), + branch_name: str | None = typer.Option(None, "--branch-name", "-b"), ): env_var = f"agebd_env={env}" - branch_var = f"branch_name=feature/add-service-{service_name_lower}" + if branch_name is None: + branch_name = f"feature/add-service-{service_name_lower}" + branch_var = f"branch_name={branch_name}" service_name_var = f"service_name_lower={service_name_lower}" run_playbook(playbook=str(IOC_PLAYBOOK_RESTART), vars=[env_var, branch_var, service_name_var])