refactor classes into respective files

meshtastic/tcp_interface.py

@@ -0,0 +1,50 @@
+import logging
+import socket
+from typing import AnyStr
+
+from .stream_interface import StreamInterface
+
+class TCPInterface(StreamInterface):
+    """Interface class for meshtastic devices over a TCP link"""
+
+    def __init__(self, hostname: AnyStr, debugOut=None, noProto=False, connectNow=True, portNumber=4403):
+        """Constructor, opens a connection to a specified IP address/hostname
+
+        Keyword Arguments:
+            hostname {string} -- Hostname/IP address of the device to connect to
+        """
+
+        logging.debug(f"Connecting to {hostname}")
+
+        server_address = (hostname, portNumber)
+        sock = socket.create_connection(server_address)
+
+        # Instead of wrapping as a stream, we use the native socket API
+        # self.stream = sock.makefile('rw')
+        self.stream = None
+        self.socket = sock
+
+        StreamInterface.__init__(
+            self, debugOut=debugOut, noProto=noProto, connectNow=connectNow)
+
+    def close(self):
+        """Close a connection to the device"""
+        logging.debug("Closing TCP stream")
+        StreamInterface.close(self)
+        # Sometimes the socket read might be blocked in the reader thread.  Therefore we force the shutdown by closing
+        # the socket here
+        self._wantExit = True
+        if not self.socket is None:
+            try:
+                self.socket.shutdown(socket.SHUT_RDWR)
+            except:
+                pass  # Ignore errors in shutdown, because we might have a race with the server
+            self.socket.close()
+
+    def _writeBytes(self, b):
+        """Write an array of bytes to our stream and flush"""
+        self.socket.send(b)
+
+    def _readBytes(self, len):
+        """Read an array of bytes from our stream"""
+        return self.socket.recv(len)