from fastapi import APIRouter, Request, Form, Depends
from fastapi.responses import RedirectResponse, JSONResponse
from app.database import get_connection
from app.auth import get_current_user
from app.models import QueryCreate

router = APIRouter(prefix="/queries", tags=["queries"])

QUERY_DEFAULTS = [
    {
        "name": "Tercero - Datos del paciente",
        "query_type": "terceros",
        "query_text": """SELECT
    p.CODIGO,
    p.TIPOIDENT,
    p.DOCIDENT,
    p.NOMBRES,
    p.APELLIDOS,
    p.DIRECCION,
    p.CIUDAD        AS COD_CIUDAD,
    c.NOMBRE        AS NOM_CIUDAD,
    p.TELEFONOS,
    p.EMAIL,
    p.F_NACIMIENTO,
    p.SEXO,
    p.TIPORES,
    p.CODETNIA
FROM PACIENTE p
LEFT JOIN CIUDAD c ON c.CODIGO = p.CIUDAD
WHERE p.DOCIDENT = :doc_num""",
        "description": "Datos del paciente por número de documento (para Tercero/Crear)"
    },
    {
        "name": "RDA Paciente por factura",
        "query_type": "transaccion",
        "query_text": """SELECT
    r.IDRECEPCION,
    r.PREFIJO,
    r.NUM_FACTURA,
    r.FECHA_RECEPCION,
    r.COD_PACIENTE,
    r.NIT_EMPRESA,
    r.DIAG_PPAL,
    r.TIPOUSU,
    r.TIPOUSUSISPRO,
    r.AUTORIZACION,
    r.CLASEPROC,
    r.HORAINICIORECEPCION,
    r.VALORTOTAL,
    rel.COD_EXAMEN,
    rel.PRECIO,
    rel.FECHA_REPORTADO,
    m.COD_ESPECIALIDAD,
    COALESCE(NULLIF(TRIM(m.CODIGO), ''), NULLIF(TRIM(r.USUARIO), ''), '') AS profesional,
    e.CODCONTRATO,
    TRIM(m.DOCIDMEDICO) AS DOCIDMEDICO
FROM RECEPCION r
JOIN RELACION rel ON rel.IDRECEPCION = r.IDRECEPCION
LEFT JOIN MEDICO m ON m.CODIGO = r.COD_MEDICO
LEFT JOIN EMPRESA e ON e.NIT = r.NIT_EMPRESA
WHERE r.NUM_FACTURA = :num_factura""",
        "description": "Servicios de una recepción por número de factura"
    },
    {
        "name": "RDA Paciente por fecha",
        "query_type": "transaccion",
        "query_text": """SELECT
    r.IDRECEPCION,
    r.PREFIJO,
    r.NUM_FACTURA,
    r.FECHA_RECEPCION,
    r.COD_PACIENTE,
    r.NIT_EMPRESA,
    r.DIAG_PPAL,
    r.TIPOUSU,
    r.TIPOUSUSISPRO,
    r.AUTORIZACION,
    r.CLASEPROC,
    r.HORAINICIORECEPCION,
    r.VALORTOTAL,
    rel.COD_EXAMEN,
    rel.PRECIO,
    rel.FECHA_REPORTADO,
    m.COD_ESPECIALIDAD,
    COALESCE(NULLIF(TRIM(m.CODIGO), ''), NULLIF(TRIM(r.USUARIO), ''), '') AS profesional,
    e.CODCONTRATO,
    TRIM(m.DOCIDMEDICO) AS DOCIDMEDICO
FROM RECEPCION r
JOIN RELACION rel ON rel.IDRECEPCION = r.IDRECEPCION
LEFT JOIN MEDICO m ON m.CODIGO = r.COD_MEDICO
LEFT JOIN EMPRESA e ON e.NIT = r.NIT_EMPRESA
WHERE r.FECHA_RECEPCION BETWEEN :fecha_ini AND :fecha_fin
ORDER BY r.IDRECEPCION""",
        "description": "Servicios en un rango de fechas (para RdaPaciente/Insertar)"
    },
]


def ensure_defaults():
    conn = get_connection()
    for q in QUERY_DEFAULTS:
        exists = conn.execute(
            "SELECT id FROM queries WHERE name = ?", (q["name"],)
        ).fetchone()
        if not exists:
            conn.execute(
                "INSERT INTO queries (name, query_type, query_text, description) VALUES (?, ?, ?, ?)",
                (q["name"], q["query_type"], q["query_text"], q["description"]),
            )
        else:
            conn.execute(
                "UPDATE queries SET query_text = ?, description = ? WHERE name = ?",
                (q["query_text"], q["description"], q["name"]),
            )
    conn.commit()
    conn.close()


@router.get("")
async def queries_page(request: Request, user: dict = Depends(get_current_user)):
    conn = get_connection()
    queries = conn.execute("SELECT * FROM queries ORDER BY query_type, name").fetchall()
    conn.close()
    return request.app.state.templates.TemplateResponse("queries.html", {
        "request": request, "user": user, "queries": queries
    })


@router.post("/create")
async def query_create(
    request: Request,
    user: dict = Depends(get_current_user),
    name: str = Form(...),
    query_type: str = Form(...),
    query_text: str = Form(...),
    description: str = Form(""),
):
    conn = get_connection()
    conn.execute(
        "INSERT INTO queries (name, query_type, query_text, description) VALUES (?, ?, ?, ?)",
        (name, query_type, query_text, description),
    )
    conn.commit()
    conn.close()
    return RedirectResponse("/queries", status_code=302)


@router.post("/update/{query_id}")
async def query_update(
    query_id: int,
    request: Request,
    user: dict = Depends(get_current_user),
    name: str = Form(...),
    query_text: str = Form(...),
    description: str = Form(""),
):
    conn = get_connection()
    conn.execute(
        "UPDATE queries SET name = ?, query_text = ?, description = ? WHERE id = ?",
        (name, query_text, description, query_id),
    )
    conn.commit()
    conn.close()
    return RedirectResponse("/queries", status_code=302)


@router.post("/run-sql")
async def run_sql(
    request: Request,
    sql: str = Form(...),
    user: dict = Depends(get_current_user),
):
    from app.services.firebird_service import get_firebird_from_config
    try:
        conn = get_connection()
        configs = {row["key"]: row["value"] for row in conn.execute("SELECT * FROM config").fetchall()}
        conn.close()

        fb, fb_success, fb_msg = get_firebird_from_config(configs)
        if not fb_success:
            return JSONResponse({"error": f"Error Firebird: {fb_msg}"})

        success, fb_err, rows = fb.execute_query(sql)
        fb.disconnect()

        if not success:
            return JSONResponse({"error": fb_err})

        limit = rows[:200] if rows else []
        return JSONResponse({
            "rows": limit,
            "total": len(rows),
            "columns": list(limit[0].keys()) if limit else []
        })
    except Exception as e:
        return JSONResponse({"error": f"Error interno: {e}"}, status_code=500)


@router.post("/delete/{query_id}")
async def query_delete(query_id: int, user: dict = Depends(get_current_user)):
    conn = get_connection()
    conn.execute("DELETE FROM queries WHERE id = ?", (query_id,))
    conn.commit()
    conn.close()
    return RedirectResponse("/queries", status_code=302)


@router.post("/esquema")
async def obtener_esquema(user: dict = Depends(get_current_user)):
    from app.services.firebird_service import get_firebird_from_config
    conn = get_connection()
    configs = {row["key"]: row["value"] for row in conn.execute("SELECT * FROM config").fetchall()}
    conn.close()

    fb, fb_success, fb_msg = get_firebird_from_config(configs)
    if not fb_success:
        return JSONResponse({"error": f"Error Firebird: {fb_msg}"}, status_code=400)

    success, fb_err, tablas = fb.execute_query("""
        SELECT RDB$RELATION_NAME as nombre
        FROM RDB$RELATIONS
        WHERE RDB$SYSTEM_FLAG = 0
          AND RDB$RELATION_NAME NOT LIKE 'RDB$%'
        ORDER BY 1
    """, {})

    if not success:
        fb.disconnect()
        return JSONResponse({"error": fb_err}, status_code=400)

    resultado = []
    for t in tablas:
        nombre = t["NOMBRE"].strip()
        ok, err, cols = fb.execute_query("""
            SELECT
                rf.RDB$FIELD_NAME as COLUMN_NAME,
                f.RDB$FIELD_TYPE as FIELD_TYPE,
                f.RDB$FIELD_LENGTH as FIELD_LENGTH
            FROM RDB$RELATION_FIELDS rf
            JOIN RDB$FIELDS f ON rf.RDB$FIELD_SOURCE = f.RDB$FIELD_NAME
            WHERE rf.RDB$RELATION_NAME = :nombre
            ORDER BY rf.RDB$FIELD_POSITION
        """, {"nombre": nombre})
        columnas = []
        if ok:
            for c in cols:
                tiponum = c["FIELD_TYPE"]
                tipos = {7: "SMALLINT", 8: "INTEGER", 10: "FLOAT", 12: "DATE", 13: "TIME",
                         14: "CHAR", 16: "BIGINT", 27: "DOUBLE", 35: "TIMESTAMP", 37: "VARCHAR",
                         40: "BLOB", 45: "BLOB_ID", 261: "BLOB"}
                columnas.append({
                    "nombre": c["COLUMN_NAME"].strip(),
                    "tipo": tipos.get(tiponum, f"UNKNOWN({tiponum})"),
                    "longitud": c["FIELD_LENGTH"]
                })
        resultado.append({"tabla": nombre, "columnas": columnas})

    fb.disconnect()
    return JSONResponse({"tablas": resultado})


@router.post("/tablas")
async def listar_tablas(user: dict = Depends(get_current_user)):
    from app.services.firebird_service import get_firebird_from_config
    conn = get_connection()
    configs = {row["key"]: row["value"] for row in conn.execute("SELECT * FROM config").fetchall()}
    conn.close()

    fb, fb_success, fb_msg = get_firebird_from_config(configs)
    if not fb_success:
        return JSONResponse({"error": f"Error Firebird: {fb_msg}"}, status_code=400)

    success, fb_err, rows = fb.execute_query("""
        SELECT RDB$RELATION_NAME as nombre
        FROM RDB$RELATIONS
        WHERE RDB$SYSTEM_FLAG = 0
          AND RDB$RELATION_NAME NOT LIKE 'RDB$%'
        ORDER BY 1
    """, {})
    fb.disconnect()

    if not success:
        return JSONResponse({"error": fb_err}, status_code=400)

    return JSONResponse({
        "tablas": [r["NOMBRE"].strip() for r in rows]
    })


@router.post("/test")
async def query_test(
    request: Request,
    query_id: int = Form(...),
    user: dict = Depends(get_current_user),
):
    from app.services.firebird_service import get_firebird_from_config

    conn = get_connection()
    q = conn.execute("SELECT * FROM queries WHERE id = ?", (query_id,)).fetchone()
    configs = {row["key"]: row["value"] for row in conn.execute("SELECT * FROM config").fetchall()}
    conn.close()

    if not q:
        return JSONResponse({"error": "Consulta no encontrada"}, status_code=404)

    fb, fb_success, fb_msg = get_firebird_from_config(configs)
    if not fb_success:
        return JSONResponse({"error": f"Error Firebird: {fb_msg}"}, status_code=400)

    success, fb_err, rows = fb.execute_query(q["query_text"], {})
    fb.disconnect()

    if not success:
        return JSONResponse({"error": fb_err}, status_code=400)

    limit = rows[:20] if rows else []
    return JSONResponse({
        "rows": limit,
        "total": len(rows),
        "columns": list(limit[0].keys()) if limit else []
    })