#!/usr/bin/env python3

import uvicorn
from contextlib import asynccontextmanager
from fastapi import FastAPI
from fastapi.middleware.cors import CORSMiddleware
from fastapi.staticfiles import StaticFiles
from opentelemetry import trace
from opentelemetry.exporter.otlp.proto.http.trace_exporter import OTLPSpanExporter
from opentelemetry.instrumentation.fastapi import FastAPIInstrumentor
from opentelemetry.sdk.resources import Resource, SERVICE_NAME
from opentelemetry.sdk.trace import TracerProvider
from opentelemetry.sdk.trace.export import BatchSpanProcessor

from backend.db import db
from dependencies import idfm_interface, settings
from routers import line, stop


@asynccontextmanager
async def lifespan(app: FastAPI):
    await db.connect(settings.db, settings.clear_static_data)
    if settings.clear_static_data:
        await idfm_interface.startup()

    yield

    await db.disconnect()


app = FastAPI(lifespan=lifespan)

app.add_middleware(
    CORSMiddleware,
    allow_origins=["https://localhost:4443", "https://localhost:3000"],
    allow_credentials=True,
    allow_methods=["*"],
    allow_headers=["*"],
)

app.mount("/widget", StaticFiles(directory="../frontend/", html=True), name="widget")

app.include_router(line.router)
app.include_router(stop.router)


FastAPIInstrumentor.instrument_app(app)

trace.set_tracer_provider(
    TracerProvider(resource=Resource.create({SERVICE_NAME: settings.app_name}))
)
trace.get_tracer_provider().add_span_processor(BatchSpanProcessor(OTLPSpanExporter()))
tracer = trace.get_tracer(settings.app_name)


if __name__ == "__main__":
    http_settings = settings.http
    uvicorn.run(
        app,
        host=http_settings.host,
        port=http_settings.port,
        ssl_certfile=http_settings.cert,
    )