Merge pull request #5 from paneidos/healthcheckv1.2.0

Add healthcheck endpoint

2 files changed, 16 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)
     ]
 )
diff --git a/ttun_server/endpoints.py b/ttun_server/endpoints.py
index b33fe65..b25ffe4 100644
--- a/ttun_server/endpoints.py
+++ b/ttun_server/endpoints.py
@@ -55,6 +55,14 @@ class Proxy(HTTPEndpoint):
         await response(self.scope, self.receive, self.send)
 
 
+class Health(HTTPEndpoint):
+    async def get(self, _) -> None:
+        response = Response(content='OK', status_code=200)
+
+        await response(self.scope, self.receive, self.send)
+
+
+
 class Tunnel(WebSocketEndpoint):
     encoding = 'json'