From 486087cdb349dbc07b479d2286a02bdca310ea38 Mon Sep 17 00:00:00 2001 From: Tom van der Lee Date: Wed, 20 Mar 2024 21:48:45 +0100 Subject: Added websocket support --- ttun_server/__init__.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) (limited to 'ttun_server/__init__.py') diff --git a/ttun_server/__init__.py b/ttun_server/__init__.py index 81f8cd4..2f8fed0 100644 --- a/ttun_server/__init__.py +++ b/ttun_server/__init__.py @@ -4,7 +4,8 @@ import os from starlette.applications import Starlette from starlette.routing import Route, WebSocketRoute, Host, Router -from ttun_server.endpoints import Proxy, Tunnel, Health +from ttun_server.endpoints import Proxy, Health +from .websockets import WebsocketProxy, Tunnel logging.basicConfig(level=getattr(logging, os.environ.get('LOG_LEVEL', 'INFO'))) @@ -18,6 +19,7 @@ server = Starlette( routes=[ Host(os.environ['TUNNEL_DOMAIN'], base_router, 'base'), Route('/{path:path}', Proxy), + WebSocketRoute('/{path:path}', WebsocketProxy) ] ) -- cgit v1.2.3