const axios = require('axios');

async function gemini({ message, instruction = '', sessionId = null }) {
    try {
        if (!message) throw new Error('Message is required.');
        
        let resumeArray = null;
        let cookie = null;
        let savedInstruction = instruction;
        
        if (sessionId) {
            try {
                const sessionData = JSON.parse(Buffer.from(sessionId, 'base64').toString());
                resumeArray = sessionData.resumeArray;
                cookie = sessionData.cookie;
                savedInstruction = instruction || sessionData.instruction || '';
            } catch (e) {
                console.error('Error parsing session:', e.message);
            }
        }
        
        if (!cookie) {
            const { headers } = await axios.post('https://gemini.google.com/_/BardChatUi/data/batchexecute?rpcids=maGuAc&source-path=%2F&bl=boq_assistant-bard-web-server_20250814.06_p1&f.sid=-7816331052118000090&hl=en-US&_reqid=173780&rt=c', 'f.req=%5B%5B%5B%22maGuAc%22%2C%22%5B0%5D%22%2Cnull%2C%22generic%22%5D%5D%5D&', {
                headers: {
                    'content-type': 'application/x-www-form-urlencoded;charset=UTF-8'
                }
            });
            
            cookie = headers['set-cookie']?.[0]?.split('; ')[0] || '';
        }
        
        const requestBody = [
            [message, 0, null, null, null, null, 0], ["en-US"], resumeArray || ["", "", "", null, null, null, null, null, null, ""],
            null, null, null, [1], 1, null, null, 1, 0, null, null, null, null, null, [[0]], 1, null, null, null, null, null,
            ["", "", savedInstruction, null, null, null, null, null, 0, null, 1, null, null, null, []],
            null, null, 1, null, null, null, null, null, null, null, 
            [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20], 1, null, null, null, null, [1]
        ];
        
        const payload = [null, JSON.stringify(requestBody)];
        
        const { data } = await axios.post('https://gemini.google.com/_/BardChatUi/data/assistant.lamda.BardFrontendService/StreamGenerate?bl=boq_assistant-bard-web-server_20250729.06_p0&f.sid=4206607810970164620&hl=en-US&_reqid=2813378&rt=c', new URLSearchParams({ 'f.req': JSON.stringify(payload) }).toString(), {
            headers: {
                'content-type': 'application/x-www-form-urlencoded;charset=UTF-8',
                'x-goog-ext-525001261-jspb': '[1,null,null,null,"9ec249fc9ad08861",null,null,null,[4]]',
                'cookie': cookie
            }
        });
        
        const match = Array.from(data.matchAll(/^\d+\n(.+?)\n/gm));
        const array = match.reverse();
        const selectedArray = array[3][1];
        const realArray = JSON.parse(selectedArray);
        const parse1 = JSON.parse(realArray[0][2]);
        
        const newResumeArray = [...parse1[1], parse1[4][0][0]];
        const text = parse1[4][0][1][0].replace(/\*\*(.+?)\*\*/g, '*$1*');
        
        const newSessionId = Buffer.from(JSON.stringify({
            resumeArray: newResumeArray,
            cookie: cookie,
            instruction: savedInstruction
        })).toString('base64');
        
        return {
            text: text,
            sessionId: newSessionId
        };
    } catch (error) {
        throw new Error(error.message);
    }
}

// Usage:
const inst = 'You are Shiroko, a friendly and cheerful AI assistant. Always answer in a friendly style.';
gemini({ message: 'Hello! My name is Rynn', instruction: inst })
    .then(r => {
        console.log('👤 User: Hello! My name is Rynn');
        console.log('🤖 Shiroko:', r.text);
        console.log()
        
        return gemini({ 
            message: 'What is my name?',
            sessionId: r.sessionId 
        });
    })
    .then(r => {
        console.log('👤 User: What is my name?');
        console.log('🤖 Shiroko:', r.text);
        console.log()
        
        return gemini({ 
            message: 'Who are you?',
            sessionId: r.sessionId 
        });
    })
    .then(r => {
        console.log('👤 User: Who are you?');
        console.log('🤖 Shiroko:', r.text);
        console.log()
    })
    .catch(err => console.error('❌ Error:', err.message));