content.js

// Listen for messages from the background script
browser.runtime.onMessage.addListener((message, sender, sendResponse) => {
    if (message.action === "getSelectedMessages") {
        try {
            // Get the message list container
            const messageList = document.querySelector('#threadTree');
            if (!messageList) {
                console.error("Could not find message list");
                sendResponse([]);
                return true;
            }

            // Get selected rows
            const selectedRows = messageList.querySelectorAll('tr.selected');
            if (!selectedRows || selectedRows.length === 0) {
                console.log("No messages selected");
                sendResponse([]);
                return true;
            }

            // Extract message IDs
            const selectedMessages = Array.from(selectedRows).map(row => {
                // Try different possible ID attributes
                const messageId = row.getAttribute('data-message-id') || 
                                row.getAttribute('data-id') || 
                                row.getAttribute('id');
                
                if (!messageId) {
                    console.warn("Row missing message ID:", row);
                    return null;
                }
                
                // Clean up the ID if needed
                const cleanId = messageId.replace(/^msg-/i, '');
                return { id: cleanId };
            }).filter(msg => msg !== null);

            console.log("Found selected messages:", selectedMessages);
            sendResponse(selectedMessages);
        } catch (error) {
            console.error("Error getting selected messages:", error);
            sendResponse([]);
        }
    } else if (message.action === 'ollamaFetch') {
        // Runs inside a tab at http://localhost:11434 to avoid CORS
        (async () => {
            try {
                const { fetchAction, model, prompt, headers, correlationId } = message;
                const base = window.location.origin;

                if (fetchAction === 'pull') {
                    const res = await fetch(`${base}/api/pull`, {
                        method: 'POST',
                        headers: Object.assign({ 'Content-Type': 'application/json' }, headers || {}),
                        body: JSON.stringify({ name: model, stream: true })
                    });
                    if (!res.ok) {
                        const t = await res.text();
                        try {
                            const j = JSON.parse(t);
                            throw new Error(j.error || t);
                        } catch (e) {
                            throw new Error(t || `HTTP ${res.status}`);
                        }
                    }
                    const reader = res.body.getReader();
                    const decoder = new TextDecoder();
                    let buffer = '';
                    while (true) {
                        const { done, value } = await reader.read();
                        if (done) break;
                        buffer += decoder.decode(value, { stream: true });
                        const lines = buffer.split('\n');
                        buffer = lines.pop();
                        for (const line of lines) {
                            if (!line.trim()) continue;
                            try {
                                const data = JSON.parse(line);
                                const payload = { action: 'ollamaPullProgress', correlationId, status: data.status || '' };
                                if (data.completed && data.total) {
                                    payload.percent = Math.round((data.completed / data.total) * 100);
                                }
                                browser.runtime.sendMessage(payload).catch(() => {});
                            } catch (e) {
                                // ignore parse errors for partial lines
                            }
                        }
                    }
                    browser.runtime.sendMessage({ action: 'ollamaPullComplete', correlationId, ok: true }).catch(() => {});
                    sendResponse({ ok: true });
                } else if (fetchAction === 'chat') {
                    const res = await fetch(`${base}/api/chat`, {
                        method: 'POST',
                        headers: Object.assign({ 'Content-Type': 'application/json' }, headers || {}),
                        body: JSON.stringify({ model, messages: [{ role: 'user', content: prompt }], stream: false })
                    });
                    const data = await res.json();
                    sendResponse({ ok: true, data });
                } else {
                    sendResponse({ ok: false, error: 'unknown fetchAction' });
                }
            } catch (err) {
                sendResponse({ ok: false, error: err.message || String(err) });
            }
        })();
        return true;
    }
    return true;
});