tools/cli: add new arg for parent host

Signed-off-by: Olivier Hoareau <olivier.hoareau@vates.tech>

Author: olivierh-pro

README.md

@@ -601,6 +601,7 @@ Take a look at an example inventory file:
 
 ```toml
 # my_inventory.toml
+parent = "ip_or_hostname" # master host
 
 [all]
 enablerepos = ["xcp-ng-base"]
@@ -618,6 +619,6 @@ enablerepos = ["xcp-ng-updates"]
 > Config values under `servers` override values under `all`. For instance, the above inventory would produce
 > the following python dict:
 >
-> `{'hosts': {'ip_or_hostname-1': {'enablerepos': ['xcp-ng-base']}, 'ip_or_hostname-2': {'enablerepos': ['xcp-ng-updates']}}}`
+> `{'parent': 'ip_or_hostname', 'hosts': {'ip_or_hostname-1': {'enablerepos': ['xcp-ng-base']}, 'ip_or_hostname-2': {'enablerepos': ['xcp-ng-updates']}}}`
 >
 > Using *enablerepo flag* `-e` with inventory is still possible, it won't be used though.

lib/tools/cli.py

@@ -17,7 +17,7 @@ def _command_update(args: argparse.Namespace) -> None:
     if args.inventory:
         inventory = load_inventory(args.inventory)
     else:
-        inventory = into_inventory(args.hosts, args.repos)
+        inventory = into_inventory(args.hosts, args.repos, args.parent_host)
 
     update_pools(inventory)
 
@@ -53,6 +53,12 @@ def cli() -> None:
         dest="repos",
         help="repositories to enable when updating",
     )
+    subparser_cmd_update.add_argument(
+        "-P",
+        "--parent-host",
+        type=HostAddress,
+        help="Address (hostname|ip) of the parent master host (works with '--nested')",
+    )
     subparser_cmd_update.set_defaults(func=_command_update)
 
     args = parser.parse_args()

lib/tools/inventory.py

@@ -17,7 +17,7 @@ class Server(TypedDict):
 
 class Inventory(TypedDict):
     hosts: Servers
-
+    parent: HostAddress | None
 
 def load_inventory(inventory_path: Path) -> Inventory:
     """Create an inventory object from loaded inventory file."""
@@ -35,10 +35,11 @@ def load_inventory(inventory_path: Path) -> Inventory:
 
     return {
         "hosts": inventory_hosts,
+        "parent": data.get("parent", None),
     }
 
 
-def into_inventory(hosts: list[HostAddress], enablerepos: list[str]) -> Inventory:
+def into_inventory(hosts: list[HostAddress], enablerepos: list[str], parent: HostAddress) -> Inventory:
     """Create an inventory object from arguments.
 
     Basically, it is used as compatibility when we don't want inventory from file.
@@ -50,4 +51,5 @@ def into_inventory(hosts: list[HostAddress], enablerepos: list[str]) -> Inventor
 
     return {
         "hosts": inventory_hosts,
+        "parent": parent,
     }