/**
 * Target: WhatsApp Web Dashboard & Auto-Invite Bot (Express + Baileys)
 * Runtime: Node.js
 */

const { default: makeWASocket, useMultiFileAuthState, DisconnectReason } = require('@whiskeysockets/baileys');
const pino = require('pino');
const express = require('express');
const fs = require('fs');

const app = express();
const PORT = process.env.PORT || 3000;

app.use(express.urlencoded({ extended: true }));
app.use(express.json());

// File penyimpanan data lokal
const DATA_FILE = './data.json';

// Inisialisasi data.json jika belum ada
if (!fs.existsSync(DATA_FILE)) {
    fs.writeFileSync(DATA_FILE, JSON.stringify({
        targetNumber: "6281234567890",
        groupIds: "120363011111111111@g.us\n120363022222222222@g.us"
    }, null, 2));
}

const sleep = (ms) => new Promise(resolve => setTimeout(resolve, ms));

let sock;
let connectionStatus = 'DISCONNECTED';
let botLogs = [];

function addLog(text) {
    const timestamp = new Date().toLocaleTimeString();
    const logMessage = `[${timestamp}] ${text}`;
    console.log(logMessage);
    botLogs.unshift(logMessage);
    if (botLogs.length > 50) botLogs.pop(); // Batasi 50 log terakhir
}

// Inisialisasi Baileys WhatsApp Bot
async function startBot() {
    const { state, saveCreds } = await useMultiFileAuthState('auth_info');
    
    sock = makeWASocket({
        auth: state,
        logger: pino({ level: 'silent' }),
        printQRInTerminal: true
    });

    sock.ev.on('connection.update', (update) => {
        const { connection, lastDisconnect, qr } = update;
        
        if (qr) {
            connectionStatus = 'WAITING_QR (Cek Terminal untuk Scan)';
        }

        if (connection === 'close') {
            connectionStatus = 'DISCONNECTED';
            const shouldReconnect = lastDisconnect?.error?.output?.statusCode !== DisconnectReason.loggedOut;
            addLog('Koneksi terputus. Mencoba menyambungkan kembali...');
            if (shouldReconnect) {
                startBot();
            }
        } else if (connection === 'open') {
            connectionStatus = 'CONNECTED (Terhubung)';
            addLog('WhatsApp berhasil terhubung!');
        }
    });

    sock.ev.on('creds.update', saveCreds);

    // Fitur Cek ID Grup via pesan "id"
    sock.ev.on('messages.upsert', async ({ messages, type }) => {
        if (type !== 'notify') return;
        
        const msg = messages[0];
        if (!msg.message || msg.key.fromMe) return;

        const remoteJid = msg.key.remoteJid;
        const messageText = msg.message.conversation || msg.message.extendedTextMessage?.text || '';

        if (remoteJid.endsWith('@g.us') && messageText.trim().toLowerCase() === 'id') {
            addLog(`Perintah 'id' diterima di grup: ${remoteJid}`);
            try {
                await sock.sendMessage(remoteJid, { 
                    text: `ID Grup ini adalah:\n\`\`\`${remoteJid}\`\`\`` 
                }, { quoted: msg });
            } catch (error) {
                addLog(`Gagal kirim ID grup: ${error.message}`);
            }
        }
    });
}

// Fungsi Eksekusi Auto-Invite
async function runAutoInvite() {
    if (!sock || connectionStatus !== 'CONNECTED (Terhubung)') {
        addLog('Gagal invite: Bot WhatsApp belum terhubung/aktif.');
        return { success: false, message: 'Bot belum terhubung!' };
    }

    try {
        const rawData = fs.readFileSync(DATA_FILE, 'utf8');
        const data = JSON.parse(rawData);
        
        const targetNumber = data.targetNumber;
        const groupIds = data.groupIds.split('\n').map(g => g.trim()).filter(g => g.length > 0);

        if (!targetNumber || groupIds.length === 0) {
            addLog('Data target atau grup kosong.');
            return { success: false, message: 'Target atau grup kosong!' };
        }

        const participantJid = `${targetNumber}@s.whatsapp.net`;
        addLog(`Memulai proses invite nomor ${targetNumber} ke ${groupIds.length} grup...`);

        // Jalankan proses invite secara asynchronous di background
        (async () => {
            for (let i = 0; i < groupIds.length; i++) {
                const groupId = groupIds[i];
                try {
                    addLog(`[${i + 1}/${groupIds.length}] Mengundang ke: ${groupId}`);
                    const resAdd = await sock.groupParticipantsUpdate(groupId, [participantJid], 'add');
                    addLog(`Sukses invite ke ${groupId}: ${JSON.stringify(resAdd)}`);

                    const delayTime = Math.floor(Math.random() * 4000) + 6000;
                    await sleep(delayTime);
                } catch (err) {
                    addLog(`Gagal invite ke ${groupId}: ${err?.message || err}`);
                }
            }
            addLog('Semua proses invite massal selesai.');
        })();

        return { success: true, message: 'Proses invite sedang berjalan di background!' };
    } catch (err) {
        return { success: false, message: err.message };
    }
}

// --- ROUTE WEB DASHBOARD ---

app.get('/', (req, res) => {
    const rawData = fs.readFileSync(DATA_FILE, 'utf8');
    const data = JSON.parse(rawData);

    const html = `
    <!DOCTYPE html>
    <html lang="id">
    <head>
        <meta charset="UTF-8">
        <title>Panel Auto-Invite WhatsApp</title>
        <style>
            body { font-family: sans-serif; background: #0f172a; color: #f8fafc; padding: 20px; }
            .container { max-width: 700px; margin: 0 auto; background: #1e293b; padding: 20px; border-radius: 8px; box-shadow: 0 4px 6px rgba(0,0,0,0.3); }
            h2 { color: #38bdf8; margin-top: 0; }
            label { display: block; margin-top: 15px; font-weight: bold; }
            input[type="text"], textarea { width: 100%; padding: 10px; margin-top: 5px; background: #0f172a; border: 1px solid #334155; color: #fff; border-radius: 4px; box-sizing: border-box; }
            textarea { height: 120px; resize: vertical; }
            .btn-group { margin-top: 20px; display: flex; gap: 10px; }
            button { background: #0284c7; color: white; border: none; padding: 10px 20px; font-size: 16px; border-radius: 4px; cursor: pointer; flex: 1; }
            button:hover { background: #0369a1; }
            button.btn-invite { background: #16a34a; }
            button.btn-invite:hover { background: #15803d; }
            .status-box { margin-top: 20px; padding: 12px; background: #0f172a; border-radius: 4px; border: 1px solid #334155; }
            .logs { background: #090d16; padding: 10px; height: 180px; overflow-y: auto; font-family: monospace; font-size: 12px; color: #38bdf8; border-radius: 4px; margin-top: 10px; }
        </style>
    </head>
    <body>
        <div class="container">
            <h2>Panel Kontrol Auto-Invite WhatsApp</h2>
            <div class="status-box">
                <strong>Status Bot:</strong> <span style="color: ${connectionStatus.includes('CONNECTED') ? '#4ade80' : '#f87171'}">${connectionStatus}</span>
            </div>
            
            <form method="POST" action="/save">
                <label>Nomor Target (Tanpa +, Cth: 628123456789):</label>
                <input type="text" name="target_number" value="${data.targetNumber}" required>
                
                <label>Daftar ID Grup (Satu ID per baris, berakhiran @g.us):</label>
                <textarea name="group_ids" required>${data.groupIds}</textarea>
                
                <div class="btn-group">
                    <button type="submit">Simpan Data</button>
                    <button type="button" class="btn-invite" onclick="triggerInvite()">🚀 Mulai Invite Sekarang</button>
                </div>
            </form>

            <label style="margin-top: 25px;">Live Logs Aktivitas:</label>
            <div class="logs" id="logContainer">${botLogs.join('<br>')}</div>
        </div>

        <script>
            function triggerInvite() {
                if(!confirm('Yakin ingin menjalankan proses invite sekarang?')) return;
                fetch('/start-invite', { method: 'POST' })
                    .then(res => res.json())
                    .then(data => {
                        alert(data.message);
                        location.reload();
                    })
                    .catch(err => alert('Gagal memicu invite: ' + err));
            }

            setInterval(() => {
                fetch('/logs').then(res => res.json()).then(data => {
                    document.getElementById('logContainer').innerHTML = data.logs.join('<br>');
                });
            }, 3000);
        </script>
    </body>
    </html>
    `;
    res.send(html);
});

// Simpan Data Form
app.post('/save', (req, res) => {
    const { target_number, group_ids } = req.body;
    fs.writeFileSync(DATA_FILE, JSON.stringify({
        targetNumber: target_number.trim(),
        groupIds: group_ids.trim()
    }, null, 2));
    res.redirect('/');
});

// Endpoint Tombol Web untuk Eksekusi Invite
app.post('/start-invite', async (req, res) => {
    const result = await runAutoInvite();
    res.json(result);
});

// Endpoint untuk Refresh Log di Frontend
app.get('/logs', (req, res) => {
    res.json({ logs: botLogs });
});

app.listen(PORT, () => {
    console.log(`Server web & bot berjalan di http://localhost:${PORT}`);
    startBot();
});