wip

terminus/app/consumers.py

@@ -0,0 +1,31 @@
+import asyncio
+from channels.generic.websocket import AsyncWebsocketConsumer
+
+
+class TCPConsumer(AsyncWebsocketConsumer):
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+        self.loop = asyncio.get_event_loop()
+
+    async def connect(self):
+        self.reader, self.writer = await asyncio.open_connection('192.168.78.233', 22)
+        self._socket_reader = self.loop.create_task(self.socket_reader())
+        await self.accept()
+
+    async def disconnect(self, close_code):
+        await self.writer.drain()
+        self.writer.close()
+        await self._socket_reader
+
+    async def receive(self, bytes_data):
+        self.writer.write(bytes_data)
+
+    async def socket_reader(self):
+        while True:
+            await self.reader._wait_for_data('read')
+            data = bytes(self.reader._buffer.copy())
+            if not data:
+                await self.close()
+                return
+            del self.reader._buffer[:]
+            await self.send(bytes_data=data)