WIPv3

1 files changed, 9 insertions, 6 deletions

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