rips_manager/app/routes/terceros.py

import json
import httpx
from datetime import datetime
from fastapi import APIRouter, Request, Form, Depends
from fastapi.responses import JSONResponse
from app.database import get_connection
from app.auth import get_current_user
from app.services.firebird_service import get_firebird_from_config
from app.services.json_generator import generar_tercero_api
from app.services.api_client import get_tns_token, TNS_BASE

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


def _load_configs() -> dict:
    conn = get_connection()
    configs = {row["key"]: row["value"] for row in conn.execute("SELECT * FROM config").fetchall()}
    conn.close()
    return configs


@router.get("")
async def terceros_page(request: Request, user: dict = Depends(get_current_user)):
    conn = get_connection()
    envios = conn.execute("""
        SELECT * FROM envios WHERE tipo = 'terceros'
        ORDER BY created_at DESC LIMIT 20
    """).fetchall()
    queries = conn.execute(
        "SELECT * FROM queries WHERE query_type = 'terceros' ORDER BY name"
    ).fetchall()
    configs = {row["key"]: row["value"] for row in conn.execute("SELECT * FROM config").fetchall()}
    conn.close()

    return request.app.state.templates.TemplateResponse("terceros.html", {
        "request": request, "user": user,
        "envios": envios, "queries": queries,
        "configs": configs,
        "firebird_host": configs.get("firebird_host", "localhost"),
        "firebird_port": configs.get("firebird_port", "3050"),
        "firebird_database": configs.get("firebird_database", ""),
    })


@router.post("/test-connection")
async def test_connection(
    request: Request,
    user: dict = Depends(get_current_user),
    host: str = Form(...),
    port: int = Form(...),
    database: str = Form(...),
    fb_user: str = Form(...),
    fb_password: str = Form(...),
):
    from app.services.firebird_service import FirebirdService
    fb = FirebirdService()
    success, msg = fb.connect(host, port, database, fb_user, fb_password)
    if success:
        fb.disconnect()
    return JSONResponse({"success": success, "message": msg})


@router.post("/preview")
async def preview_query(
    request: Request,
    user: dict = Depends(get_current_user),
    query_id: int = Form(...),
    doc_num: str = Form(""),
):
    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({"success": False, "message": "Consulta no encontrada"})

    fb, ok, msg = get_firebird_from_config(configs)
    if not ok:
        return JSONResponse({"success": False, "message": f"Error Firebird: {msg}"})

    if ":doc_num" in q["query_text"] and not doc_num:
        fb.disconnect()
        return JSONResponse({"success": False, "message": "Ingresa el número de documento para buscar el paciente"})

    params = {}
    if ":doc_num" in q["query_text"]:
        params["doc_num"] = doc_num
    if ":fecha_ini" in q["query_text"]:
        params["fecha_ini"] = "1900-01-01"
        params["fecha_fin"] = "2100-12-31"
    if ":factura" in q["query_text"]:
        params["factura"] = doc_num if doc_num else ""

    success, error, rows = fb.execute_query(q["query_text"], params if params else None)
    fb.disconnect()

    if not success:
        return JSONResponse({"success": False, "message": error})

    json_result = generar_tercero_api(rows[0]) if rows else None

    return JSONResponse({
        "success": True,
        "rows_count": len(rows),
        "columns": list(rows[0].keys()) if rows else [],
        "preview": rows[:5],
        "generated_json": json_result,
    })


@router.post("/send")
async def send_terceros(
    request: Request,
    user: dict = Depends(get_current_user),
    query_id: int = Form(...),
    doc_num: str = Form(""),
):
    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({"success": False, "message": "Consulta no encontrada"})

    fb, ok, msg = get_firebird_from_config(configs)
    if not ok:
        return JSONResponse({"success": False, "message": f"Error Firebird: {msg}"})

    if ":doc_num" in q["query_text"] and not doc_num:
        fb.disconnect()
        return JSONResponse({"success": False, "message": "Ingresa el número de documento"})

    params = {}
    if ":doc_num" in q["query_text"]:
        params["doc_num"] = doc_num

    success, error, rows = fb.execute_query(q["query_text"], params if params else None)
    fb.disconnect()

    if not success:
        return JSONResponse({"success": False, "message": error})
    if not rows:
        return JSONResponse({"success": False, "message": "No se encontraron datos"})

    tercero_json = generar_tercero_api(rows[0])

    token, token_err = await get_tns_token(
        configs.get("tns_empresa", ""),
        configs.get("tns_usuario", ""),
        configs.get("tns_password", ""),
    )
    if not token:
        return JSONResponse({"success": False, "message": f"Error login TNS: {token_err}"})

    headers = {"Content-Type": "application/json", "Authorization": f"Bearer {token}"}
    endpoint = f"{TNS_BASE}/v2/tablas/Tercero/Crear"

    resp_ok = False
    resp_text = ""
    resp_code = 0
    try:
        async with httpx.AsyncClient(timeout=int(configs.get("api_timeout", 30))) as client:
            resp = await client.post(endpoint, json=tercero_json, headers=headers)
        resp_ok = resp.is_success
        resp_text = resp.text[:2000]
        resp_code = resp.status_code
    except Exception as e:
        resp_text = str(e)

    conn = get_connection()
    conn.execute("""
        INSERT INTO envios (user_id, tipo, status, json_enviado, respuesta_api, created_at)
        VALUES (?, ?, ?, ?, ?, ?)
    """, (
        user["user_id"], "terceros",
        "success" if resp_ok else "error",
        json.dumps(tercero_json, indent=2, ensure_ascii=False),
        resp_text[:1000],
        datetime.now().isoformat(),
    ))
    conn.commit()
    conn.close()

    return JSONResponse({
        "success": resp_ok,
        "status_code": resp_code,
        "message": "Envío exitoso" if resp_ok else f"Error: {resp_text}",
        "cuv": resp_text[:200] if resp_ok else None,
    })