remove useless logs

Signed-off-by: Frank Villaro-Dixon <frank@villaro-dixon.eu>

README.md

@@ -27,6 +27,46 @@ You'll need to edit the kubeconfig file and change the api host to be your local
 
 
 
+## How to use it
+
+### Pod
+
+Launch a pod (through a deployment, or sts, or something else) on your cluster.
+You can use the following image `forge.k3s.fr/frank/kube-escape:latest`
+
+Don't forget to give it the following env values:
+
+- WEBSOCKET_ROOT_URL
+- WS_ID: facultative, can auto generate itself
+
+Then look at its logs and you'll see the ws url to use when connecting to it
+
+### WS Proxy
+
+You should spawn a WS proxy that will receive connections from the client and the pod.
+It should be accessible by both.
+
+You can override the command of the image and use `./proxy.py`
+
+### Client
+
+Launch your client with:
+```bash
+./client.py <ws_URL_given_by_pod>
+```
+
+This will open a listening socket on localhost port 6443
+
+### Kubectl
+
+Change your kubeconfig's server to `https://localhost:6443`
+
+And then, enjoy!
+
+
+## Considerations
+
+
 ### Security
 
 I guess you could proxy your websockets through an HTTPs endpoint. Wouldn't be bad.

pod.py

@@ -16,7 +16,7 @@ async def handle_socket_read(socketid, tcpreader, ws):
         print(f"New socket: {socketid}. Waiting on recv")
         while True:
             data = await tcpreader.read(2024)
-            print(f"TCP@{socketid} Received {len(data)} bytes")
+            #print(f"TCP@{socketid} Received {len(data)} bytes")
             if data == b'':
                 print(f"TCP@{socketid} Connection closed")
                 c = conn.WSMsg(socketid, conn.MsgType.DISCONNECT)
@@ -24,7 +24,7 @@ async def handle_socket_read(socketid, tcpreader, ws):
                 break
 
             c = conn.WSMsg(socketid, conn.MsgType.DATA, data)
-            print(f'TCP>WS: {c}')
+            #print(f'TCP>WS: {c}')
             await ws.send(c.to_bytes())
 
     except Exception as e:
@@ -62,9 +62,8 @@ async def handle_ws_incoming(cfg, ws, sockets):
 
     elif c.msg == conn.MsgType.DATA:
         tcpreader, tcpwriter = sockets[socketid]
-        print(f'WS>TCP: {c}')
+        #print(f'WS>TCP: {c}')
         tcpwriter.write(c.payload)
-        print('written')
 
 
 def get_config():

proxy.py

@@ -29,12 +29,12 @@ async def handler(websocket):
         print(f"Connection closed: {e}")
     finally:
         # Unregister the client
-        connected_clients.remove(websocket)
+        connected_clients.get(websocket.request.path, set()).remove(websocket)
 
 async def main():
     # Start the WebSocket server
     ws_port = os.environ.get("WS_PORT", 9999)
-    server = await websockets.asyncio.server.serve(handler, "::", ws_port)
+    server = await websockets.asyncio.server.serve(handler, "", ws_port)
     print(f"WebSocket server listening on ws://[::]:{ws_port}")
     await server.wait_closed()