Merge pull request #44 from looran/main

add UDP support

Author: JPHutchins

smpmgr/common.py

@@ -13,6 +13,7 @@
 from smpclient.generics import SMPRequest, TEr1, TEr2, TRep
 from smpclient.transport.ble import SMPBLETransport
 from smpclient.transport.serial import SMPSerialTransport
+from smpclient.transport.udp import SMPUDPTransport
 
 logger = logging.getLogger(__name__)
 
@@ -26,6 +27,7 @@
 class TransportDefinition:
     port: str | None
     ble: str | None
+    ip: str | None
 
 
 @dataclass(frozen=True)
@@ -56,6 +58,14 @@ def get_custom_smpclient(options: Options, smp_client_cls: Type[TSMPClient]) ->
             SMPBLETransport(),
             options.transport.ble,
         )
+    elif options.transport.ip is not None:
+        logger.info(
+            f"Initializing SMPClient with the SMPUDPTransport, {options.transport.ip=} and MTU 1024"
+        )
+        return smp_client_cls(
+            SMPUDPTransport(1024),
+            options.transport.ip,
+        )
     else:
         typer.echo(
             f"A transport option is required; "

smpmgr/main.py

@@ -46,6 +46,7 @@
 @app.callback(invoke_without_command=True)
 def options(
     ctx: typer.Context,
+    ip: str = typer.Option(None, help="The IP address to connect to for UDP transport"),
     port: str = typer.Option(
         None, help="The serial port to connect to, e.g. COM1, /dev/ttyACM0, etc."
     ),
@@ -70,7 +71,9 @@ def options(
 
     setup_logging(loglevel, logfile)
 
-    ctx.obj = Options(timeout=timeout, transport=TransportDefinition(port=port, ble=ble), mtu=mtu)
+    ctx.obj = Options(
+        timeout=timeout, transport=TransportDefinition(port=port, ble=ble, ip=ip), mtu=mtu
+    )
     logger.info(ctx.obj)
 
     if ctx.invoked_subcommand is None: