From 639fe495a7416df73b6b23633258cc7953d33cb0 Mon Sep 17 00:00:00 2001
From: mic <m@micz.it>
Date: Tue, 18 Jun 2024 21:08:03 +0200
Subject: [PATCH] replying to the right message. fixes #71

---
 js/mzta-chatgpt.js |  8 ++++----
 mzta-background.js | 30 +++++++++++++++++++++---------
 2 files changed, 25 insertions(+), 13 deletions(-)

diff --git a/js/mzta-chatgpt.js b/js/mzta-chatgpt.js
index 5bf7e4de..9d2f3d28 100644
--- a/js/mzta-chatgpt.js
+++ b/js/mzta-chatgpt.js
@@ -119,7 +119,7 @@ function chatpgt_scrollToBottom () {
     catch (err) { console.error('[ThunderAI] ', err); }
 }
 
-function addCustomDiv(prompt_action,tabId) {
+function addCustomDiv(prompt_action,tabId,mailMessageId) {
     // Create <style> element for the CSS
     var style = document.createElement('style');
     style.textContent = ".mzta-header-fixed {position: fixed;bottom: 0;left: 0;height:100px;width: 100%;background-color: #333;color: white;text-align: center;padding: 10px 0;z-index: 1000;border-top: 3px solid white;}"
@@ -205,7 +205,7 @@ function addCustomDiv(prompt_action,tabId) {
             btn_ok.textContent = browser.i18n.getMessage("chatgpt_win_get_answer");
             btn_ok.onclick = async function() {
                 const response = await chatgpt_getFromDOM('last');
-                browser.runtime.sendMessage({command: "chatgpt_replyMessage", text: response, tabId: tabId});
+                browser.runtime.sendMessage({command: "chatgpt_replyMessage", text: response, tabId: tabId, mailMessageId: mailMessageId});
                 //console.log(response);
                 browser.runtime.sendMessage({command: "chatgpt_close"});
             };
@@ -215,7 +215,7 @@ function addCustomDiv(prompt_action,tabId) {
             btn_ok.onclick = async function() {
                 const response = await chatgpt_getFromDOM('last');
                 //console.log('replace text: '+tabId)
-                browser.runtime.sendMessage({command: "chatgpt_replaceSelectedText", text: response, tabId: tabId});
+                browser.runtime.sendMessage({command: "chatgpt_replaceSelectedText", text: response, tabId: tabId, mailMessageId: mailMessageId});
                 //console.log(response);
                 browser.runtime.sendMessage({command: "chatgpt_close"});
             };
@@ -344,7 +344,7 @@ browser.runtime.onMessage.addListener((message, sender, sendResponse) => {
             // User not logged in
             alert(browser.i18n.getMessage("chatgpt_user_not_logged_in"));
         }else{
-            addCustomDiv(message.action,message.tabId);
+            addCustomDiv(message.action,message.tabId,message.mailMessageId);
             (async () => {
                 if(mztaDoCustomText === 1){
                     current_message = message;
diff --git a/mzta-background.js b/mzta-background.js
index a3981046..1f2eefcb 100644
--- a/mzta-background.js
+++ b/mzta-background.js
@@ -76,14 +76,22 @@ messenger.runtime.onMessage.addListener(async (message, sender, sendResponse) =>
                     replyType = 'replyToSender';
                 }
                 //console.log('replyType: ' + replyType);
-                browser.messageDisplay.getDisplayedMessage(message.tabId).then(async (mailMessage) => {
-                    let reply_tab = await browser.compose.beginReply(mailMessage.id, replyType, {
-                        type: "reply",
-                        //body:  paragraphsHtmlString,
-                        isPlainText: false,
-                        identityId: await getCurrentIdentity(mailMessage),
-                    })
-                    
+                // browser.messageDisplay.getDisplayedMessage(message.tabId).then(async (mailMessage) => {
+                //     let reply_tab = await browser.compose.beginReply(mailMessage.id, replyType, {
+                //         type: "reply",
+                //         //body:  paragraphsHtmlString,
+                //         isPlainText: false,
+                //         identityId: await getCurrentIdentity(mailMessage),
+                //     })
+                //console.log(">>>>>>>>>>>> message.mailMessageId: " + message.mailMessageId);
+                browser.messages.get(message.mailMessageId).then(async (mailMessage) => {
+                let curr_idn = await getCurrentIdentity(mailMessage)
+                let reply_tab = await browser.compose.beginReply(mailMessage.id, replyType, {
+                    type: "reply",
+                    //body:  paragraphsHtmlString,
+                    isPlainText: false,
+                    identityId: curr_idn,
+                })
                     // Wait for tab loaded.
                     await new Promise(resolve => {
                         const tabIsLoaded = tab => {
@@ -92,6 +100,7 @@ messenger.runtime.onMessage.addListener(async (message, sender, sendResponse) =>
                         const listener = (tabId, changeInfo, updatedTab) => {
                             if (tabIsLoaded(updatedTab)) {
                                 browser.tabs.onUpdated.removeListener(listener);
+                                console.log(">>>>>>>>>>>> reply_tab: " + tabId);
                                 resolve();
                             }
                         }
@@ -174,7 +183,10 @@ async function openChatGPT(promptText, action, curr_tabId, do_custom_text = 0) {
     browser.tabs.executeScript(createdTab.id, { code: pre_script + mzta_script, matchAboutBlank: false })
         .then(async () => {
             console.log("[ThunderAI] Script injected successfully");
-            browser.tabs.sendMessage(createdTab.id, { command: "chatgpt_send", prompt: promptText, action: action, tabId: 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});
         })
         .catch(err => {
             console.error("[ThunderAI] Error injecting the script: ", err);