"kiwi init" as a KiwiCommand

kiwi_scp/commands/cli.py

@@ -1,7 +1,7 @@
 import os
 import sys
 from enum import Enum, auto
-from typing import List, Tuple, Iterable
+from typing import List, Tuple, Iterable, Any, Type
 
 import click
 
@@ -53,6 +53,28 @@ class KiwiCommand:
                 (item, "blue"),
             )
 
+    @staticmethod
+    def user_query(description: str, default: Any, cast_to: Type[Any] = str):
+        # prompt user as per argument
+        while True:
+            try:
+                prompt = \
+                    click.style(f"Enter {description} [", fg="green") + \
+                    click.style(default, fg="blue") + \
+                    click.style("] ", fg="green")
+                str_value = input(prompt).strip()
+                if str_value:
+                    return cast_to(str_value)
+                else:
+                    return default
+
+            except EOFError:
+                click.echo("Input aborted.")
+                return default
+
+            except Exception as e:
+                click.echo(f"Invalid input: {e}")
+
     @classmethod
     def run_for_instance(cls, instance: Instance, **kwargs) -> None:
         for project in instance.config.projects:

kiwi_scp/commands/cmd_init.py

@@ -5,20 +5,18 @@ from pathlib import Path
 
 import click
 
-from .decorators import _pass_instance as pass_instance
+from .cli import KiwiCommandType, KiwiCommand
+from .decorators import kiwi_command
 from .._constants import KIWI_CONF_NAME
 from ..config import KiwiConfig
 from ..instance import Instance
-from ..misc import user_query
+
+_logger = logging.getLogger(__name__)
 
 
-@click.command(
-    "init",
-    short_help="Initializes kiwi-scp",
-)
 @click.option(
-    "-o",
+    "-d",
-    "--output",
+    "--directory",
     help=f"initialize a kiwi-scp instance in another directory",
     type=click.Path(
         path_type=Path,
@@ -31,46 +29,45 @@ from ..misc import user_query
     "--force/--no-force",
     help=f"use default values even if {KIWI_CONF_NAME} is present",
 )
-@click.option(
+@kiwi_command(
-    "-s/-S",
+    "init",
-    "--show/--no-show",
+    KiwiCommandType.INSTANCE,
-    help=f"show effective {KIWI_CONF_NAME} contents instead",
+    short_help="Initializes kiwi-scp",
 )
-@pass_instance
+class CMD(KiwiCommand):
-def CMD(ctx: Instance, output: Path, force: bool, show: bool):
     """Initialize or reconfigure a kiwi-scp instance"""
 
-    if output is not None:
+    @classmethod
-        ctx.directory = output
+    def run_for_instance(cls, instance: Instance, output: Path = None, force: bool = None, **kwargs) -> None:
+        if output is not None:
+            instance.directory = output
 
-    current_config = KiwiConfig() if force else ctx.config
+        current_config = KiwiConfig() if force else instance.config
 
-    if show:
+        # check force switch
-        # just show the currently effective kiwi.yml
+        if force and os.path.isfile(KIWI_CONF_NAME):
-        click.echo_via_pager(current_config.kiwi_yml)
+            _logger.warning(f"About to overwrite an existing '{KIWI_CONF_NAME}'!")
-        return
 
-    # check force switch
+        # build new kiwi dict
-    if force and os.path.isfile(KIWI_CONF_NAME):
+        kiwi_dict = current_config.kiwi_dict
-        logging.warning(f"Overwriting an existing '{KIWI_CONF_NAME}'!")
+        kiwi_dict.update({
+            "version": KiwiCommand.user_query("kiwi-scp version to use in this instance", current_config.version),
+            "storage": {
+                "directory": KiwiCommand.user_query("local directory for service data",
+                                                    current_config.storage.directory, Path),
+            },
+            "network": {
+                "name": KiwiCommand.user_query("name for local network hub", current_config.network.name),
+                "cidr": KiwiCommand.user_query("CIDRv4 block for local network hub", current_config.network.cidr,
+                                               IPv4Network),
+            },
+        })
 
-    # build new kiwi dict
+        # ensure output directory exists
-    kiwi_dict = current_config.kiwi_dict
+        if not os.path.isdir(instance.directory):
-    kiwi_dict.update({
+            os.mkdir(instance.directory)
-        "version": user_query("kiwi-scp version to use in this instance", current_config.version),
-        "storage": {
-            "directory": user_query("local directory for service data", current_config.storage.directory, Path),
-        },
-        "network": {
-            "name": user_query("name for local network hub", current_config.network.name),
-            "cidr": user_query("CIDRv4 block for local network hub", current_config.network.cidr, IPv4Network),
-        },
-    })
 
-    # ensure output directory exists
+        # write out the new kiwi.yml
-    if not os.path.isdir(ctx.directory):
+        cfg = KiwiConfig.parse_obj(kiwi_dict)
-        os.mkdir(ctx.directory)
+        with open(instance.directory.joinpath(KIWI_CONF_NAME), "w") as file:
-
+            cfg.dump_kiwi_yml(file)
-    # write out the new kiwi.yml
-    with open(ctx.directory.joinpath(KIWI_CONF_NAME), "w") as file:
-        KiwiConfig.parse_obj(kiwi_dict).dump_kiwi_yml(file)

kiwi_scp/misc.py

@@ -1,31 +1,12 @@
 import re
-from typing import Any, Type, Optional
+from typing import Optional
 
-import click
 import ruamel.yaml
 import ruamel.yaml.compat
 
 from ._constants import HEADER_KIWI_CONF_NAME
 
 
-def user_query(description: str, default: Any, cast_to: Type[Any] = str):
-    # prompt user as per argument
-    while True:
-        try:
-            str_value = input(f"Enter {description} [{default}] ").strip()
-            if str_value:
-                return cast_to(str_value)
-            else:
-                return default
-
-        except EOFError:
-            click.echo("Input aborted.")
-            return default
-
-        except Exception as e:
-            click.echo(f"Invalid input: {e}")
-
-
 class YAML(ruamel.yaml.YAML):
     def __init__(self, *args, **kwargs):
         super().__init__(*args, **kwargs)