#!/usr/bin/env python3
"""
Script complet: PDF → OCR → Oracle (cu PACK_CONTAFIN)
Usage: python process_and_save.py <path_to_pdf> [--save]
"""
import sys
sys.path.insert(0, "/workspace/roa2web")

import asyncio
import oracledb
from datetime import datetime
from decimal import Decimal
from pathlib import Path

ORACLE_CONFIG = {
    "user": "MARIUSM_AUTO",
    "password": "ROMFASTSOFT",
    "dsn": "10.0.20.121:1521/ROA"
}

CUI_TO_CONT = {
    "11201891": "6022",  # MOL
    "1590082": "6022",   # OMV Petrom
    "14991381": "6022",  # MOL Romania
    "10562600": "6021",  # Dedeman
}

def get_cont_cheltuiala(cui: str) -> str:
    cui_clean = cui.upper().replace("RO", "").strip()
    return CUI_TO_CONT.get(cui_clean, "6028")

async def process_pdf(pdf_path: Path):
    from backend.modules.data_entry.services.ocr_service import ocr_service
    
    mime_type = "application/pdf" if pdf_path.suffix.lower() == ".pdf" else "image/jpeg"
    
    print(f"\n[OCR] Processing: {pdf_path.name}")
    print("-" * 50)
    
    success, message, result = await ocr_service.process_image(pdf_path, mime_type)
    
    if not success:
        print(f"ERROR: {message}")
        return None
    
    print(f"Partner: {result.partner_name}")
    print(f"CUI: {result.cui}")
    print(f"Data: {result.receipt_date}")
    print(f"Numar: {result.receipt_number}")
    print(f"Total: {result.amount}")
    print(f"TVA: {result.tva_total}")
    print(f"Confidence: {result.overall_confidence:.0%}")
    
    return result

def save_to_oracle_with_pack(result, do_commit: bool = False):
    mode = "SAVE" if do_commit else "DRY RUN"
    print(f"\n[Oracle + PACK_CONTAFIN] {mode}")
    print("-" * 50)
    
    conn = oracledb.connect(**ORACLE_CONFIG)
    cursor = conn.cursor()
    
    try:
        an = result.receipt_date.year if result.receipt_date else datetime.now().year
        luna = result.receipt_date.month if result.receipt_date else datetime.now().month
        receipt_date = result.receipt_date or datetime.now().date()
        
        # Parametri
        id_util = 0  # ID utilizator implicit
        id_sucursala = 0
        
        # 1. INITIALIZEAZA
        print("[1] INITIALIZEAZA_SCRIERE_ACT_RUL...")
        cursor.callproc('PACK_CONTAFIN.INITIALIZEAZA_SCRIERE_ACT_RUL', [
            id_util,
            datetime.now(),
            an,
            luna,
            0,  # suprascriere_cod
            0,  # suprascriere_anluna
            0,  # scrie_sterge (0=scrie)
            id_sucursala
        ])
        print("   OK")
        
        # Obține COD din secvență sau calculează
        cursor.execute("SELECT NVL(MAX(COD), 0) + 1 FROM ACT WHERE AN = :an AND LUNA = :luna", an=an, luna=luna)
        cod = cursor.fetchone()[0]
        
        # Partner
        cui_clean = (result.cui or "").upper().replace("RO", "").strip()
        cursor.execute("SELECT ID_PART FROM NOM_PARTENERI WHERE COD_FISCAL = :cui OR COD_FISCAL = :cui2", 
                      cui=cui_clean, cui2="RO"+cui_clean)
        row = cursor.fetchone()
        id_part = row[0] if row else 0
        
        total = float(result.amount or 0)
        tva = float(result.tva_total or 0)
        fara_tva = total - tva
        nract = int(result.receipt_number) if result.receipt_number and result.receipt_number.isdigit() else 0
        cont = get_cont_cheltuiala(result.cui or "")
        expl = f"OCR: {result.partner_name or 'N/A'}"
        
        print(f"   COD: {cod}, Partner ID: {id_part}, Cont: {cont}")
        
        # 2. INSERT în ACT_TEMP
        print("[2] INSERT ACT_TEMP...")
        
        lines = [
            (cont, "401", fara_tva, expl, id_part, 0),  # cheltuială - partener pe credit
            ("401", "5311", total, f"Plata {expl}", 0, id_part),  # plată - partener pe debit
        ]
        if tva > 0:
            lines.insert(1, ("4426", "401", tva, f"TVA {expl}", id_part, 0))  # TVA - partener pe credit
        
        for scd, scc, suma, e, id_partc, id_partd in lines:
            cursor.execute("""
                INSERT INTO ACT_TEMP (
                    LUNA, AN, COD, DATAIREG, DATAACT, NRACT,
                    EXPLICATIA, SCD, SCC, SUMA, 
                    ID_PARTC, ID_PARTD, ID_UTIL, DATAORA
                ) VALUES (
                    :luna, :an, :cod, TRUNC(SYSDATE), :dataact, :nract,
                    :expl, :scd, :scc, :suma, 
                    :id_partc, :id_partd, :id_util, SYSDATE
                )
            """, luna=luna, an=an, cod=cod, dataact=receipt_date, nract=nract,
                expl=e, scd=scd, scc=scc, suma=suma, 
                id_partc=id_partc, id_partd=id_partd, id_util=id_util)
            print(f"   {scd} = {scc}: {suma:.2f}")
        
        # 3. FINALIZEAZA
        print("[3] FINALIZEAZA_SCRIERE_ACT_RUL...")
        mesaj = cursor.var(oracledb.STRING, 4000)
        cursor.callproc('PACK_CONTAFIN.FINALIZEAZA_SCRIERE_ACT_RUL', [
            id_util,
            cod,
            0,  # scrie_sterge
            0,  # modificare_nota
            0,  # scrie_cump_vanz
            mesaj
        ])
        result_msg = mesaj.getvalue()
        print(f"   Mesaj: {result_msg}")
        
        if do_commit:
            conn.commit()
            print(f"\n✅ SALVAT în Oracle (COD={cod})")
        else:
            conn.rollback()
            print(f"\n⚠️ DRY RUN - rollback (COD ar fi fost {cod})")
        
        return cod, result_msg
        
    except Exception as e:
        print(f"❌ Eroare: {e}")
        conn.rollback()
        raise
    finally:
        cursor.close()
        conn.close()

async def main():
    if len(sys.argv) < 2:
        print("Usage: python process_and_save.py <pdf_path> [--save]")
        return
    
    pdf_path = Path(sys.argv[1])
    if not pdf_path.exists():
        pdf_path = Path(f"/workspace/roa2web/tests/fixtures/ocr-samples/{sys.argv[1]}")
    
    if not pdf_path.exists():
        print(f"File not found: {sys.argv[1]}")
        return
    
    do_save = "--save" in sys.argv
    
    print("=" * 50)
    print("PDF -> OCR -> Oracle (PACK_CONTAFIN)")
    print("=" * 50)
    
    result = await process_pdf(pdf_path)
    if result:
        save_to_oracle_with_pack(result, do_commit=do_save)
    
    print("\n" + "=" * 50)

if __name__ == "__main__":
    asyncio.run(main())