ThunderAI/js/mzta-prompts.js

/*
 *  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 prompts text derived from https://github.com/ali-raheem/Aify/blob/13ff87583bc520fb80f555ab90a90c5c9df797a7/plugin/html/globals.js

/*  ================= PROMPTS PROPERTIES ========================================

    ================ BASE PROPERTIES

    Types (type attribute):
    0: always show (when composing a part of the text must be selected if need_selected = 1)
    1: show when reading an email
    2: show when composing a mail (a part of the text must be selected if need_selected = 1)

    Actions (action attribute):
    0: close button
    1: do reply
    2: substitute text

    Only if text selected (need_selected attribute):
    0: no selection needed (use all the message body)
    1: need a selection

    Signature (need_signature attribute):
    0: No signature needed
    1: Signature needed

    Custom Text (need_custom_text attribute):
    0: No custom text needed
    1: Custom text needed

    ================ USER PROPERTIES
    Enabled (enabled attribute):
    0: Disabled
    1: Enabled

    Position (position attribute):
    <num>: position number

*/

const defaultPrompts = [
    {
        id: 'prompt_reply',
        name: "__MSG_prompt_reply__",
        text: "Reply to the following email.",
        type: 1,
        action: 1,
        need_selected: 0,
        need_signature: 1,
        need_custom_text: 0,
    },
    {
        id: 'prompt_rewrite_polite',
        name: "__MSG_prompt_rewrite_polite__",
        text: "Rewrite the following text to be more polite. Reply with only the re-written text and with no extra comments or other text.",
        type: 2,
        action: 2,
        need_selected: 1,
        need_signature: 0,
        need_custom_text: 0,
    },
    {
        id: 'prompt_rewrite_formal',
        name: "__MSG_prompt_rewrite_formal__",
        text: "Rewrite the following text to be more formal. Reply with only the re-written text and with no extra comments or other text.",
        type: 2,
        action: 2,
        need_selected: 1,
        need_signature: 0,
        need_custom_text: 0,
    },
    {
        id: 'prompt_classify',
        name: "__MSG_prompt_classify__",
        text: "Classify the following text in terms of Politeness, Warmth, Formality, Assertiveness, Offensiveness giving a percentage for each category. Reply with only the category and score with no extra comments or other text.",
        type: 0,
        action: 0,
        need_selected: 0,
        need_signature: 0,
        need_custom_text: 0,
    },
    {
        id: 'prompt_summarize_this',
        name: "__MSG_prompt_summarize_this__",
        text: "Summarize the following email into a bullet point list.",
        type: 0,
        action: 0,
        need_selected: 0,
        need_signature: 0,
        need_custom_text: 0,
    },
    {
        id: 'prompt_translate_this',
        name: "__MSG_prompt_translate_this__",
        text: "Translate the following email in ",
        type: 0,
        action: 0,
        need_selected: 0,
        need_signature: 0,
        need_custom_text: 0,
    },
    {
        id: 'prompt_this',
        name: "__MSG_prompt_this__",
        text: "Reply with only the needed text and with no extra comments or other text.",
        type: 2,
        action: 2,
        need_selected: 1,
        need_signature: 0,
        need_custom_text: 0,
    },
];


export async function getPrompts(){
    //return await Promise.all([getDefaultPrompts_withProps(), getCustomPrompts()]).then(([defaultPrompts, customPrompts]) => [...defaultPrompts, ...customPrompts]);
    const defaultPrompts = await getDefaultPrompts_withProps();
    const customPrompts = await getCustomPrompts();
    return defaultPrompts.concat(customPrompts);
}


async function getDefaultPrompts_withProps() {
    let prefs = await browser.storage.sync.get({_default_prompts_properties: null});
    let defaultPrompts_prop = [...defaultPrompts];
    if(prefs._default_prompts_properties === null){     // no default prompts properties saved
        let pos = 1;
        defaultPrompts_prop.forEach((prompt) => {
            prompt.position = pos;
            prompt.enabled = 1;
            pos++;
        })
    } else {    // we have saved the default prompts properties
        defaultPrompts_prop.forEach((prompt) => {
            prompt.position = prefs._default_prompts_properties[prompt.id].position;
            prompt.enabled = prefs._default_prompts_properties[prompt.id].enabled;
        })
    }
    return defaultPrompts_prop;
}


async function getCustomPrompts() {
    let prefs = await browser.storage.sync.get({_custom_prompt: null});
    if(prefs._custom_prompt === null){
        return [];
    } else {
        return prefs._custom_prompt;
    }
}

export async function setDefaultPromptsProperties(prompts) {
    let default_prompts_properties = {};
    prompts.forEach((prompt) => {
        default_prompts_properties[prompt.id] = {position: prompt.position, enabled: prompt.enabled};
    });
    await browser.storage.sync.set({_default_prompts_properties: default_prompts_properties});
}


export async function setCustomPrompts(prompts) {
    await browser.storage.sync.set({_custom_prompt: prompts});
}