const express = require('express');
const { chromium } = require('playwright');
const admin = require('firebase-admin');
const cors = require('cors');

// --- CONFIGURACIÓN FIREBASE ---
try {
    if (process.env.FIREBASE_PRIVATE_KEY) {
        if (!admin.apps.length) {
            admin.initializeApp({
                credential: admin.credential.cert({
                    projectId: process.env.FIREBASE_PROJECT_ID,
                    clientEmail: process.env.FIREBASE_CLIENT_EMAIL,
                    privateKey: process.env.FIREBASE_PRIVATE_KEY.replace(/\\n/g, '\n'),
                }),
            });
        }
        console.log("✅ Firebase inicializado.");
    }
} catch (e) { console.error("❌ Error Firebase:", e.message); }

const db = admin.apps.length ? admin.firestore() : null;
const APPOINTMENTS_COL = "appointments"; 

const URLS = {
    MULTI_LOGIN: "https://web.multiasistencia.com/w3multi/acceso.php",
    MULTI_LIST: "https://web.multiasistencia.com/w3multi/cerrados.php",
    HS_LOGIN: "https://www.clientes.homeserve.es/cgi-bin/fccgi.exe?w3exec=PROF_PASS",
    HS_LIQ: "https://www.clientes.homeserve.es/cgi-bin/fccgi.exe?w3exec=CONSULTALIQ_WEB"
};

const app = express();
app.use(cors({ origin: '*' }));
app.use(express.json({ limit: '50mb' }));

// --- ENDPOINT DE STREAMING ---
app.post('/api/robot-cobros', async (req, res) => {
    // Configurar cabeceras para STREAMING (Evita timeout 504)
    res.setHeader('Content-Type', 'text/plain; charset=utf-8');
    res.setHeader('Transfer-Encoding', 'chunked');
    
    // Función auxiliar para enviar mensajes al navegador
    const send = (type, payload) => {
        res.write(JSON.stringify({ type, payload }) + "\n");
    };

    const { action, url, provider, dataToSave, month, year } = req.body; 
    console.log(`🔔 Orden: ${action} [${provider}]`);

    try {
        if (provider === 'MULTI') {
            if (action === 'scan') {
                await runMultiStream(month, year, send);
            } 
            else if (action === 'save_data') {
                const count = await runSaver(dataToSave, 'MULTI');
                send('DONE', { count });
            }
        } 
        else {
            // HOMESERVE (Mantenemos modo normal por ahora, adaptado a respuesta simple)
            if (action === 'scan') {
                const lista = await runHSScanner();
                send('HS_DATES', lista);
            } 
            else if (action === 'analyze') {
                const analisis = await runHSAnalyzer(url);
                send('HS_DATA', analisis);
            }
            else if (action === 'save_data') {
                const count = await runSaver(dataToSave, 'HS');
                send('DONE', { count });
            }
        }
    } catch (err) {
        console.error("❌ CRASH:", err.message);
        send('ERROR', err.message);
    } finally {
        res.end(); // Cerrar conexión al terminar
    }
});

// ==================================================================
// 🤖 LÓGICA MULTIASISTENCIA (STREAMING)
// ==================================================================

async function runMultiStream(mes, anio, send) {
    let browser = null;
    const valD1 = `${mes}_${anio}`;
    const valD2 = `${anio}`;

    try {
        send('LOG', `🚀 Iniciando robot para: ${valD1}`);
        const { browser: b, page } = await loginMulti();
        browser = b;

        send('LOG', "📂 Accediendo a Servicios Cerrados...");
        await page.goto(URLS.MULTI_LIST, { waitUntil: 'domcontentloaded' });
        
        // Filtro Fecha
        if (await page.isVisible('select[name="D1"]')) {
            send('LOG', `📅 Aplicando filtro de fecha...`);
            await page.selectOption('select[name="D1"]', valD1);
            await page.selectOption('select[name="D2"]', valD2);
            await page.click('input[name="continuar"]'); 
            await page.waitForTimeout(2000);
        }

        // Recolectar IDs
        let idsServicios = [];
        let tieneSiguiente = true;
        let pagActual = 1;

        while (tieneSiguiente && pagActual <= 10) { 
            send('LOG', `📄 Leyendo página ${pagActual}...`);
            const nuevosIds = await page.evaluate(() => {
                const celdas = Array.from(document.querySelectorAll('td.tdet'));
                const lista = [];
                celdas.forEach(td => {
                    if (/^\d{8}$/.test(td.innerText.trim())) lista.push(td.innerText.trim());
                });
                return lista;
            });
            
            const unicosPag = [...new Set(nuevosIds)];
            idsServicios.push(...unicosPag);
            send('LOG', `   -> Encontrados ${unicosPag.length} servicios.`);

            const haySiguiente = await page.$('a:has-text("Página siguiente")');
            if (haySiguiente) {
                await haySiguiente.click();
                await page.waitForTimeout(3000);
                pagActual++;
            } else { tieneSiguiente = false; }
        }

        idsServicios = [...new Set(idsServicios)];
        send('LOG', `🔍 Total servicios a revisar: ${idsServicios.length}`);

        // PROCESAR UNO A UNO Y ENVIAR AL MOMENTO
        for (const [index, idServicio] of idsServicios.entries()) {
            send('PROGRESS', { current: index + 1, total: idsServicios.length });
            
            const urlPresupuesto = `https://web.multiasistencia.com/w3multi/valprincipal.php?reparacion=${idServicio}&modo=0`;
            
            try {
                await page.goto(urlPresupuesto, { timeout: 45000, waitUntil: 'domcontentloaded' });
                
                // ESPERA EXPLÍCITA DEL PRECIO (Clave para que no salga 0€)
                try {
                    // Esperamos hasta 5 segundos a que aparezca el elemento del precio
                    await page.waitForSelector('td[data-label*="TOTAL REPARACION"]', { timeout: 5000 });
                } catch(e) { /* Si falla, seguimos, quizás no hay precio */ }

                const info = await page.evaluate(() => {
                    const clean = (t) => t ? t.trim() : "Desconocido";
                    let cliente = "", direccion = "", totalStr = "0";

                    // Datos Cliente
                    const bloques = Array.from(document.querySelectorAll('.policy-info-block'));
                    bloques.forEach(b => {
                        const t = b.querySelector('.policy-info-title')?.innerText.toUpperCase() || "";
                        const v = b.querySelector('.policy-info-value')?.innerText || "";
                        if (t.includes("CLIENTE")) cliente = v;
                        if (t.includes("DIRECCIÓN")) direccion = v;
                    });

                    // Fallback antiguo
                    if (!cliente) {
                        const tds = Array.from(document.querySelectorAll('td'));
                        for(let i=0; i<tds.length; i++) {
                            const txt = tds[i].innerText.toUpperCase();
                            if(txt.includes("NOMBRE CLIENTE") && tds[i+1]) cliente = tds[i+1].innerText;
                            if(txt.includes("DIRECCIÓN") && tds[i+1]) direccion = tds[i+1].innerText;
                        }
                    }

                    // PRECIO (Selector exacto que me diste)
                    const celdaTotal = document.querySelector('td[data-label*="TOTAL REPARACION"]');
                    if (celdaTotal) totalStr = celdaTotal.innerText;
                    
                    return { cliente: clean(cliente), direccion: clean(direccion), totalStr };
                });

                let importe = 0;
                if (info.totalStr) {
                    let cleanMoney = info.totalStr.replace(/[^\d.,-]/g, '').replace(',', '.');
                    importe = Math.abs(parseFloat(cleanMoney) || 0);
                }

                // Cruzar con BD
                let docId = null;
                let enBD = false;
                if (db) {
                    const q = await db.collection(APPOINTMENTS_COL).where("serviceNumber", "==", idServicio).get();
                    if (!q.empty) {
                        docId = q.docs[0].id;
                        enBD = true;
                    }
                }

                // ENVIAR DATO EN TIEMPO REAL AL NAVEGADOR
                send('ITEM_FOUND', {
                    servicio: idServicio,
                    direccion: info.direccion,
                    importe: importe,
                    enBD: enBD,
                    docId: docId
                });

            } catch (errDetail) { 
                console.error(errDetail);
                send('LOG', `⚠️ Error en ${idServicio}: ${errDetail.message}`);
            }
        }

        send('FINISH_SCAN', {});

    } catch (e) { throw e; } finally { if(browser) await browser.close(); }
}

async function loginMulti() {
    let user = "", pass = "";
    if (db) {
        const doc = await db.collection("providerCredentials").doc("multiasistencia").get();
        if (doc.exists) { user = doc.data().user; pass = doc.data().pass; }
    }
    if(!user) throw new Error("Faltan credenciales");

    const browser = await chromium.launch({ headless: true, args: ['--no-sandbox'] });
    const page = await browser.newPage();

    await page.goto(URLS.MULTI_LOGIN, { timeout: 60000 });

    const userFilled = await page.evaluate((u) => {
      const el = document.querySelector('input[name="usuario"]') || document.querySelector('input[type="text"]');
      if (el) { el.value = u; el.dispatchEvent(new Event('input', { bubbles: true })); return true; }
      return false;
    }, user);

    if (!userFilled) await page.fill('input[name="usuario"]', user);
    await page.fill('input[type="password"]', pass);
    await page.click('input[type="submit"]'); 
    await page.waitForTimeout(4000);

    return { browser, page };
}

// ... (Resto de funciones HS y Saver se mantienen igual que la V10) ...
// IMPORTANTE: Asegúrate de incluirlas aquí abajo (runHSScanner, runHSAnalyzer, runSaver, loginHS)
// Copialas del mensaje anterior si hace falta, no cambian.

async function runHSScanner() {
    let browser = null;
    try {
        const { browser: b, page } = await loginHS();
        browser = b;
        const liquidaciones = await page.evaluate(() => {
            const links = Array.from(document.querySelectorAll('a'));
            const results = [];
            links.forEach(l => {
                const txt = l.innerText.trim();
                if (/\d{2}\/\d{2}\/\d{4}/.test(txt)) results.push({ fecha: txt, url: l.href });
            });
            return results;
        });
        return liquidaciones;
    } catch (e) { throw e; } finally { if(browser) await browser.close(); }
}

async function runHSAnalyzer(targetUrl) {
    let browser = null;
    try {
        const { browser: b, page } = await loginHS(); 
        browser = b;
        await page.goto(targetUrl, { timeout: 60000 });
        await page.waitForTimeout(1500);
        const botonPulsado = await page.evaluate(() => {
            const el = Array.from(document.querySelectorAll('input, button, a')).find(el => (el.value || el.innerText || "").toUpperCase().includes("DESGLOSE"));
            if (el) { el.click(); return true; } return false;
        });
        if (botonPulsado) await page.waitForTimeout(3000);
        const datos = await page.evaluate(() => {
            const rows = Array.from(document.querySelectorAll('tr'));
            return rows.map(tr => {
                const tds = tr.querySelectorAll('td');
                if (tds.length >= 6 && /^\d{5,}$/.test(tds[0].innerText.trim())) {
                    return { servicio: tds[0].innerText.trim(), direccion: tds[1].innerText.trim(), saldoRaw: tds[5].innerText.trim() };
                }
            }).filter(Boolean);
        });
        const encontrados = [], noEncontrados = [];
        if (db) {
            for (const item of datos) {
                let importe = Math.abs(parseFloat(item.saldoRaw.replace(/[^\d.-]/g, '')) || 0);
                const q = await db.collection(APPOINTMENTS_COL).where("serviceNumber", "==", item.servicio).get();
                if (!q.empty) q.forEach(doc => encontrados.push({ servicio: item.servicio, direccion: item.direccion, importe, docId: doc.id }));
                else noEncontrados.push({ servicio: item.servicio, direccion: item.direccion, importe });
            }
        }
        return { encontrados, noEncontrados };
    } catch (e) { throw e; } finally { if(browser) await browser.close(); }
}

async function loginHS() {
    let user = "", pass = "";
    if (db) {
        const doc = await db.collection("providerCredentials").doc("homeserve").get();
        if (doc.exists) { user = doc.data().user; pass = doc.data().pass; }
    }
    const browser = await chromium.launch({ headless: true, args: ['--no-sandbox'] });
    const page = await browser.newPage();
    await page.goto(URLS.HS_LOGIN, { timeout: 60000 });
    if (await page.isVisible('input[name="CODIGO"]')) {
        await page.fill('input[name="CODIGO"]', user);
        await page.fill('input[type="password"]', pass);
        await page.keyboard.press('Enter');
        await page.waitForTimeout(4000);
    }
    await page.goto(URLS.HS_LIQ);
    await page.waitForTimeout(2000);
    return { browser, page };
}

async function runSaver(items, providerType) {
    if (!db) return 0;
    const batch = db.batch();
    const nowISO = new Date().toISOString();
    let count = 0;
    items.forEach(item => {
        const ref = db.collection(APPOINTMENTS_COL).doc(item.docId);
        const updateData = { paidAmount: item.importe, paymentState: "Pagado", status: 'completed', paymentDate: nowISO, lastUpdatedByRobot: nowISO };
        if (providerType === 'MULTI') updateData.multiasistenciaPaymentStatus = 'paid_verified';
        else updateData.homeservePaymentStatus = 'paid_saldo';
        batch.update(ref, updateData);
        count++;
    });
    await batch.commit();
    return count;
}

const PORT = process.env.PORT || 3000;
app.listen(PORT, () => console.log(`🚀 Server on port ${PORT}`));