arm-core/api/src/app.py

"""Точка входа FastAPI. Каркас из шаблона команды: инфраструктура без доменного кода.

Что здесь есть: настройки, подключение S3 (MinIO), NATS (`src/core/nats.py`: core
NATS без JetStream, подключение в фоне при NATS_ENABLED=true независимо от DEBUG, подписка `recognition.result` в queue
group `arm-core`), SMB (горячие папки ИС Сканирование), middleware (gzip, proxy headers, CORS, РАИСА при DEBUG=False),
healthcheck (uptime, version, nats: ok|down|off), раздача собранного SPA.
Недоступность S3 или NATS при старте не роняет и не задерживает приложение: ошибка логируется, healthcheck показывает
состояние. Outbox-паблишер и вахтёр сроков — задача 23 (docs/context/03-architecture.md).
"""

import os
from contextlib import asynccontextmanager
from datetime import UTC, datetime
from http import HTTPStatus

import smbclient
from fastapi import FastAPI, Request
from fastapi.exceptions import HTTPException, StarletteHTTPException
from fastapi.middleware.cors import CORSMiddleware
from fastapi.staticfiles import StaticFiles
from loguru import logger
from miniopy_async import Minio
from starlette.middleware.gzip import GZipMiddleware
from uvicorn.middleware.proxy_headers import ProxyHeadersMiddleware

from src.api.router import API_PREFIX, router
from src.core.clients import S3Storage
from src.core.config import get_settings
from src.core.exception_handlers import register_exception_handlers
from src.core.nats import NatsConnection
from src.services.recognition_exchange import handle_message

settings = get_settings()

VERSION = settings.VERSION
UP_TIME = datetime.now(UTC)


# --- РАИСА: защищённый API (только вне DEBUG) ---------------------------------------------------
protected_middleware_available = False
if not settings.DEBUG:
    try:
        from raisa_fastapi_protected_api import (  # noqa: F401  # pyright: ignore[reportMissingImports]
            AuthorizationMiddleware,
            OpenEndpoint,
            ProtectedOAuthSettings,
            ProtectedRolesSettings,
            ProtectedSettings,
            SearchType,
        )

        protected_middleware_available = True
    except ImportError:
        protected_middleware_available = False


def _normalize_root_path(value: str) -> str:
    if not value:
        return ""
    value = value.strip()
    if value == "/":
        return ""
    return "/" + value.strip("/")


_APP_ROOT_PATH = _normalize_root_path(settings.ROOT_PATH)


async def _connect_s3() -> S3Storage | None:
    """Клиент S3 с проверкой бакета; при недоступности — ``None`` (приложение всё равно стартует)."""
    s3_client = Minio(
        settings.OPENBAO.MINIO.MINIO_URL,
        access_key=settings.OPENBAO.MINIO.MINIO_ADMIN_LOGIN,
        secret_key=settings.OPENBAO.MINIO.MINIO_ADMIN_PASSWORD,
        cert_check=False,
    )
    try:
        bucket_exists = await s3_client.bucket_exists(settings.OPENBAO.MINIO.MINIO_BUCKET)
    except Exception as exc:
        logger.exception(
            "Не удалось проверить наличие бакета. error_category=s3_error, error_type={error_type}, error={error}",
            error_type=type(exc).__name__,
            error=repr(exc),
        )
        await s3_client.close_session()
        return None
    logger.info("Бакет проверен. bucket_exists={bucket_exists}", bucket_exists=bucket_exists)
    return S3Storage(client=s3_client, bucket_name=settings.OPENBAO.MINIO.MINIO_BUCKET)


async def _subscribe_results(nats: NatsConnection) -> None:
    """Подписки после первого подключения к NATS: `recognition.result` в queue group `arm-core`."""
    await nats.subscribe_results(handle_message)


def _create_nats() -> NatsConnection | None:
    """Соединение NATS при NATS_ENABLED=true; при false — ``None`` (healthcheck покажет ``off``)."""
    if not settings.NATS_ENABLED:
        logger.info("NATS выключен (NATS_ENABLED=false)")
        return None
    return NatsConnection(settings.OPENBAO.NATS, on_connected=_subscribe_results)


@asynccontextmanager
async def lifespan(app: FastAPI):
    app.state.storage = None
    app.state.nats = None

    try:
        smbclient.register_session(
            settings.OPENBAO.SMB.SMB_HOST, settings.OPENBAO.SMB.SMB_LOGIN, settings.OPENBAO.SMB.SMB_PASSWORD
        )
    except Exception:
        logger.exception("Ошибка при регистрации сессии SMB")

    try:
        app.state.storage = await _connect_s3()
        app.state.nats = _create_nats()
        if app.state.nats is not None:
            await app.state.nats.start()  # подключение в фоне: старт приложения не ждёт NATS

        logger.info("Приложение запущено")
        yield
    finally:
        logger.info("Приложение закрывается")
        smbclient.reset_connection_cache()
        if app.state.nats is not None:
            await app.state.nats.stop()
        if app.state.storage:
            await app.state.storage.client.close_session()


app = FastAPI(title="АРМ КА", version=VERSION, root_path=settings.ROOT_PATH, lifespan=lifespan)


@app.get("/healthcheck", status_code=200)
async def healthcheck(request: Request):
    """Живость приложения. ``nats``: ``off`` — NATS выключен, ``ok`` — подключён, ``down`` — включён, но нет связи."""
    nats_status = "off"
    if settings.NATS_ENABLED:
        nc = getattr(request.app.state, "nats", None)
        nats_status = "ok" if nc is not None and nc.is_connected else "down"
    return {"uptime": UP_TIME, "version": VERSION, "nats": nats_status}


app.include_router(router)

register_exception_handlers(app)

app.add_middleware(GZipMiddleware, minimum_size=1000, compresslevel=1)
app.add_middleware(ProxyHeadersMiddleware, trusted_hosts=["*"])
app.add_middleware(
    CORSMiddleware,
    allow_origins=["*"],
    allow_credentials=True,
    allow_methods=["*"],
    allow_headers=["*"],
)


# --- Раздача собранного SPA (web/) ---------------------------------------------------------------
os.makedirs("web", exist_ok=True)


def sep_static(path: str, stat_dir: str) -> str:
    sep_path = path.split("/")
    index = sep_path.index(stat_dir)
    return "/".join(sep_path[index:])


API_STATIC_PATH = API_PREFIX.strip("/")


class SPAStaticFiles(StaticFiles):
    async def get_response(self, path: str, scope):
        # Несуществующий маршрут API — честный 404 (в т.ч. для POST), а не index.html SPA и не 405 от StaticFiles:
        # иначе dev-маршруты, которых при выключенных флагах нет, выглядели бы как «существующие»
        if path == API_STATIC_PATH or path.startswith(API_STATIC_PATH + "/"):
            raise HTTPException(status_code=HTTPStatus.NOT_FOUND)
        try:
            return await super().get_response(path, scope)
        except (HTTPException, StarletteHTTPException) as ex:
            if ex.status_code == HTTPStatus.NOT_FOUND:
                if "images" in path:
                    return await super().get_response(sep_static(path, "images"), scope)
                if "static" in path:
                    return await super().get_response(sep_static(path, "static"), scope)
                return await super().get_response("index.html", scope)
            raise


app.mount("/", SPAStaticFiles(directory="web", html=True), name="web")