From 90fbe807d6dfd66645f58ee4651220f7b79660b8 Mon Sep 17 00:00:00 2001 From: lassulus Date: Tue, 28 Jan 2025 10:09:15 +0100 Subject: [PATCH] clan-cli vars upload: add optional --directory --- pkgs/clan-cli/clan_cli/vars/upload.py | 17 ++++++++++++++--- 1 file changed, 14 insertions(+), 3 deletions(-) diff --git a/pkgs/clan-cli/clan_cli/vars/upload.py b/pkgs/clan-cli/clan_cli/vars/upload.py index c59285135..605216dce 100644 --- a/pkgs/clan-cli/clan_cli/vars/upload.py +++ b/pkgs/clan-cli/clan_cli/vars/upload.py @@ -1,6 +1,7 @@ import argparse import importlib import logging +from pathlib import Path from clan_cli.completions import add_dynamic_completer, complete_machines from clan_cli.machines.machines import Machine @@ -8,15 +9,21 @@ from clan_cli.machines.machines import Machine log = logging.getLogger(__name__) -def upload_secret_vars(machine: Machine) -> None: +def upload_secret_vars(machine: Machine, directory: Path | None = None) -> None: secret_store_module = importlib.import_module(machine.secret_vars_module) secret_store = secret_store_module.SecretStore(machine=machine) - secret_store.upload(phases=["activation", "users", "services"]) + if directory: + secret_store.populate_dir(directory, phases=["activation", "users", "services"]) + else: + secret_store.upload(phases=["activation", "users", "services"]) def upload_command(args: argparse.Namespace) -> None: machine = Machine(name=args.machine, flake=args.flake) - upload_secret_vars(machine) + directory = None + if args.directory: + directory = Path(args.directory) + upload_secret_vars(machine, directory) def register_upload_parser(parser: argparse.ArgumentParser) -> None: @@ -26,4 +33,8 @@ def register_upload_parser(parser: argparse.ArgumentParser) -> None: ) add_dynamic_completer(machines_parser, complete_machines) + parser.add_argument( + "--directory", + help="If provided, secrets will be copied into this directory instead of the remote server", + ) parser.set_defaults(func=upload_command)