from fastapi import APIRouter
from .. import database

router = APIRouter()

@router.get("/health")
async def health_check():
    result = {"oracle": "error", "sqlite": "error"}

    # Check Oracle
    try:
        if database.pool:
            with database.pool.acquire() as conn:
                with conn.cursor() as cur:
                    cur.execute("SELECT SYSDATE FROM DUAL")
                    cur.fetchone()
            result["oracle"] = "ok"
    except Exception as e:
        result["oracle"] = str(e)

    # Check SQLite
    try:
        db = await database.get_sqlite()
        await db.execute("SELECT 1")
        await db.close()
        result["sqlite"] = "ok"
    except Exception as e:
        result["sqlite"] = str(e)

    return result