tools: rpc: timeouts from RPC wrapper

Allow RPC wrappers to overwrite the default 10 seconds timeout.

Signed-off-by: Jordan Yates <jordan@embeint.com>

Author: JordanYates

src/infuse_iot/commands.py

@@ -60,6 +60,10 @@ def auth_level(self) -> Auth:
         """Authentication level to run command with"""
         return Auth.DEVICE
 
+    def command_timeout_ms(self) -> int:
+        """Duration to wait for the RPC response in milliseconds"""
+        return 10000
+
     def request_struct(self) -> ctypes.LittleEndianStructure | bytes:
         """RPC_CMD request structure"""
         raise NotImplementedError

src/infuse_iot/tools/rpc.py

@@ -85,6 +85,7 @@ def run(self):
             with self._client.connection(self._id, types, self._args.conn_timeout) as mtu:
                 self._max_payload = mtu
                 rpc_client = RpcClient(self._client, mtu, self._id, self.rx_handler)
+                rpc_client.set_timeout(self._command.command_timeout_ms())
                 params = bytes(self._command.request_struct())
 
                 if hasattr(self._command.response, "vla_from_buffer_copy"):  # type: ignore