109 lines
4.3 KiB
JavaScript
109 lines
4.3 KiB
JavaScript
/*
|
|
* ThunderAI [https://micz.it/thunderdbird-addon-thunderai/]
|
|
* Copyright (C) 2024 Mic (m@micz.it)
|
|
|
|
* This program is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
|
|
// Modified version derived from https://github.com/ali-raheem/Aify/blob/cfadf52f576b7be3720b5b73af7c8d3129c054da/plugin/html/actions.js
|
|
|
|
import { defaultPrompts } from '../js/mzta-prompts.js';
|
|
import { getLanguageDisplayName } from '../js/mzta-utils.js'
|
|
|
|
const addAction = (curr_prompt, promptsContainer) => {
|
|
const promptDiv = document.createElement("div");
|
|
promptDiv.classList.add("button");
|
|
const nameInput = document.createElement("p");
|
|
nameInput.classList.add("flat");
|
|
nameInput.innerText = curr_prompt.name;
|
|
nameInput.classList.add("prompt-name");
|
|
|
|
const getHighlight = async () => {
|
|
const tabs = await browser.tabs.query({ active: true, currentWindow: true });
|
|
return {tabId: tabs[0].id,
|
|
selection: await browser.tabs.sendMessage(tabs[0].id, { command: "getSelectedText" }),
|
|
text: await browser.tabs.sendMessage(tabs[0].id, { command: "getTextOnly" })
|
|
};
|
|
};
|
|
|
|
nameInput.onclick = async () => {
|
|
const msg_text = await getHighlight();
|
|
|
|
//check if a selection is needed
|
|
if(curr_prompt.need_selected && (msg_text.selection==='')){
|
|
//A selection is needed, but nothing is selected!
|
|
alert(browser.i18n.getMessage('prompt_selection_needed'));
|
|
return;
|
|
}
|
|
|
|
var body_text = '';
|
|
if (msg_text.selection!=='') {
|
|
body_text = msg_text.selection.replace(/\s+/g, ' ').trim();
|
|
} else {
|
|
body_text = msg_text.text.replace(/\s+/g, ' ').trim();
|
|
}
|
|
//open chatgpt window
|
|
//console.log("Click menu item...");
|
|
let prefs = await browser.storage.sync.get({default_chatgpt_lang: getLanguageDisplayName(browser.i18n.getUILanguage())});
|
|
let chatgpt_lang = prefs.default_chatgpt_lang;
|
|
|
|
var fullPrompt = curr_prompt.text + (curr_prompt.need_signature ? " " + await getDefaultSignature():"") + " " + browser.i18n.getMessage("prompt_lang") + chatgpt_lang + ". \"" + body_text + "\" ";
|
|
|
|
switch(curr_prompt.id){
|
|
case 'prompt_translate_this':
|
|
fullPrompt = curr_prompt.text + chatgpt_lang + ". \"" + body_text + "\" ";
|
|
break;
|
|
case 'prompt_reply':
|
|
fullPrompt += "Do not add the subject line to the response."
|
|
break;
|
|
default:
|
|
break;
|
|
}
|
|
|
|
const tabs = await browser.tabs.query({ active: true, currentWindow: true });
|
|
browser.runtime.sendMessage({command: "chatgpt_open", prompt: fullPrompt, action: curr_prompt.action, tabId: tabs[0].id});
|
|
};
|
|
promptDiv.appendChild(nameInput);
|
|
promptsContainer.appendChild(promptDiv);
|
|
};
|
|
|
|
async function getDefaultSignature(){
|
|
let prefs = await browser.storage.sync.get({default_sign_name: ''});
|
|
if(prefs.default_sign_name===''){
|
|
return '';
|
|
}else{
|
|
return "Sign the message as " + prefs.default_sign_name + ".";
|
|
}
|
|
}
|
|
|
|
async function checkCompose(){
|
|
return await browser.windows.getCurrent().then((currWindow) => {
|
|
//console.log("currWindow.type: "+currWindow.type);
|
|
return currWindow.type == 'messageCompose';
|
|
});
|
|
}
|
|
|
|
document.addEventListener("DOMContentLoaded", async () => {
|
|
let inCompose = await checkCompose();
|
|
//console.log("inCompose: "+inCompose)
|
|
const promptsContainer = document.getElementById("prompts-container");
|
|
defaultPrompts.forEach((prompt) => {
|
|
if((prompt.type == 0)||((prompt.type == 1)&&(!inCompose))||((prompt.type == 2)&&(inCompose))){
|
|
addAction(prompt, promptsContainer)
|
|
}
|
|
});
|
|
i18n.updateDocument();
|
|
});
|
|
|