tools: enable --server-port arguments

Added the `--server-port` argument to any tools that act or interact
with a gateway.

Signed-off-by: Aeyohan Furtado <aeyohan@embeint.com>

Author: Aeyohan

src/infuse_iot/tools/annotate_events.py

@@ -25,10 +25,9 @@
     ClientNotificationConnectionDropped,
     GatewayRequestConnectionRequest,
     LocalClient,
-    default_multicast_address,
 )
 from infuse_iot.time import InfuseTime
-from infuse_iot.util.argparse import InfuseDeviceId, ValidFile
+from infuse_iot.util.argparse import InfuseDeviceId, ValidFile, add_server_port_parser
 from infuse_iot.util.console import choose_one
 from infuse_iot.zephyr.errno import errno
 
@@ -121,6 +120,8 @@ def add_parser(cls, parser):
 
         parser.add_argument("--id", type=InfuseDeviceId, help="Device to log events to")
 
+        add_server_port_parser(parser)
+
     def __init__(self, args):
         self._label_type = args.labels
         self._time_check = args.time or TimeCheckType.DEFAULT
@@ -153,7 +154,7 @@ def __init__(self, args):
             self._labels = []
 
         self._logger: rpc_enum_data_logger = args.logger
-        self._client = LocalClient(default_multicast_address(), 1.0)
+        self._client = LocalClient(args.server_sock, 1.0)
         self._device_id = args.id
         self.rpc_client: RpcClient | None = None
         self.connected = False

src/infuse_iot/tools/audio_record.py

@@ -19,10 +19,9 @@
     ClientNotificationEpacketReceived,
     GatewayRequestConnectionRequest,
     LocalClient,
-    default_multicast_address,
 )
 from infuse_iot.tdf import TDF
-from infuse_iot.util.argparse import InfuseDeviceId
+from infuse_iot.util.argparse import InfuseDeviceId, add_server_port_parser
 from infuse_iot.util.console import Console
 
 
@@ -32,7 +31,7 @@ class SubCommand(InfuseCommand):
     DESCRIPTION = "Record audio data to a file from TDF"
 
     def __init__(self, args):
-        self._client = LocalClient(default_multicast_address(), 1.0)
+        self._client = LocalClient(args.server_sock, 1.0)
         self._decoder = TDF()
         if args.gateway:
             self._id = InfuseID.GATEWAY
@@ -54,6 +53,8 @@ def add_parser(cls, parser):
         )
         parser.add_argument("--name", type=str, help="Filename prefix")
 
+        add_server_port_parser(parser)
+
     def handle_channel(self, channel: str, stack: ExitStack, tdf: TDF.Reading):
         if channel == "left":
             chan = self._left

src/infuse_iot/tools/auto_activate.py

@@ -18,9 +18,9 @@
 from infuse_iot.socket_comms import (
     GatewayRequestConnectionRequest,
     LocalClient,
-    default_multicast_address,
 )
 from infuse_iot.tdf import TDF
+from infuse_iot.util.argparse import add_server_port_parser
 
 
 class SubCommand(InfuseCommand):
@@ -31,7 +31,7 @@ class SubCommand(InfuseCommand):
     def __init__(self, args):
         self.app_ids = args.app
         self.active = args.active or False
-        self.client = LocalClient(default_multicast_address(), 1.0)
+        self.client = LocalClient(args.server_sock, 1.0)
         self.decoder = TDF()
         self.state = "Scanning"
         self.name = "Active" if args.active else "Inactive"
@@ -47,6 +47,8 @@ def add_parser(cls, parser):
         mode_group.add_argument("--active", action="store_true", help="Move all devices to active state")
         mode_group.add_argument("--inactive", action="store_true", help="Move all devices to inactive state")
 
+        add_server_port_parser(parser)
+
     def progress_table(self):
         table = Table()
         table.add_column()

src/infuse_iot/tools/bt_log.py

@@ -15,10 +15,9 @@
     ClientNotificationEpacketReceived,
     GatewayRequestConnectionRequest,
     LocalClient,
-    default_multicast_address,
 )
 from infuse_iot.tdf import TDF
-from infuse_iot.util.argparse import InfuseDeviceId
+from infuse_iot.util.argparse import InfuseDeviceId, add_server_port_parser
 from infuse_iot.util.console import Console
 
 
@@ -28,7 +27,7 @@ class SubCommand(InfuseCommand):
     DESCRIPTION = "Connect to remote Bluetooth device serial logs"
 
     def __init__(self, args):
-        self._client = LocalClient(default_multicast_address(), 1.0)
+        self._client = LocalClient(args.server_sock, 1.0)
         self._decoder = TDF()
         self._id = args.id
         self._data = args.data
@@ -41,6 +40,7 @@ def add_parser(cls, parser):
         parser.add_argument(
             "--conn-timeout", type=int, default=10000, help="Timeout to wait for a connection to the device (ms)"
         )
+        add_server_port_parser(parser)
 
     def run(self):
         if not self._client.comms_check():

src/infuse_iot/tools/data_logger_sync.py

@@ -28,9 +28,8 @@
 from infuse_iot.socket_comms import (
     GatewayRequestConnectionRequest,
     LocalClient,
-    default_multicast_address,
 )
-from infuse_iot.util.argparse import ValidDir
+from infuse_iot.util.argparse import ValidDir, add_server_port_parser
 
 
 class DeviceState:
@@ -65,7 +64,7 @@ class SubCommand(InfuseCommand):
     DESCRIPTION = "Synchronise data logger state from remote devices"
 
     def __init__(self, args):
-        self._client = LocalClient(default_multicast_address(), 1.0)
+        self._client = LocalClient(args.server_sock, 1.0)
         self._min_rssi: int | None = args.rssi
         self._app = args.app
         self._out = args.out
@@ -112,6 +111,8 @@ def add_parser(cls, parser):
             help="Synchronise removable loggers",
         )
 
+        add_server_port_parser(parser)
+
     def progress_table(self):
         table = Table()
         table.add_column("Device ID")

src/infuse_iot/tools/gateway.py

@@ -51,9 +51,8 @@
     GatewayRequestEpacketSend,
     GatewayRequestObservedDevices,
     LocalServer,
-    default_multicast_address,
 )
-from infuse_iot.util.argparse import ValidFile
+from infuse_iot.util.argparse import ValidFile, add_server_port_parser
 from infuse_iot.util.console import Console
 from infuse_iot.util.os import is_wsl
 from infuse_iot.util.threading import SignaledThread
@@ -529,7 +528,7 @@ def add_parser(cls, parser):
         )
         parser.add_argument("--baud", type=int, default=115200, help="Baudrate for serial port")
         parser.add_argument("--root", type=ValidFile, help="Root identity certificate to use instead of cloud")
-        parser.add_argument("--server-port", type=int, help="Alternate multicast port to use")
+        add_server_port_parser(parser)
 
     def __init__(self, args: argparse.Namespace):
         self.port: SerialLike
@@ -545,8 +544,7 @@ def __init__(self, args: argparse.Namespace):
         if args.display_only:
             self.server = None
         else:
-            addr = default_multicast_address(args.server_port) if args.server_port else default_multicast_address()
-            self.server = LocalServer(addr)
+            self.server = LocalServer(args.server_sock)
         self.rpc_server = LocalRpcServer(self.ddb)
         self._common = CommonThreadState(self.server, self.port, self.ddb, self.rpc_server)
         self.log = args.log

src/infuse_iot/tools/localhost.py

@@ -26,10 +26,10 @@
 from infuse_iot.socket_comms import (
     ClientNotificationEpacketReceived,
     LocalClient,
-    default_multicast_address,
 )
 from infuse_iot.tdf import TDF
 from infuse_iot.time import InfuseTime
+from infuse_iot.util.argparse import add_server_port_parser
 from infuse_iot.util.console import Console
 from infuse_iot.util.threading import SignaledThread
 
@@ -42,6 +42,7 @@ class SubCommand(InfuseCommand):
     @classmethod
     def add_parser(cls, parser):
         parser.add_argument("--port", "-p", type=int, default=8080, help="Port number for localhost server")
+        add_server_port_parser(parser)
 
     def __init__(self, args):
         self._data_lock = threading.Lock()
@@ -51,7 +52,7 @@ def __init__(self, args):
         self._data: dict[int, dict] = {}
         self._port: int = args.port
 
-        self._client = LocalClient(default_multicast_address(), 1.0)
+        self._client = LocalClient(args.server_sock, 1.0)
         self._decoder = TDF()
 
     # Serve the HTML file

src/infuse_iot/tools/native_bt.py

@@ -43,9 +43,8 @@
     GatewayRequestConnectionRequest,
     GatewayRequestEpacketSend,
     LocalServer,
-    default_multicast_address,
 )
-from infuse_iot.util.argparse import BtLeAddress, ValidFile
+from infuse_iot.util.argparse import BtLeAddress, ValidFile, add_server_port_parser
 from infuse_iot.util.console import Console
 
 
@@ -212,11 +211,12 @@ class SubCommand(InfuseCommand):
     @classmethod
     def add_parser(cls, parser):
         parser.add_argument("--root", type=ValidFile, help="Root identity certificate to use instead of cloud")
+        add_server_port_parser(parser)
 
     def __init__(self, args: argparse.Namespace):
         self.infuse_manu = 0x0DE4
         self.database = DeviceDatabase(args.root)
-        self.server = LocalServer(default_multicast_address())
+        self.server = LocalServer(args.server_sock)
         self.bleak_mapping: dict[int, BLEDevice] = {}
         self.unknown_networks: set[int] = set()
         Console.init()

src/infuse_iot/tools/ota_upgrade.py

@@ -28,9 +28,8 @@
 from infuse_iot.socket_comms import (
     GatewayRequestConnectionRequest,
     LocalClient,
-    default_multicast_address,
 )
-from infuse_iot.util.argparse import InfuseDeviceId, ValidFile, ValidRelease
+from infuse_iot.util.argparse import InfuseDeviceId, ValidFile, ValidRelease, add_server_port_parser
 from infuse_iot.util.crc import crc16_ccitt
 from infuse_iot.zephyr.errno import errno
 
@@ -41,7 +40,7 @@ class SubCommand(InfuseCommand):
     DESCRIPTION = "Automatically OTA upgrade observed devices"
 
     def __init__(self, args):
-        self._client = LocalClient(default_multicast_address(), 1.0)
+        self._client = LocalClient(args.server_sock, 1.0)
         self._conn_timeout = args.conn_timeout
         self._min_rssi: int | None = args.rssi
         self._explicit_ids: list[int] = []
@@ -107,6 +106,8 @@ def add_parser(cls, parser):
         explicit.add_argument("--id", type=InfuseDeviceId, help="Single device to upgrade")
         explicit.add_argument("--list", type=ValidFile, help="File containing a list of IDs to upgrade")
 
+        add_server_port_parser(parser)
+
     def progress_table(self):
         table = Table()
         table.add_column(f"{self._app_name}\n{self._new_ver}")

src/infuse_iot/tools/rpc.py

@@ -20,9 +20,8 @@
     ClientNotificationEpacketReceived,
     GatewayRequestConnectionRequest,
     LocalClient,
-    default_multicast_address,
 )
-from infuse_iot.util.argparse import InfuseDeviceId
+from infuse_iot.util.argparse import InfuseDeviceId, add_server_port_parser
 
 
 class SubCommand(InfuseCommand):
@@ -56,9 +55,11 @@ def add_parser(cls, parser):
             cmd_parser.set_defaults(rpc_class=cmd_cls)
             cmd_cls.add_parser(cmd_parser)
 
+        add_server_port_parser(parser)
+
     def __init__(self, args: argparse.Namespace):
         self._args = args
-        self._client = LocalClient(default_multicast_address(), 1.0)
+        self._client = LocalClient(args.server_sock, 1.0)
         self._command: InfuseRpcCommand = args.rpc_class(args)
         self._request_id = random.randint(0, 2**32 - 1)
         self._max_payload = 0