version set to 2.2.0_i143_v5

api_webchat/controller.js

@@ -25,8 +25,16 @@ const urlParams = new URLSearchParams(window.location.search);
 const llm = urlParams.get('llm');
 const call_id = urlParams.get('call_id');
 
-// console.log(">>>>>>>>>>> llm: " + llm);
-// console.log(">>>>>>>>>>> call_id: " + call_id);
+// Data received from the user
+let promptData = null;
+
+const messageInput = document.querySelector('message-input');
+const messagesArea = document.querySelector('messages-area');
+
+console.log(">>>>>>>>>> controller.js DOMContentLoaded");
+
+ console.log(">>>>>>>>>>> llm: " + llm);
+ console.log(">>>>>>>>>>> call_id: " + call_id);
 
 // The controller wires up all the components and workers together,
 // managing the dependencies. A kind of "DI" class.
@@ -34,30 +42,31 @@ let worker = null;
 
 switch (llm) {
     case "chatgpt_api":
-        browser.runtime.sendMessage({command: "openai_api_ready_" + call_id, window_id: (await browser.windows.getCurrent()).id});
+        console.log(">>>>>>>> [ThunderAI] chatgpt_api sending I'm ready message...");
+        // browser.runtime.sendMessage({command: "openai_api_ready_" + call_id, window_id: (await browser.windows.getCurrent()).id});
+        console.log(">>>>>> [ThunderAI] chatgpt_api I'm ready message sent.");
         worker = new Worker('model-worker-openai.js', { type: 'module' });
+        console.log(">>>>>> [ThunderAI] chatgpt_api worker initialized.");
         break;
     case "ollama_api":
-        browser.runtime.sendMessage({command: "ollama_api_ready_" + call_id, window_id: (await browser.windows.getCurrent()).id});
+        console.log(">>>>>>>> [ThunderAI] ollama_api sending I'm ready message...");
+        // browser.runtime.sendMessage({command: "ollama_api_ready_" + call_id, window_id: (await browser.windows.getCurrent()).id});
+        console.log(">>>>>> [ThunderAI] ollama_api I'm ready message sent.");
         worker = new Worker('model-worker-ollama.js', { type: 'module' });
+        console.log(">>>>>> [ThunderAI] ollama_api worker initialized.");
         break;
     case "openai_comp_api":
-        browser.runtime.sendMessage({command: "openai_comp_api_ready_" + call_id, window_id: (await browser.windows.getCurrent()).id});
+        // browser.runtime.sendMessage({command: "openai_comp_api_ready_" + call_id, window_id: (await browser.windows.getCurrent()).id});
         worker = new Worker('model-worker-openai_comp.js', { type: 'module' });
         break;
 }
 
-const messagesArea = document.querySelector('messages-area');
 messagesArea.init(worker);
 
 // Initialize the messageInput component and pass the worker to it
-const messageInput = document.querySelector('message-input');
 messageInput.init(worker);
 messageInput.setMessagesArea(messagesArea);
 
-// Data received from the user
-let promptData = null;
-
 // ============================== TESTING
 // let browser = {
 //     i18n: {
@@ -82,10 +91,14 @@ switch (llm) {
         messageInput.setModel(prefs_api.chatgpt_model);
         messagesArea.setLLMName("ChatGPT");
         worker.postMessage({ type: 'init', chatgpt_api_key: prefs_api.chatgpt_api_key, chatgpt_model: prefs_api.chatgpt_model});
+        console.log(">>>>>> [ThunderAI] chatgpt_api worker post init message.");
         messagesArea.appendUserMessage(browser.i18n.getMessage("chagpt_api_connecting") + " " +browser.i18n.getMessage("AndModel") + " \"" + prefs_api.chatgpt_model + "\"...", "info");
+        console.log(">>>>> [ThunderAI] messagesArea.appendUserMessage done.");
+        browser.runtime.sendMessage({command: "openai_api_ready_" + call_id, window_id: (await browser.windows.getCurrent()).id});
         break;
     case "ollama_api": {
-        let prefs_api = await browser.storage.sync.get({ollama_host: '', ollama_model: ''});
+        let prefs_api = {};
+        prefs_api = await browser.storage.sync.get({ollama_host: '', ollama_model: ''});
         let i18nStrings = {};
         i18nStrings["ollama_api_request_failed"] = browser.i18n.getMessage('ollama_api_request_failed');
         i18nStrings["error_connection_interrupted"] = browser.i18n.getMessage('error_connection_interrupted');
@@ -93,7 +106,14 @@ switch (llm) {
         messageInput.setModel(prefs_api.ollama_model);
         messagesArea.setLLMName("Ollama Local");
         worker.postMessage({ type: 'init', ollama_host: prefs_api.ollama_host, ollama_model: prefs_api.ollama_model, i18nStrings: i18nStrings});
+        console.log(">>>>>> [ThunderAI] Ollama init done.");
+        browser.runtime.sendMessage({command: "ollama_api_ready_" + call_id, window_id: (await browser.windows.getCurrent()).id});
+        try{
             messagesArea.appendUserMessage(browser.i18n.getMessage("ollama_api_connecting") + " \"" + prefs_api.ollama_host + "\" " +browser.i18n.getMessage("AndModel") + " \"" + prefs_api.ollama_model + "\"...", "info");
+        }catch(e){
+            console.error(">>>>>> [ThunderAI] messagesArea.appendUserMessage failed: " + JSON.stringify(e));
+        }
+        console.log(">>>>>> [ThunderAI] messagesArea.appendUserMessage done.");
         break;
     }
     case "openai_comp_api": {
@@ -105,13 +125,14 @@ switch (llm) {
         messagesArea.setLLMName(prefs_api.openai_comp_chat_name);
         worker.postMessage({ type: 'init', openai_comp_host: prefs_api.openai_comp_host, openai_comp_model: prefs_api.openai_comp_model, i18nStrings: i18nStrings});
         messagesArea.appendUserMessage(browser.i18n.getMessage("OpenAIComp_api_connecting") + " \"" + prefs_api.openai_comp_host + "\" " +browser.i18n.getMessage("AndModel") + " \"" + prefs_api.openai_comp_model + "\"...", "info");
+        browser.runtime.sendMessage({command: "openai_comp_api_ready_" + call_id, window_id: (await browser.windows.getCurrent()).id});
         break;
     }
 }
 
-
 // Event listeners for worker messages
 worker.onmessage = function(event) {
+    console.log(">>>>>>>>>> [ThunderAI] controller.js onmessage: " + JSON.stringify(event));
     const { type, payload } = event.data;
     switch (type) {
         case 'messageSent':
@@ -136,7 +157,7 @@ worker.onmessage = function(event) {
 
 // handling commands from the backgound page
 browser.runtime.onMessage.addListener((message, sender, sendResponse) => {
-    //console.log(">>>>>>>>>>>>> controller.js onMessage: " + JSON.stringify(message));
+    console.log(">>>>>>>>>>>>> controller.js onMessage: " + JSON.stringify(message));
     switch (message.command) {
         case "api_send":
             //send the received prompt to the llm api

api_webchat/messagesArea.js

@@ -143,7 +143,7 @@ class MessagesArea extends HTMLElement {
 
     appendUserMessage(messageText, type="user") {
         this.fullTextHTML = "";
-        // console.log("[ThunderAI] appendUserMessage: " + messageText);
+        console.log(">>>>>> [ThunderAI] appendUserMessage: " + messageText);
         const header = document.createElement('h2');
         let source = "You";
         switch (type) {
@@ -161,11 +161,12 @@ class MessagesArea extends HTMLElement {
         messageElement.classList.add('message', type);
         messageElement.textContent = messageText;
         this.messages.appendChild(messageElement);
+        console.log(">>>>>> [ThunderAI] appendUserMessage done.");
         this.scrollToBottom();
     }
 
     appendBotMessage(messageText, type="bot") {
-        // console.log("[ThunderAI] appendBotMessage: " + messageText);
+        console.log("(>>>>>> [ThunderAI] appendBotMessage: " + messageText);
 
         this.fullTextHTML = messageText;
 
@@ -182,6 +183,7 @@ class MessagesArea extends HTMLElement {
         messageElement.classList.add('message', type);
         messageElement.textContent = messageText;
         this.messages.appendChild(messageElement);
+        console.log("(>>>>>> [ThunderAI] appendBotMessage done.");
         this.scrollToBottom();
     }
 

manifest.json

@@ -2,7 +2,7 @@
     "manifest_version": 2,
     "name": "ThunderAI",
     "description": "__MSG_extensionDescription__",
-    "version": "2.2.0pre4",
+    "version": "2.2.0_i143_v5",
     "author": "Mic (m@micz.it)",
     "homepage_url": "https://micz.it/thunderbird-addon-thunderai/",
     "browser_specific_settings": {

mzta-background.js

@@ -239,6 +239,8 @@ async function openChatGPT(promptText, action, curr_tabId, prompt_name = '', do_
         return;
     }
 
+    let mailMessage = await browser.messageDisplay.getDisplayedMessage(curr_tabId);
+
     switch(prefs.connection_type){
         case 'chatgpt_web':
         // We are using the ChatGPT web interface
@@ -256,18 +258,12 @@ async function openChatGPT(promptText, action, curr_tabId, prompt_name = '', do_
             win_options.width = prefs.chatgpt_win_width,
             win_options.height = prefs.chatgpt_win_height
         }
-        let newWindow = await browser.windows.create(win_options);
-
-        let newWindowId = newWindow.id;
-        let createdTab = newWindow.tabs[0];
-
-        const listener = async (message, sender, sendResponse) => {
-
-            if (message.command === "chatgpt_web_ready_" + rand_call_id) {
 
+        const listener = (message, sender, sendResponse) => {
+            async function handleChatGptWeb(createdTab) {
                 taLog.log("[ThunderAI] ChatGPT web interface script started...");
         
-                let pre_script = `let mztaWinId = `+ newWindowId +`;
+                let pre_script = `let mztaWinId = `+ createdTab.windowId +`;
                 let mztaStatusPageDesc="`+ browser.i18n.getMessage("prefs_status_page") +`";
                 let mztaForceCompletionDesc="`+ browser.i18n.getMessage("chatgpt_force_completion") +`";
                 let mztaForceCompletionTitle="`+ browser.i18n.getMessage("chatgpt_force_completion_title") +`";
@@ -280,7 +276,7 @@ async function openChatGPT(promptText, action, curr_tabId, prompt_name = '', do_
                 taLog.log("Waiting 1 sec done");
                 
                 await browser.tabs.executeScript(createdTab.id, { code: pre_script + mzta_script, matchAboutBlank: false });
-                let mailMessage = await browser.messageDisplay.getDisplayedMessage(curr_tabId);
+                // let mailMessage = await browser.messageDisplay.getDisplayedMessage(curr_tabId);
                 let mailMessageId = -1;
                 if(mailMessage) mailMessageId = mailMessage.id;
                 browser.tabs.sendMessage(createdTab.id, { command: "chatgpt_send", prompt: promptText, action: action, tabId: curr_tabId, mailMessageId: mailMessageId});
@@ -288,9 +284,15 @@ async function openChatGPT(promptText, action, curr_tabId, prompt_name = '', do_
                 taLog.log("[ThunderAI] ChatGPT Web script injected successfully");
                 browser.runtime.onMessage.removeListener(listener);
             }
+        
+            if (message.command === "chatgpt_web_ready_" + rand_call_id) {
+                return handleChatGptWeb(sender.tab)
+            }
+            return false;
         }
 
         browser.runtime.onMessage.addListener(listener);
+        await browser.windows.create(win_options);
         break;  // chatgpt_web - END
 
     case 'chatgpt_api':
@@ -298,6 +300,35 @@ async function openChatGPT(promptText, action, curr_tabId, prompt_name = '', do_
 
         let rand_call_id2 = '_openai_' + generateCallID();
 
+        const listener2 = (message, sender, sendResponse) => {
+
+            function handleChatGptApi(createdTab) {
+                let mailMessageId2 = -1;
+                if(mailMessage) mailMessageId2 = mailMessage.id;
+
+                // check if the config is present, or give a message error
+                if (prefs.chatgpt_api_key == '') {
+                    browser.tabs.sendMessage(createdTab.id, { command: "api_error", error: browser.i18n.getMessage('chatgpt_empty_apikey')});
+                    return;
+                }
+                if (prefs.chatgpt_model == '') {
+                    browser.tabs.sendMessage(createdTab.id, { command: "api_error", error: browser.i18n.getMessage('chatgpt_empty_model')});
+                    return;
+                }
+                //console.log(">>>>>>>>>> sender: " + JSON.stringify(sender));
+                browser.tabs.sendMessage(createdTab.id, { command: "api_send", prompt: promptText, action: action, tabId: curr_tabId, mailMessageId: mailMessageId2, do_custom_text: do_custom_text});
+                taLog.log('[OpenAI ChatGPT] Connection succeded!');
+                browser.runtime.onMessage.removeListener(listener2);
+            }
+
+            if (message.command === "openai_api_ready_"+rand_call_id2) {
+                return handleChatGptApi(sender.tab);
+            }
+            return false;
+        }
+
+        browser.runtime.onMessage.addListener(listener2);
+
         let win_options2 = {
             url: browser.runtime.getURL('api_webchat/index.html?llm='+prefs.connection_type+'&call_id='+rand_call_id2),
             type: "popup",
@@ -312,35 +343,6 @@ async function openChatGPT(promptText, action, curr_tabId, prompt_name = '', do_
 
         await browser.windows.create(win_options2);
 
-        const listener2 = async (message, sender, sendResponse) => {
-
-            if (message.command === "openai_api_ready_"+rand_call_id2) {
-
-                let newWindow2 = await browser.windows.get(message.window_id, {populate: true});
-                let createdTab2 = newWindow2.tabs[0];
-
-                let mailMessage = await browser.messageDisplay.getDisplayedMessage(curr_tabId);
-                let mailMessageId2 = -1;
-                if(mailMessage) mailMessageId2 = mailMessage.id;
-
-                // check if the config is present, or give a message error
-                if (prefs.chatgpt_api_key == '') {
-                    browser.tabs.sendMessage(createdTab2.id, { command: "api_error", error: browser.i18n.getMessage('chatgpt_empty_apikey')});
-                    return;
-                }
-                if (prefs.chatgpt_model == '') {
-                    browser.tabs.sendMessage(createdTab2.id, { command: "api_error", error: browser.i18n.getMessage('chatgpt_empty_model')});
-                    return;
-                }
-
-                browser.tabs.sendMessage(createdTab2.id, { command: "api_send", prompt: promptText, action: action, tabId: curr_tabId, mailMessageId: mailMessageId2, do_custom_text: do_custom_text});
-                taLog.log('[OpenAI ChatGPT] Connection succeded!');
-                browser.runtime.onMessage.removeListener(listener2);
-            }
-        }
-
-        browser.runtime.onMessage.addListener(listener2);
-
         break;  // chatgpt_api - END
 
         case 'ollama_api':
@@ -350,6 +352,41 @@ async function openChatGPT(promptText, action, curr_tabId, prompt_name = '', do_
 
             let rand_call_id3 = '_ollama_' + generateCallID();
 
+            const listener3 = async (message, sender, sendResponse) => {
+
+                function handleOllamaApi(createdTab3) {
+                    taLog.log("Ollama API window ready.");
+                    taLog.log("message.window_id: " + message.window_id)
+                    taLog.log(">>>>>> createdTab3.id: " + createdTab3.id)
+                    // let mailMessage3 = await browser.messageDisplay.getDisplayedMessage(curr_tabId);
+                    let mailMessageId3 = -1;
+                    if(mailMessage) mailMessageId3 = mailMessage.id;
+                    taLog.log(">>>>>> mailMessageId3: " + mailMessageId3)
+            
+                    // check if the config is present, or give a message error
+                    if (prefs.ollama_host == '') {
+                        browser.tabs.sendMessage(createdTab3.id, { command: "api_error", error: browser.i18n.getMessage('ollama_empty_host')});
+                        return;
+                    }
+                    if (prefs.ollama_model == '') {
+                        browser.tabs.sendMessage(createdTab3.id, { command: "api_error", error: browser.i18n.getMessage('ollama_empty_model')});
+                        return;
+                    }
+                    console.log(">>>>>> [ThunderAI] Ollama API about to send message to createdTab3.id: " + createdTab3.id);
+                    browser.tabs.sendMessage(createdTab3.id, { command: "api_send", prompt: promptText, action: action, tabId: curr_tabId, mailMessageId: mailMessageId3, do_custom_text: do_custom_text});
+                    taLog.log('[Ollama API] Connection succeded!');
+                    browser.runtime.onMessage.removeListener(listener3);
+                }
+
+                if (message.command === "ollama_api_ready_"+rand_call_id3) {
+                    return handleOllamaApi(sender.tab);
+                }else{
+                    return false;
+                }
+            }
+
+            browser.runtime.onMessage.addListener(listener3);
+
             let win_options3 = {
                 url: browser.runtime.getURL('api_webchat/index.html?llm='+prefs.connection_type+'&call_id='+rand_call_id3),
                 type: "popup",
@@ -364,37 +401,6 @@ async function openChatGPT(promptText, action, curr_tabId, prompt_name = '', do_
 
             await browser.windows.create(win_options3);
 
-            const listener3 = async (message, sender, sendResponse) => {
-
-                if (message.command === "ollama_api_ready_"+rand_call_id3) {
-                    taLog.log("Ollama API window ready.");
-                    taLog.log("message.window_id: " + message.window_id)
-                    let newWindow3 = await browser.windows.get(message.window_id, {populate: true});
-                    let createdTab3 = newWindow3.tabs[0];
-                    taLog.log(">>>>>> createdTab3.id: " + createdTab3.id)
-                    let mailMessage3 = await browser.messageDisplay.getDisplayedMessage(curr_tabId);
-                    let mailMessageId3 = -1;
-                    if(mailMessage3) mailMessageId3 = mailMessage3.id;
-                    taLog.log(">>>>>> mailMessageId3: " + mailMessageId3)
-            
-                    // check if the config is present, or give a message error
-                    if (prefs.ollama_host == '') {
-                        await browser.tabs.sendMessage(createdTab3.id, { command: "api_error", error: browser.i18n.getMessage('ollama_empty_host')});
-                        return;
-                    }
-                    if (prefs.ollama_model == '') {
-                        await browser.tabs.sendMessage(createdTab3.id, { command: "api_error", error: browser.i18n.getMessage('ollama_empty_model')});
-                        return;
-                    }
-
-                    await browser.tabs.sendMessage(createdTab3.id, { command: "api_send", prompt: promptText, action: action, tabId: curr_tabId, mailMessageId: mailMessageId3, do_custom_text: do_custom_text});
-                    taLog.log('[Ollama API] Connection succeded!');
-                    browser.runtime.onMessage.removeListener(listener3);
-                }
-            }
-
-            browser.runtime.onMessage.addListener(listener3);
-
             break;  // ollama_api - END
 
             case 'openai_comp_api':
@@ -422,7 +428,7 @@ async function openChatGPT(promptText, action, curr_tabId, prompt_name = '', do_
                         let newWindow4 = await browser.windows.get(message.window_id, {populate: true});
                         let createdTab4 = newWindow4.tabs[0];
         
-                        let mailMessage = await browser.messageDisplay.getDisplayedMessage(curr_tabId);
+                        // let mailMessage = await browser.messageDisplay.getDisplayedMessage(curr_tabId);
                         let mailMessageId4 = -1;
                         if(mailMessage) mailMessageId4 = mailMessage.id;