feat: Add 'image erase' command

Signed-off-by: Tormod Volden <[email protected]>

Author: tormodvolden

smpmgr/image_management.py

@@ -20,7 +20,7 @@
 from smpclient import SMPClient
 from smpclient.generics import error, success
 from smpclient.mcuboot import ImageInfo
-from smpclient.requests.image_management import ImageStatesRead, ImageStatesWrite
+from smpclient.requests.image_management import ImageErase, ImageStatesRead, ImageStatesWrite
 from typing_extensions import Annotated
 
 from smpmgr.common import Options, connect_with_spinner, get_smpclient, smp_request
@@ -88,6 +88,36 @@ async def f() -> None:
     asyncio.run(f())
 
 
+@app.command()
+def erase(
+    ctx: typer.Context,
+    slot: Annotated[
+        int,
+        typer.Option(help="Image slot to erase, as displayed by image state-read (0-indexed)"),
+    ],
+) -> None:
+    """Request to erase an image slot on the SMP Server."""
+
+    options = cast(Options, ctx.obj)
+    smpclient = get_smpclient(options)
+
+    async def f() -> None:
+        await connect_with_spinner(smpclient, options.timeout)
+
+        r = await smp_request(
+            smpclient, options, ImageErase(slot=slot), "Waiting for image erase..."
+        )
+
+        if error(r):
+            print(r)
+        elif success(r):
+            pass
+        else:
+            raise Exception("Unreachable")
+
+    asyncio.run(f())
+
+
 async def upload_with_progress_bar(
     smpclient: SMPClient, file: typer.FileBinaryRead | BufferedReader, slot: int = 0
 ) -> None: