tools/cli: add new arg for parent host

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

Author: olivierh-pro

README.md

@@ -603,6 +603,7 @@ Take a look at an example inventory file:
 
 ```toml
 # my_inventory.toml
+parent = "ip_or_hostname" # master host
 
 [all]
 nested = true
@@ -621,6 +622,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, args.nested)
+        inventory = into_inventory(args.hosts, args.repos, args.nested, args.parent_host)
 
     update_pools(inventory)
 
@@ -56,6 +56,12 @@ def cli() -> None:
     subparser_cmd_update.add_argument(
         "--nested", action="store_true", default=False, help="Indicate whether hosts are nested or not"
     )
+    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

@@ -31,10 +31,13 @@ def load_inventory(inventory_path: Path) -> dict[str, dict[str, list[str]]]:
         inventory_hosts[server] = host
 
     inventory["hosts"] = inventory_hosts
+    inventory["parent"] = data.get("parent", None)
 
     return inventory
 
-def into_inventory(hosts: list[HostAddress], enablerepos: list[str], nested: bool) -> dict[HostAddress, dict[str, list[str]]]:
+def into_inventory(
+    hosts: list[HostAddress], enablerepos: list[str], nested: bool, parent: HostAddress
+) -> dict[HostAddress, dict[str, list[str]]]:
     """Create an inventory object from arguments.
 
     Basically, it is used as compatibility when we don't want inventory from file.
@@ -50,5 +53,6 @@ def into_inventory(hosts: list[HostAddress], enablerepos: list[str], nested: boo
         inventory_hosts[h] = host
 
     inventory["hosts"] = inventory_hosts
+    inventory["parent"] = parent
 
     return inventory