index.js

const fs = require('fs');
const { execSync, spawn } = require('child_process');
const { Client, LocalAuth } = require('whatsapp-web.js');
const qrcode = require('qrcode-terminal');
require('dotenv').config();

const pyenv = process.env.VIRTUAL_ENV;
// ── Persistent Python/Whisper process ────────────────────────────────────────
let whisperReady = false;
const whisperProc = spawn(`${pyenv}/bin/python`, ['transcribe.py'], {
    env:{
        ...process.env,
    }
});

whisperProc.stdout.once('data', (data) => {
    if (data.toString().trim() === 'READY') {
        whisperReady = true;
        console.log('🧠 Whisper caricato in memoria e pronto!');
    }
});

whisperProc.stderr.on('data', (d) => console.error('🐍 Python:', d.toString()));
whisperProc.on('close', (code) => console.error(`🐍 Python terminato con codice ${code}`));

function transcribeAudio(mp3File) {
    return new Promise((resolve, reject) => {
        const onData = (data) => {
            const line = data.toString().trim();
            if (line.startsWith('OK ')) {
                whisperProc.stdout.off('data', onData);
                resolve(line.slice(3));
            } else if (line.startsWith('ERR ')) {
                whisperProc.stdout.off('data', onData);
                reject(new Error(line.slice(4)));
            }
        };
        whisperProc.stdout.on('data', onData);
        whisperProc.stdin.write(mp3File + '\n');
    });
}

// ── WhatsApp client ───────────────────────────────────────────────────────────
const client = new Client({
    authStrategy: new LocalAuth(),
    puppeteer: {
        headless: true,
        executablePath: '/usr/bin/chromium',
        args: ['--no-sandbox', '--disable-setuid-sandbox', '--disable-dev-shm-usage', '--disable-gpu']
    }   
});

client.on('qr', qr => {
    console.log('--- NUOVO QR CODE ---');
    qrcode.generate(qr, { small: true });
});

client.on('ready', () => {
    console.log('✅ BOT PRONTO');
    console.log('   - Vocale in arrivo → trascrizione automatica');
    console.log('   - Rispondi a un vocale con "!t" → trascrizione on-demand');
});

client.on('message_create', async (msg) => {

    // ON-DEMAND: rispondi a un vocale con "!t"
    if (msg.body === '!t' && msg.hasQuotedMsg) {
        const quoted = await msg.getQuotedMessage();
        if (quoted.hasMedia && (quoted.type === 'audio' || quoted.type === 'ptt')) {
            await transcribe(quoted);
        } else {
            console.log('❌ Il messaggio citato non è un audio.');
        }
        return;
    }

});

async function getSenderName(msg) {
    try {
        // in gruppi prende il nome del contatto
        const contact = await msg.getContact();
        return contact.pushname || contact.name || contact.number || msg.from;
    } catch {
        return msg.from;
    }
}

async function transcribe(audioMsg) {
    let rawFile = null;
    let mp3File = null;

    try {
        if (!whisperReady) {
            console.log('⏳ Whisper ancora in caricamento, riprova tra qualche secondo...');
            return;
        }

        const media = await audioMsg.downloadMedia();

        if (!media || !media.data) {
            console.error('❌ Impossibile scaricare l\'audio.');
            return;
        }

        const mimeToExt = {
            'audio/ogg': 'ogg',
            'audio/mpeg': 'mp3',
            'audio/mp4': 'mp4',
            'audio/webm': 'webm',
            'audio/wav': 'wav'
        };
        const baseMime = media.mimetype.split(';')[0].trim();
        const ext = mimeToExt[baseMime] || 'ogg';
        const timestamp = Date.now();

        rawFile = `./audio_${timestamp}.${ext}`;
        mp3File = `./audio_${timestamp}.mp3`;

        fs.writeFileSync(rawFile, media.data, 'base64');
        execSync(`ffmpeg -y -i "${rawFile}" -ar 16000 -ac 1 -c:a libmp3lame "${mp3File}"`, { stdio: 'pipe' });

        const [text, senderName] = await Promise.all([
            transcribeAudio(mp3File),
            getSenderName(audioMsg)
        ]);

        const time = new Date().toLocaleTimeString('it-IT');

        // ✅ Output pulito, solo quando trascrivi
        console.log(`\n┌─────────────────────────────────`);
        console.log(`│ 👤 ${senderName}`);
        console.log(`│ 🕐 ${time}`);
        console.log(`│`);
        console.log(`│ 📝 ${text}`);
        console.log(`└─────────────────────────────────\n`);

    } catch (err) {
        console.error('❌ Errore:', err.message);
    } finally {
        for (const f of [rawFile, mp3File]) {
            if (f && fs.existsSync(f)) {
                try { fs.unlinkSync(f); } catch (_) {}
            }
        }
    }
}

client.initialize();