Starting server and forwarding it to onion works

dragonion_server/modules/server/__init__.py

@@ -0,0 +1 @@
+from .server import run

dragonion_server/modules/server/integration.py

@@ -0,0 +1,9 @@
+from dragonion_server.utils.onion import Onion
+
+
+def integrate_onion(port: int, name: str) -> Onion:
+    onion = Onion()
+    onion.connect()
+    onion.write_onion_service(name, port)
+    print(f'Available on {onion.start_onion_service(name)}')
+    return onion

dragonion_server/modules/server/routes.py

@@ -0,0 +1,10 @@
+from fastapi import APIRouter, Request
+from fastapi.responses import PlainTextResponse
+
+
+router = APIRouter()
+
+
+@router.get("/", response_model=str)
+async def root(request: Request):
+    return PlainTextResponse("dragonion-server")

dragonion_server/modules/server/server.py

@@ -0,0 +1,20 @@
+from fastapi import FastAPI
+import uvicorn
+from dragonion_server.utils.onion import get_available_port
+from .integration import integrate_onion
+from .routes import router
+
+
+def get_app(port: int, name: str) -> FastAPI:
+    onion = integrate_onion(port, name)
+    return FastAPI(
+        title=f'dragonion-server: {name}',
+        description=f'Secure dragonion chat endpoint server - service {name}',
+        on_shutdown=[onion.cleanup]
+    )
+
+
+def run(name: str, port: int = get_available_port()):
+    app = get_app(port, name)
+    app.include_router(router)
+    uvicorn.run(app, host='0.0.0.0', port=port)