websocket: inject messages via flow

1 files changed, 23 insertions, 0 deletions

diff --git a/examples/complex/websocket_inject_message.py b/examples/complex/websocket_inject_message.py
new file mode 100644
index 00000000..e9c3ea0c
--- /dev/null
+++ b/examples/complex/websocket_inject_message.py
@@ -0,0 +1,23 @@
+"""
+This example shows how to inject a WebSocket message to the client.
+Every new WebSocket connection will trigger a new asyncio task that
+periodically injects a new message to the client.
+"""
+import asyncio
+import mitmproxy.websocket
+
+
+class InjectWebSocketMessage:
+
+    async def inject(self, flow: mitmproxy.websocket.WebSocketFlow):
+        i = 0
+        while not flow.ended and not flow.error:
+            await asyncio.sleep(5)
+            flow.inject_message(flow.client_conn, 'This is the #{} an injected message!'.format(i))
+            i += 1
+
+    def websocket_start(self, flow):
+        asyncio.get_event_loop().create_task(self.inject(flow))
+
+
+addons = [InjectWebSocketMessage()]