278 lines
12 KiB
JavaScript
278 lines
12 KiB
JavaScript
/*
|
|
* ThunderAI [https://micz.it/thunderbird-addon-thunderai/]
|
|
* Copyright (C) 2024 - 2025 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 {
|
|
getAccountsList,
|
|
normalizeStringList,
|
|
isAPIKeyValue
|
|
} from "../../js/mzta-utils.js";
|
|
import {
|
|
initializeSpecificIntegrationUI
|
|
} from "../_lib/connection-ui.js";
|
|
|
|
let autocompleteSuggestions = [];
|
|
let taLog = new taLogger("mzta-summarize-page", true);
|
|
|
|
document.addEventListener("DOMContentLoaded", async () => {
|
|
|
|
let specialPrompts = await getSpecialPrompts();
|
|
let summarize_prompt = specialPrompts.find((prompt) => prompt.id === 'prompt_summarize');
|
|
let summarize_email_template = specialPrompts.find((prompt) => prompt.id === 'prompt_summarize_email_template');
|
|
let summarize_email_separator = specialPrompts.find((prompt) => prompt.id === 'prompt_summarize_email_separator');
|
|
|
|
if (summarize_prompt && summarize_prompt.api_type && summarize_prompt.api_type !== '') {
|
|
let update_prefs = {};
|
|
update_prefs['summarize_connection_type'] = summarize_prompt.api_type;
|
|
|
|
let integration = summarize_prompt.api_type.replace('_api', '');
|
|
if (integration_options_config && integration_options_config[integration]) {
|
|
for (const key of Object.keys(integration_options_config[integration])) {
|
|
if (summarize_prompt[key] !== undefined) {
|
|
update_prefs[`summarize_${integration}_${key}`] = summarize_prompt[key];
|
|
}
|
|
}
|
|
}
|
|
await browser.storage.sync.set(update_prefs);
|
|
}
|
|
|
|
await initializeSpecificIntegrationUI({
|
|
prefix: 'summarize',
|
|
promptId: 'prompt_summarize',
|
|
taLog: taLog,
|
|
restoreOptionsCallback: restoreOptions
|
|
});
|
|
|
|
i18n.updateDocument();
|
|
|
|
document.querySelectorAll(".option-input").forEach(element => {
|
|
element.addEventListener("change", saveOptions);
|
|
});
|
|
let prefs_summarize = await browser.storage.sync.get({ summarize_enabled_accounts: [], connection_type: 'chatgpt_web' });
|
|
|
|
let summarize_textarea = document.getElementById("summarize_prompt_text");
|
|
let summarize_save_btn = document.getElementById("btn_save_prompt");
|
|
let summarize_reset_btn = document.getElementById("btn_reset_prompt");
|
|
let summarize_textarea_email_template = document.getElementById("summarize_email_template_text");
|
|
let summarize_reset_email_template_btn = document.getElementById("btn_reset_email_template");
|
|
let summarize_save_email_template_btn = document.getElementById("btn_save_email_template");
|
|
let summarize_email_separator_textarea = document.getElementById("summarize_email_separator_text");
|
|
let summarize_email_separator_save_btn = document.getElementById("btn_save_email_separator");
|
|
let summarize_email_separator_reset_btn = document.getElementById("btn_reset_email_separator");
|
|
|
|
|
|
// on changing textareas
|
|
summarize_textarea.addEventListener("input", (event) => {
|
|
summarize_reset_btn.disabled = (event.target.value === browser.i18n.getMessage('prompt_summarize_full_text'));
|
|
summarize_save_btn.disabled = (event.target.value === summarize_prompt.text);
|
|
});
|
|
|
|
summarize_textarea_email_template.addEventListener("input", (event) => {
|
|
summarize_reset_email_template_btn.disabled = (event.target.value === browser.i18n.getMessage('prompt_summarize_email_template_full_text'));
|
|
summarize_save_email_template_btn.disabled = (event.target.value === summarize_email_template.text);
|
|
});
|
|
|
|
summarize_email_separator_textarea.addEventListener("input", (event) => {
|
|
summarize_email_separator_reset_btn.disabled = (event.target.value === browser.i18n.getMessage('prompt_summarize_email_separator_full_text'));
|
|
summarize_email_separator_save_btn.disabled = (event.target.value === summarize_email_separator.text);
|
|
});
|
|
|
|
|
|
// on clicking buttons, reset
|
|
summarize_reset_email_template_btn.addEventListener("click", () => {
|
|
summarize_textarea_email_template.value = browser.i18n.getMessage("prompt_summarize_email_template_full_text");
|
|
summarize_reset_email_template_btn.disabled = true;
|
|
let event = new Event("input", { bubbles: true, cancelable: true });
|
|
summarize_textarea_email_template.dispatchEvent(event);
|
|
});
|
|
|
|
summarize_reset_btn.addEventListener("click", () => {
|
|
summarize_textarea.value = browser.i18n.getMessage("prompt_summarize_full_text");
|
|
summarize_reset_btn.disabled = true;
|
|
let event = new Event("input", { bubbles: true, cancelable: true });
|
|
summarize_textarea.dispatchEvent(event);
|
|
});
|
|
|
|
summarize_email_separator_reset_btn.addEventListener("click", () => {
|
|
summarize_email_separator_textarea.value = browser.i18n.getMessage("prompt_summarize_email_separator_full_text");
|
|
summarize_email_separator_reset_btn.disabled = true;
|
|
let event = new Event("input", { bubbles: true, cancelable: true });
|
|
summarize_email_separator_textarea.dispatchEvent(event);
|
|
});
|
|
|
|
// on clicking buttons, save
|
|
summarize_save_email_template_btn.addEventListener("click", () => {
|
|
specialPrompts.find(prompt => prompt.id === 'prompt_summarize_email_template').text = summarize_textarea_email_template.value;
|
|
setSpecialPrompts(specialPrompts);
|
|
summarize_save_email_template_btn.disabled = true;
|
|
browser.runtime.sendMessage({ command: "reload_menus" });
|
|
});
|
|
|
|
summarize_save_btn.addEventListener("click", () => {
|
|
specialPrompts.find(prompt => prompt.id === 'prompt_summarize').text = summarize_textarea.value;
|
|
setSpecialPrompts(specialPrompts);
|
|
summarize_save_btn.disabled = true;
|
|
browser.runtime.sendMessage({ command: "reload_menus" });
|
|
});
|
|
|
|
summarize_email_separator_save_btn.addEventListener("click", () => {
|
|
specialPrompts.find(prompt => prompt.id === 'prompt_summarize_email_separator').text = summarize_email_separator_textarea.value;
|
|
setSpecialPrompts(specialPrompts);
|
|
summarize_email_separator_save_btn.disabled = true;
|
|
browser.runtime.sendMessage({ command: "reload_menus" });
|
|
});
|
|
|
|
|
|
if(summarize_prompt.text === 'prompt_summarize_full_text'){
|
|
summarize_prompt.text = browser.i18n.getMessage(summarize_prompt.text);
|
|
}
|
|
if(summarize_email_template.text === 'prompt_summarize_email_template_full_text'){
|
|
summarize_email_template.text = browser.i18n.getMessage(summarize_email_template.text);
|
|
}
|
|
if(summarize_email_separator.text === 'prompt_summarize_email_separator_full_text'){
|
|
summarize_email_separator.text = browser.i18n.getMessage(summarize_email_separator.text);
|
|
}
|
|
|
|
summarize_textarea_email_template.value = summarize_email_template.text;
|
|
summarize_reset_email_template_btn.disabled = (summarize_textarea_email_template.value === browser.i18n.getMessage("prompt_summarize_email_template_full_text"));
|
|
summarize_textarea.value = summarize_prompt.text;
|
|
summarize_reset_btn.disabled = (summarize_textarea.value === browser.i18n.getMessage("prompt_summarize_full_text"));
|
|
summarize_email_separator_textarea.value = summarize_email_separator.text;
|
|
summarize_email_separator_reset_btn.disabled = (summarize_email_separator_textarea.value === browser.i18n.getMessage("prompt_summarize_email_separator_full_text"));
|
|
|
|
autocompleteSuggestions = (await getPlaceholders(true))
|
|
.filter((p) => p.id !== "additional_text")
|
|
.map(mapPlaceholderToSuggestion);
|
|
|
|
textareaAutocomplete(summarize_textarea, autocompleteSuggestions, 1);
|
|
textareaAutocomplete(summarize_textarea_email_template, autocompleteSuggestions, 1);
|
|
textareaAutocomplete(summarize_email_separator_textarea, autocompleteSuggestions, 1);
|
|
|
|
});
|
|
|
|
// Methods to manage options, derived from: /options/mzta-options.js
|
|
|
|
function saveOptions(e) {
|
|
e.preventDefault();
|
|
let options = {};
|
|
let element = e.target;
|
|
// console.log(">>>>>>>>>> Saving option: " + element.id + " = " + element.value);
|
|
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':
|
|
// console.log(">>>>>>>>>> Saving option [select-one]: " + element.id + " = " + element.value);
|
|
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 => {
|
|
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':
|
|
let default_number_value = 0;
|
|
if(element.id == 'chatgpt_win_height') default_number_value = prefs_default.chatgpt_win_height;
|
|
if(element.id == 'chatgpt_win_width') default_number_value = prefs_default.chatgpt_win_width;
|
|
element.value = result[element.id] ?? default_number_value;
|
|
break;
|
|
case 'text':
|
|
case 'textarea':
|
|
case 'password':
|
|
let default_text_value = '';
|
|
if(element.id == 'default_chatgpt_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 = '';
|
|
const restoreValue = result[element.id] || default_select_value;
|
|
// Check if option exists
|
|
let optionExists = Array.from(element.options).some(opt => opt.value === restoreValue);
|
|
// If it doesn't exist and restoreValue is not empty, create it
|
|
if (!optionExists && restoreValue !== '') {
|
|
let newOption = new Option(restoreValue, restoreValue);
|
|
element.add(newOption);
|
|
}
|
|
// Set value
|
|
element.value = restoreValue;
|
|
if (element.value === '') {
|
|
element.selectedIndex = -1;
|
|
}
|
|
}else{
|
|
console.error("[ThunderAI] Unhandled input type:", element.type);
|
|
}
|
|
}
|
|
});
|
|
}
|
|
|
|
let getting = await browser.storage.sync.get(prefs_default);
|
|
|
|
let specialPrompts = await getSpecialPrompts();
|
|
let addtags_prompt = specialPrompts.find(prompt => prompt.id === 'prompt_summarize');
|
|
|
|
if (addtags_prompt) {
|
|
if (addtags_prompt.api_type && addtags_prompt.api_type !== '') {
|
|
getting['summarize_connection_type'] = addtags_prompt.api_type;
|
|
} else {
|
|
getting['summarize_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 (addtags_prompt[propName] !== undefined && addtags_prompt[propName] !== '') {
|
|
getting[`summarize_${propName}`] = addtags_prompt[propName];
|
|
} else {
|
|
getting[`summarize_${propName}`] = getting[propName];
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
setCurrentChoice(getting);
|
|
}
|