Added cfengine add/remove/search/update to add/remove/search for modules or update cfbs project

Author: SimonThalvorsen

src/cfengine_cli/cfengine_wrapper/arg_parse.py

@@ -12,6 +12,48 @@
 
 
 def parse_wrapper_args(subp: argparse._SubParsersAction):
+    update_parser = subp.add_parser(
+        "update",
+        help="Updates the current cfbs project",
+        description="A wrapper around the cfbs `update` function",
+    )
+    update_parser.add_argument(
+        "to_update",
+        nargs="*",
+        help="Directory of cfbs-project to update",
+    )
+    remove_parser = subp.add_parser(
+        "remove",
+        help="Removes the specified module(s) from cfbs project",
+        description="A wrapper around the cfbs `remove` function",
+    )
+    remove_parser.add_argument(
+        "module",
+        nargs="+",
+        help="Module(s) for which to remove",
+    )
+
+    add_parser = subp.add_parser(
+        "add",
+        help="Adds the specified module(s) to cfbs project",
+        description="A wrapper around the cfbs `add` function",
+    )
+    add_parser.add_argument(
+        "module",
+        nargs="+",
+        help="Module(s) for which to add",
+    )
+    search_parser = subp.add_parser(
+        "search",
+        help="Searches the build-index for specified module(s)",
+        description="A wrapper around the cfbs `search` function",
+    )
+    search_parser.add_argument(
+        "module",
+        nargs="+",
+        help="Module(s) for which to lookup",
+    )
+
     input_parser = subp.add_parser(
         "input",
         help="Sets/updates input.json for selected module(s)",

src/cfengine_cli/cfengine_wrapper/cfengine_commands.py

@@ -8,7 +8,13 @@
 from cf_remote.commands import show as show_command
 from cf_remote.remote import run_command, transfer_file
 from cf_remote.commands import connect_cmd
-from cfbs.commands import input_command
+from cfbs.commands import (
+    input_command,
+    add_command,
+    remove_command,
+    update_command,
+    search_command,
+)
 
 from cfengine_cli.utils import UserError
 from cfengine_cli.cfengine_wrapper.cfengine_objects import (
@@ -297,5 +303,21 @@ def connect(host) -> int:
     return connect_cmd(host)
 
 
-def cfbs_input(modules: list[str] | None = None) -> int:
+def cfbs_input(modules: list[str]) -> int:
     return input_command(modules, "cfengine input")
+
+
+def cfbs_add(modules: list[str]) -> int:
+    return add_command(modules, "cfengine input")
+
+
+def cfbs_remove(modules: list[str] | None = None) -> int:
+    return remove_command(modules, "cfengine input")
+
+
+def cfbs_update(to_update) -> int:
+    return update_command(to_update)
+
+
+def cfbs_search(modules: list[str]) -> int:
+    return search_command(modules)

src/cfengine_cli/main.py

@@ -224,9 +224,19 @@ def run_command_with_args(args) -> int:
     if args.command == "build":
         return cfengine_commands.build(args.hub, args.non_interactive)
     if args.command == "deploy":
-        return cfengine_commands.deploy(args.hub, args.masterfiles, args.non_interactive)
+        return cfengine_commands.deploy(
+            args.hub, args.masterfiles, args.non_interactive
+        )
     if args.command == "input":
         return cfengine_commands.cfbs_input(args.module)
+    if args.command == "add":
+        return cfengine_commands.cfbs_add(args.module)
+    if args.command == "remove":
+        return cfengine_commands.cfbs_remove(args.module)
+    if args.command == "search":
+        return cfengine_commands.cfbs_search(args.module)
+    if args.command == "update":
+        return cfengine_commands.cfbs_update(args.to_update)
     if args.command == "format":
         return commands.format(args.files, args.line_length, args.check)
     if args.command == "lint":