From 12f2e24e2154113a6329d74aa556ae23506c34e1 Mon Sep 17 00:00:00 2001 From: Tom van der Lee Date: Tue, 30 Jun 2026 22:46:55 +0200 Subject: WIP --- ttun_server/__init__.py | 15 +++++++++------ 1 file changed, 9 insertions(+), 6 deletions(-) (limited to 'ttun_server/__init__.py') diff --git a/ttun_server/__init__.py b/ttun_server/__init__.py index 2f8fed0..6c77858 100644 --- a/ttun_server/__init__.py +++ b/ttun_server/__init__.py @@ -1,28 +1,31 @@ import logging import os -from starlette.applications import Starlette -from starlette.routing import Route, WebSocketRoute, Host, Router +from fastapi import FastAPI +from starlette.routing import Host, Route, Router, WebSocketRoute -from ttun_server.endpoints import Proxy, Health +from ttun_server.endpoints import health, proxy from .websockets import WebsocketProxy, Tunnel logging.basicConfig(level=getattr(logging, os.environ.get('LOG_LEVEL', 'INFO'))) base_router = Router(routes=[ - Route('/health/', Health), + Route('/health/', health), WebSocketRoute('/tunnel/', Tunnel) ]) -server = Starlette( +server = FastAPI( debug=True, routes=[ Host(os.environ['TUNNEL_DOMAIN'], base_router, 'base'), - Route('/{path:path}', Proxy), + Route('/{path:path}', proxy), WebSocketRoute('/{path:path}', WebsocketProxy) ] ) +server.post() + + try: from ._version import version __version__ = version -- cgit v1.2.3