ScanOS/backend/app/main.py

import threading, logging
from contextlib import asynccontextmanager
from typing import Annotated

from fastapi import FastAPI, Depends

from app.data import models
from app.data.database import SessionLocal, engine

from uvicorn.logging import DefaultFormatter

from app.scanner.scanner import Scanner
from app.scanner.scanner import Status as ScannerStatus

# Set up logging
logger = logging.getLogger()
__syslog = logging.StreamHandler()
__syslog.setFormatter(DefaultFormatter(fmt="%(levelprefix)s %(message)s", use_colors=True))
logger.setLevel(logging.INFO)
logger.addHandler(__syslog)

# Create database
models.Base.metadata.create_all(bind=engine)

# Set up scanner instance
__scanner = Scanner("/var/www/html/img", logger)

# Preload scanner after FastAPI start
@asynccontextmanager
async def __lifespan(app: FastAPI):
    threading.Thread(target=__scanner.preload).start()
    yield

app = FastAPI(lifespan=__lifespan)

# SQLAlchemiy session dependency
def get_db():
    db = SessionLocal()
    try:
        yield db
    finally:
        db.close()

def get_scanner():
    return __scanner

from app.routers import power
app.include_router(power.router)

from app.routers import scan
app.include_router(scan.router)

@app.get("/api/ready")
async def __ready(scanner: Annotated[Scanner, Depends(get_scanner)]):
    return scanner.get_status() != ScannerStatus.INITIALIZED