49 lines
1.4 KiB
Python
49 lines
1.4 KiB
Python
import uvicorn
|
|
from fastapi import FastAPI
|
|
|
|
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:
|
|
"""
|
|
Creates FastAPI object and runs integrate_onion
|
|
:param port: Must be same with port on which uvicorn is running
|
|
:param name: Name of service
|
|
:return: FastAPI object with onion.cleanup function on shutdown
|
|
"""
|
|
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 | None = get_available_port()):
|
|
"""
|
|
Runs service with specified name and starts onion
|
|
:param name: Name of service
|
|
:param port: Port where to start service, if not specified - gets random available
|
|
:return:
|
|
"""
|
|
if port is None:
|
|
port = get_available_port()
|
|
app = get_app(port, name)
|
|
app.include_router(router)
|
|
uvicorn.run(app, host="0.0.0.0", port=port)
|
|
|
|
|
|
def run_without_onion(name: str, port: int | None = get_available_port()):
|
|
if port is None:
|
|
port = get_available_port()
|
|
|
|
app = FastAPI(
|
|
title=f"dragonion-server: {name}",
|
|
description=f"Secure dragonion chat endpoint server - service {name}",
|
|
)
|
|
app.include_router(router)
|
|
uvicorn.run(app, host="0.0.0.0", port=port)
|