65 lines
2.7 KiB
JavaScript
65 lines
2.7 KiB
JavaScript
// Modified version derived from https://github.com/ali-raheem/Aify/blob/cfadf52f576b7be3720b5b73af7c8d3129c054da/plugin/html/actions.js
|
|
|
|
import { defaultPrompts } from '../js/mzta-prompts.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...");
|
|
var fullPrompt = curr_prompt.text + " " + browser.i18n.getMessage("prompt_lang") + browser.i18n.getMessage("prompt_translate_lang") + " \"" + msg_text.text + "\" ";
|
|
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 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 actionsContainer = document.getElementById("prompts-container");
|
|
defaultPrompts.forEach((prompt) => {
|
|
if((prompt.type == 0)||((prompt.type == 1)&&(!inCompose))||((prompt.type == 2)&&(inCompose))){
|
|
addAction(prompt, actionsContainer)
|
|
}
|
|
});
|
|
i18n.updateDocument();
|
|
});
|
|
|