231 lines
8.6 KiB
JavaScript
231 lines
8.6 KiB
JavaScript
/*
|
|
* ThunderAI [https://micz.it/thunderbird-addon-thunderai/]
|
|
* Copyright (C) 2024 - 2026 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/>.
|
|
*/
|
|
|
|
import {
|
|
prefs_default,
|
|
integration_options_config
|
|
} from '../../options/mzta-options-default.js';
|
|
import { taLogger } from "../../js/mzta-logger.js";
|
|
import {
|
|
getSpecialPrompts,
|
|
setSpecialPrompts
|
|
} from "../../js/mzta-prompts.js";
|
|
import {
|
|
getPlaceholders,
|
|
mapPlaceholderToSuggestion
|
|
} from "../../js/mzta-placeholders.js";
|
|
import { textareaAutocomplete } from "../../js/mzta-placeholders-autocomplete.js";
|
|
import {
|
|
normalizeStringList,
|
|
isAPIKeyValue,
|
|
setTomSelectBorder
|
|
} from "../../js/mzta-utils.js";
|
|
import {
|
|
initializeSpecificIntegrationUI
|
|
} from "../_lib/connection-ui.js";
|
|
|
|
let autocompleteSuggestions = [];
|
|
let taLog = new taLogger("mzta-translate-page", true);
|
|
|
|
document.addEventListener("DOMContentLoaded", async () => {
|
|
|
|
let specialPrompts = await getSpecialPrompts();
|
|
let translate_prompt = specialPrompts.find((prompt) => prompt.id === 'prompt_translate_this');
|
|
|
|
if (translate_prompt && translate_prompt.api_type && translate_prompt.api_type !== '') {
|
|
let update_prefs = {};
|
|
update_prefs['translate_connection_type'] = translate_prompt.api_type;
|
|
|
|
let integration = translate_prompt.api_type.replace('_api', '');
|
|
if (integration_options_config && integration_options_config[integration]) {
|
|
for (const key of Object.keys(integration_options_config[integration])) {
|
|
if (translate_prompt[key] !== undefined) {
|
|
update_prefs[`translate_${integration}_${key}`] = translate_prompt[key];
|
|
}
|
|
}
|
|
}
|
|
await browser.storage.sync.set(update_prefs);
|
|
}
|
|
|
|
await initializeSpecificIntegrationUI({
|
|
prefix: 'translate',
|
|
promptId: 'prompt_translate_this',
|
|
taLog: taLog,
|
|
restoreOptionsCallback: restoreOptions
|
|
});
|
|
|
|
i18n.updateDocument();
|
|
|
|
document.querySelectorAll(".option-input").forEach(element => {
|
|
element.addEventListener("change", saveOptions);
|
|
});
|
|
let translate_textarea = document.getElementById("translate_prompt_text");
|
|
let translate_save_btn = document.getElementById("btn_save_prompt");
|
|
let translate_reset_btn = document.getElementById("btn_reset_prompt");
|
|
|
|
// on changing textarea
|
|
translate_textarea.addEventListener("input", (event) => {
|
|
translate_reset_btn.disabled = (event.target.value === browser.i18n.getMessage('prompt_translate_this_full_text'));
|
|
translate_save_btn.disabled = (event.target.value === translate_prompt.text);
|
|
});
|
|
|
|
// on clicking reset button
|
|
translate_reset_btn.addEventListener("click", () => {
|
|
translate_textarea.value = browser.i18n.getMessage("prompt_translate_this_full_text");
|
|
translate_reset_btn.disabled = true;
|
|
let event = new Event("input", { bubbles: true, cancelable: true });
|
|
translate_textarea.dispatchEvent(event);
|
|
});
|
|
|
|
// on clicking save button
|
|
translate_save_btn.addEventListener("click", () => {
|
|
specialPrompts.find(prompt => prompt.id === 'prompt_translate_this').text = translate_textarea.value;
|
|
setSpecialPrompts(specialPrompts);
|
|
translate_save_btn.disabled = true;
|
|
browser.runtime.sendMessage({ command: "reload_menus" });
|
|
});
|
|
|
|
if(translate_prompt.text === 'prompt_translate_this_full_text'){
|
|
translate_prompt.text = browser.i18n.getMessage(translate_prompt.text);
|
|
}
|
|
|
|
translate_textarea.value = translate_prompt.text;
|
|
translate_reset_btn.disabled = (translate_textarea.value === browser.i18n.getMessage("prompt_translate_this_full_text"));
|
|
|
|
autocompleteSuggestions = (await getPlaceholders(true))
|
|
.filter((p) => p.id !== "additional_text")
|
|
.map(mapPlaceholderToSuggestion);
|
|
|
|
textareaAutocomplete(translate_textarea, autocompleteSuggestions, 1);
|
|
|
|
});
|
|
|
|
// Methods to manage options, derived from: /options/mzta-options.js
|
|
|
|
function saveOptions(e) {
|
|
e.preventDefault();
|
|
let options = {};
|
|
let element = e.target;
|
|
switch (element.type) {
|
|
case 'checkbox':
|
|
options[element.id] = element.checked;
|
|
break;
|
|
case 'number':
|
|
options[element.id] = element.valueAsNumber;
|
|
break;
|
|
case 'text':
|
|
case 'password':
|
|
options[element.id] = element.value.trim();
|
|
break;
|
|
case 'select-one':
|
|
if (element.id === 'translate_auto') {
|
|
options[element.id] = parseInt(element.value, 10);
|
|
} else {
|
|
options[element.id] = element.value;
|
|
}
|
|
break;
|
|
case 'textarea':
|
|
options[element.id] = normalizeStringList(element.value);
|
|
break;
|
|
default:
|
|
console.error("[ThunderAI] Unhandled input type:", element.type);
|
|
}
|
|
|
|
browser.storage.sync.set(options);
|
|
}
|
|
|
|
async function restoreOptions() {
|
|
function setCurrentChoice(result) {
|
|
document.querySelectorAll(".option-input").forEach(element => {
|
|
if(!element.id) return;
|
|
taLog.log("Options restoring " + element.id + " = " + (isAPIKeyValue(element.id) ? "****************" : result[element.id]));
|
|
switch (element.type) {
|
|
case 'checkbox':
|
|
element.checked = result[element.id] || false;
|
|
break;
|
|
case 'number':
|
|
element.value = result[element.id] ?? 0;
|
|
break;
|
|
case 'text':
|
|
case 'textarea':
|
|
case 'password':
|
|
let default_text_value = '';
|
|
if(element.id == 'translate_lang') default_text_value = prefs_default.default_chatgpt_lang;
|
|
element.value = result[element.id] || default_text_value;
|
|
break;
|
|
default:
|
|
if (element.tagName === 'SELECT') {
|
|
let default_select_value = 0;
|
|
if (element.id === 'translate_auto') {
|
|
default_select_value = prefs_default.translate_auto;
|
|
}
|
|
const restoreValue = result[element.id] ?? default_select_value;
|
|
let optionExists = Array.from(element.options).some(opt => opt.value === String(restoreValue));
|
|
if (element.tomselect) {
|
|
if (!optionExists && restoreValue !== '') {
|
|
element.tomselect.addOption({ value: String(restoreValue), text: String(restoreValue) });
|
|
}
|
|
element.tomselect.setValue(String(restoreValue), true);
|
|
setTomSelectBorder(element.tomselect);
|
|
} else {
|
|
if (!optionExists && restoreValue !== '') {
|
|
let newOption = new Option(restoreValue, restoreValue);
|
|
element.add(newOption);
|
|
}
|
|
element.value = restoreValue;
|
|
if (element.value === '') {
|
|
element.selectedIndex = 0;
|
|
}
|
|
}
|
|
}else{
|
|
console.error("[ThunderAI] Unhandled input type:", element.type);
|
|
}
|
|
}
|
|
});
|
|
}
|
|
|
|
let getting = await browser.storage.sync.get(prefs_default);
|
|
|
|
let specialPrompts = await getSpecialPrompts();
|
|
let translate_prompt = specialPrompts.find(prompt => prompt.id === 'prompt_translate_this');
|
|
|
|
if (translate_prompt) {
|
|
if (translate_prompt.api_type && translate_prompt.api_type !== '') {
|
|
getting['translate_connection_type'] = translate_prompt.api_type;
|
|
} else {
|
|
getting['translate_connection_type'] = getting['connection_type'];
|
|
}
|
|
for (const [integration, options] of Object.entries(integration_options_config)) {
|
|
for (const key of Object.keys(options)) {
|
|
const propName = `${integration}_${key}`;
|
|
if (translate_prompt[propName] !== undefined && translate_prompt[propName] !== '') {
|
|
getting[`translate_${propName}`] = translate_prompt[propName];
|
|
} else {
|
|
getting[`translate_${propName}`] = getting[propName];
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
// If translate_lang is empty, show default_chatgpt_lang as placeholder/default
|
|
if (!getting['translate_lang']) {
|
|
getting['translate_lang'] = '';
|
|
}
|
|
|
|
setCurrentChoice(getting);
|
|
}
|