const { WebcastPushConnection } = require('tiktok-live-connector');
const mysql = require('mysql2/promise');

const [,, username, dbHost, dbUser, dbPass, dbName] = process.argv;

if (!username) {
    console.error('Usage: node tiktok_bridge.js <username> <dbHost> <dbUser> <dbPass> <dbName>');
    process.exit(1);
}

async function start() {
    console.log(`TikTok Bridge starting for @${username}`);
    
    let connection;
    try {
        connection = await mysql.createConnection({
            host: dbHost || '127.0.0.1',
            user: dbUser || 'root',
            password: dbPass || '',
            database: dbName || 'app_db'
        });
        console.log('Connected to database');
    } catch (err) {
        console.error('DB Connection failed:', err);
        process.exit(1);
    }

    const tiktokConnection = new WebcastPushConnection(username);

    tiktokConnection.connect().then(state => {
        console.log(`Connected to TikTok Live @${username} (Room ID: ${state.roomId})`);
    }).catch(err => {
        console.error('TikTok Connection failed:', err);
        process.exit(1);
    });

    tiktokConnection.on('chat', async (data) => {
        console.log(`[Chat] ${data.uniqueId}: ${data.comment}`);
        try {
            await connection.execute(
                'INSERT INTO tiktok_history (username, comment_author, comment_text, created_at) VALUES (?, ?, ?, NOW())',
                [username, data.uniqueId, data.comment]
            );
        } catch (err) {
            console.error('Error inserting chat:', err);
        }
    });

    tiktokConnection.on('gift', async (data) => {
        const giftMsg = `sent ${data.repeatCount}x ${data.giftName}!`;
        console.log(`[Gift] ${data.uniqueId}: ${giftMsg}`);
        try {
            await connection.execute(
                'INSERT INTO tiktok_history (username, comment_author, comment_text, created_at) VALUES (?, ?, ?, NOW())',
                [username, data.uniqueId, giftMsg]
            );
        } catch (err) {
            console.error('Error inserting gift:', err);
        }
    });

    tiktokConnection.on('disconnected', () => {
        console.log('TikTok connection disconnected');
        process.exit(0);
    });

    tiktokConnection.on('error', (err) => {
        console.error('TikTok error:', err);
    });

    // Keep alive
    setInterval(async () => {
        try {
            await connection.query('SELECT 1');
        } catch (err) {
            console.error('DB connection lost, reconnecting...');
            connection = await mysql.createConnection({
                host: dbHost, user: dbUser, password: dbPass, database: dbName
            });
        }
    }, 30000);
}

start();