feat: troppa roba davvero
This commit is contained in:
+21
-15
@@ -2,19 +2,22 @@ import os
|
||||
import sys
|
||||
from contextlib import asynccontextmanager
|
||||
|
||||
from api.auth.router import router as auth_router
|
||||
from api.editor.router import router as editor_router
|
||||
from api.proxy.router import router as proxy_router
|
||||
from core.config import settings
|
||||
from core.mongodb import close_mongodb, init_mongodb
|
||||
from core.security import hash_password
|
||||
from fastapi import FastAPI
|
||||
from integrations.docker import get_all_services_in_dir
|
||||
from integrations.docker.discovery import service_scan_root
|
||||
from repositories.auth import ensure_auth_indexes, ensure_single_account
|
||||
from repositories.examples import ensure_example_indexes
|
||||
from repositories.proxy import ensure_proxy_indexes
|
||||
from services.mitmproxy.recovery import reconcile_runtime
|
||||
|
||||
from app.api.auth.router import router as auth_router
|
||||
from app.api.editor.router import router as editor_router
|
||||
from app.api.proxy.router import router as proxy_router
|
||||
from app.core.config import settings
|
||||
from app.core.mongodb import close_mongodb, init_mongodb
|
||||
from app.core.security import hash_password
|
||||
from app.integrations.docker import get_all_services_in_dir
|
||||
from app.integrations.docker.discovery import service_scan_root
|
||||
from app.repositories.auth import ensure_auth_indexes, ensure_single_account
|
||||
from app.repositories.health import ensure_health_indexes
|
||||
from app.repositories.proxy import ensure_proxy_indexes
|
||||
from app.repositories.templates import ensure_template_indexes
|
||||
from app.services.mitmproxy.health_monitor import health_monitor
|
||||
from app.services.mitmproxy.recovery import reconcile_runtime
|
||||
|
||||
|
||||
@asynccontextmanager
|
||||
@@ -26,7 +29,8 @@ async def lifespan(_: FastAPI):
|
||||
password_hash=hash_password(settings.AUTH_PASSWORD),
|
||||
)
|
||||
await ensure_proxy_indexes()
|
||||
await ensure_example_indexes()
|
||||
await ensure_template_indexes()
|
||||
await ensure_health_indexes()
|
||||
recovered = await reconcile_runtime()
|
||||
if recovered:
|
||||
recovered_count = len(recovered)
|
||||
@@ -42,8 +46,8 @@ async def lifespan(_: FastAPI):
|
||||
)
|
||||
|
||||
scan_root = service_scan_root()
|
||||
app.state.services = get_all_services_in_dir(scan_root) or []
|
||||
if not app.state.services:
|
||||
services = get_all_services_in_dir(scan_root) or []
|
||||
if not services:
|
||||
warning = (
|
||||
"\n" + "!" * 60 + "\n"
|
||||
" WARNING: No services found in scan root!\n"
|
||||
@@ -52,8 +56,10 @@ async def lifespan(_: FastAPI):
|
||||
print(warning, file=sys.stderr)
|
||||
|
||||
try:
|
||||
await health_monitor.start()
|
||||
yield
|
||||
finally:
|
||||
await health_monitor.stop()
|
||||
await close_mongodb()
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user