1 files changed, 8 insertions, 3 deletions

diff --git a/ttun_server/__init__.py b/ttun_server/__init__.py
index cf589cc..17a8e7a 100644
--- a/ttun_server/__init__.py
+++ b/ttun_server/__init__.py
@@ -2,16 +2,21 @@ import logging
 import os
 
 from starlette.applications import Starlette
-from starlette.routing import Route, WebSocketRoute
+from starlette.routing import Route, WebSocketRoute, Host, Router
 
-from ttun_server.endpoints import Proxy, Tunnel
+from ttun_server.endpoints import Proxy, Tunnel, Health
 
 logging.basicConfig(level=getattr(logging, os.environ.get('LOG_LEVEL', 'INFO')))
 
+base_router = Router(routes=[
+    Route('/health/', Health),
+    WebSocketRoute('/tunnel/', Tunnel)
+])
+
 server = Starlette(
     debug=True,
     routes=[
+        Host(os.environ['TUNNEL_DOMAIN'], base_router, 'base'),
         Route('/{path:path}', Proxy),
-        WebSocketRoute('/tunnel/', Tunnel)
     ]
 )