2 files changed, 5 insertions, 7 deletions

diff --git a/ttun_server/__init__.py b/ttun_server/__init__.py
index d54227a..4af6ca0 100644
--- a/ttun_server/__init__.py
+++ b/ttun_server/__init__.py
@@ -2,10 +2,11 @@ import logging
 import os
 
 from fastapi import FastAPI
-from starlette.routing import Host, Route, WebSocketRoute
+from starlette.routing import Host, Route, WebSocketRoute, Mount
 
+from authentication.endpoints import authentication, oauth
 from proxy import app as proxy_app
-from ttun_server.endpoints import health, base_endpoints
+from ttun_server.endpoints import health
 from ttun_server.websockets import tunnel
 
 logging.basicConfig(level=getattr(logging, os.environ.get('LOG_LEVEL', 'INFO')))
@@ -13,6 +14,8 @@ logging.basicConfig(level=getattr(logging, os.environ.get('LOG_LEVEL', 'INFO')))
 app = FastAPI(
     debug=True,
     routes=[
+        Mount('/auth/', app=authentication),
+        Mount('/oauth/', app=oauth),
         WebSocketRoute('/tunnel/', endpoint=tunnel),
         Route('/health/', endpoint=health),
         Host(f'{{subdomain}}.{os.environ['TUNNEL_DOMAIN']}', app=proxy_app)
diff --git a/ttun_server/endpoints.py b/ttun_server/endpoints.py
index 51c17aa..61e3f21 100644
--- a/ttun_server/endpoints.py
+++ b/ttun_server/endpoints.py
@@ -1,7 +1,2 @@
-from fastapi import FastAPI
-
-base_endpoints = FastAPI()
-
-@base_endpoints.get('/health/')
 async def health():
     return 'OK'