/*
 *  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/>.
 */


// Some original methods derived from https://github.com/KudoAI/chatgpt.js/blob/7eb8463cd61143fa9e1d5a8ec3c14d3c1b286e54/chatgpt.js
// Using a full string to inject it in the ChatGPT page to avoid any security error

export const mzta_script = `
let force_go = false;
let do_force_completion = false;
let current_message = null;
let current_action = null;
let current_tabId = null;
let current_mailMessageId = null;
let selectionChangeTimeout = null;
let isDragging = false;
let delay_wait_completion = 7000; // milliseconds
let _customTextArray = [];
let _currentCustomTextIndex = 0;
let lastSelectedHtml = "";

async function chatgpt_sendMsg(msg, method ='') {       // return -1 send button not found, -2 textarea not found
    let textArea = document.getElementById('prompt-textarea')
    //check if the textarea has been found
    if(!textArea) {
        console.error("[ThunderAI] Textarea not found!");
        return -2;
    }
    // #prompt-textarea is a ProseMirror contenteditable. Place the caret at
    // the end of the inner paragraph and let execCommand('insertText') feed
    // the text through the browser's input pipeline so PM can reconcile.
    textArea.focus();
    const pmTarget = textArea.querySelector('p') || textArea;
    const sel = window.getSelection();
    const range = document.createRange();
    range.selectNodeContents(pmTarget);
    range.collapse(false);
    sel.removeAllRanges();
    sel.addRange(range);
    document.execCommand('insertText', false, msg);
    //wait for the button to change from the audio button to the send button (from nov-2024)
    await new Promise(resolve => setTimeout(resolve, 1000));
    let sendButton = document.querySelector('[data-testid="send-button"]') // pre-GPT-4o
        || document.querySelector('path[d*="M15.1918 8.90615C15.6381"]')?.parentNode.parentNode  // from sept-2024;
        || document.querySelector('path[d*="M15.192 8.906a1.143"]')?.parentNode.parentNode;  // post-GPT-4o;
    //check if the sendbutton has been found
    if (!sendButton) {
        console.error("[ThunderAI] Send button not found!");
        return -1;
    }
    const delaySend = setInterval(() => {
        //console.log(">>>>>>>>>> sendButton disabled: " + sendButton?.hasAttribute('disabled'));
        if (!sendButton?.hasAttribute('disabled')) { // send msg
            // Always click the send button: a synthetic KeyboardEvent with
            // only keyCode:13 desyncs ProseMirror and crashes the React app.
            sendButton.click();
            clearInterval(delaySend);
        }else{
            //console.error(">>>>>>>>>> The sendButton seems to be disabled! Trying again...");
         sendButton = document.querySelector('path[d*="M15.192 8.906a1.143"]')?.parentNode.parentNode  // post-GPT-4o;
            || document.querySelector('[data-testid="send-button"]');
        }
    }, 25);
    return 0;   //everything is ok
}

async function chatgpt_isIdle() {
    return new Promise(resolve => {
        const intervalId = setInterval(() => {
              if (chatgpt_getRegenerateButton() || do_force_completion) {
                 clearInterval(intervalId); resolve(true);
              }
            }, 100);});}

function chatgpt_getRegenerateButton() {
    for (const mainSVG of document.querySelectorAll('main svg.icon')) {
        if (mainSVG.querySelector('path[d^="M3.502 16.6663V13"]') || mainSVG.querySelector('path[d^="M3.06957"]') || mainSVG.querySelector('path[d^="M10.9153 1.83987L11.2942 1.88772L11.4749"]') || document.querySelector("button[data-testid=good-response-turn-action-button]")){ // regen icon or thumb-up icon found
            //console.log(">>>>>>>>>> found regen icon!");
            return mainSVG.parentNode.parentNode;
        }
    }
    for (const mainSVG of document.querySelectorAll('main svg.icon')) {
        if (mainSVG.querySelector('path[d^="M9.75122 4.09203C9.75122"]') || mainSVG.querySelector('path[d^="M11 4.9099C11 4.47485"]')){ // read aloud icon found
            //console.log(">>>>>>>>>> found read aloud icon!");
            return mainSVG.parentNode.parentNode;
        }
    }
}

function chatpgt_scrollToBottom () {
    try { 
        //document.querySelector('button[class*="cursor"][class*="bottom"]').click();
        document.querySelector('path[d^="M9.33468 3.33333C9.33468"]')?.parentNode?.parentNode?.click();
    }
    catch (err) { console.error('[ThunderAI] ', err); }
}

function addCustomDiv(prompt_action,tabId,mailMessageId) {
    // Create <style> element for the CSS
    var style = document.createElement('style');
    style.textContent = ".mzta-header-fixed {position:fixed;bottom:0;left: 0;height:100px;width:100%;background-color: #333;color: white;text-align: center;padding: 10px 0;z-index: 1000;border-top: 3px solid white;}"
    style.textContent += "body {padding-bottom: 100px !important;} [id^='headlessui-dialog-panel-:r']{padding-bottom: 100px !important;} [data-testid='screen-thread']{padding-bottom: 100px !important;} [slot='content']{padding-bottom: 100px !important;}";
    style.textContent += ".mzta-btn {background-color: #007bff;border: none;color: white;padding: 8px 15px;text-align: center;text-decoration: none;display: inline-block;font-size: 16px;margin: 4px 2px;transition-duration: 0.4s;cursor: pointer;border-radius: 5px;}";
    style.textContent += ".mzta-btn:hover {background-color:#0056b3;color:white;}";
    style.textContent += ".btn_disabled {background-color: #6a829b !important;color: white !important;cursor: not-allowed;}";
    style.textContent += ".btn_disabled:hover {background-color:#6a829b !important;color:white !important;}";
    style.textContent += "#mzta-loading{height:50px;display:inline-block;}";
    style.textContent += "#mzta-model_warn{position:absolute;top:50%;left:50%;transform:translate(-50%,-50%);max-height:100%px;min-width:30%;max-width:50%;padding:3px;border-radius:5px;text-align:center;background-color:#FFBABA;border:1px solid;font-size:13px;color:#D8000C;display:none;}#mzta-model_warn a{color:blue;text-decoration: underline;}";
    style.textContent += "#mzta-btn_model {background-color: #007bff;border: none;color: white;padding: 2px 4px;text-align: center;text-decoration: none;display: none;font-size: 13px;margin-left: 4px;transition-duration: 0.4s;cursor: pointer;border-radius: 2px;}";
    style.textContent += "#mzta-status-page{position:fixed;bottom:0;left:0;padding-left:5px;font-size:13px;font-style:italic;text-decoration:underline;color:#919191;}";
    style.textContent += "#mzta-force-completion{cursor:pointer;position:fixed;bottom:0;right:0;padding-right:5px;font-size:13px;font-style:italic;text-decoration:underline;color:#919191;}";
    style.textContent += "#mzta-status-page:hover, #mzta-force-completion:hover{color:#007bff;}";
    style.textContent += "#mzta-custom_text{padding:10px;width:auto;max-width:80%;height:auto;max-height:80%;border-radius:5px;overflow:auto;position:fixed;top:50%;left:50%;display:none;transform:translate(-50%,-50%);text-align:center;background:#333;color:white;border:3px solid white;}";
    style.textContent += "#mzta-custom_loading{height:50px;display:none;}";
    style.textContent += "#mzta-custom_textarea{color:black;padding:1px;font-size:15px;width:100%;}";
    style.textContent += "#mzta-custom_info{text-align:center;width:100%;padding-bottom:10px;font-size:15px;}";
    style.textContent += "#mzta-custom_info span{font-size:0.8em;}";
    style.textContent += "#mzta-custom_step{position: absolute;bottom: 5px;right: 10px;font-size: 12px;color: #ccc;}";
    style.textContent += "#mzta-prompt-name{font-size:13px;font-style:italic;color:#919191;position:fixed;bottom:75px;;left:0;padding-left:5px;}";
    style.textContent += "#mzta-diff-overlay{position: fixed;top:0;left:0;width:100vw;height:100vh;background: rgba(0, 0, 0, 0.5);display:flex;justify-content:center;align-items:center;z-index:999;}";
    style.textContent += "#mzta-diff{padding:10px;border:2px solid white;border-radius:1em;position:fixed;top:50%;left:50%;width:80%;height:30em;transform:translate(-50%,-50%);z-index:9999;background-color: #333;color: white;}";
    style.textContent += "#mzta-diff_title{font-size:16px;font-weight:bold;text-align:center;padding-bottom:10px;}";
    style.textContent += "#mzta-diff_content{overflow-y:scroll;text-align:justify;width:100%;height:22.5em;}";
    style.textContent += "#mzta-diff span.added{background-color: rgb(0, 94, 0);display:inline;} #mzta-diff span.removed{background-color: rgb(90, 0, 0);display:inline;text-decoration:line-through;}";
    style.textContent += "#mzta-btn_close_diff{position:absolute;bottom:10px;left:50%;transform:translateX(-50%);}";
    style.textContent += "#mzta-btn_change_reply_type{padding-left:4px;padding-right:10px;margin-left:0px;border-bottom-left-radius:0px;border-top-left-radius:0px;height:2.7em;}";
    style.textContent += ".mzta-btn_reply{padding-right:4px;margin-right:0px;border-bottom-right-radius:0px;border-top-right-radius:0px;}";
    style.textContent += "#mzta-forcecomp-hint{display:none;position:fixed;bottom:30px;right:90px;width:20em;background: #c3cefeff;color: #111827;padding:10px 14px;border-radius:12px;box-shadow:0 4px 10px rgba(0,0,0,.2);font-size:14px;font-weight:bold;z-index:2000;pointer:default;}";
    style.textContent += "#mzta-forcecomp-hint .label{display:block;line-height:1.3;}";
    style.textContent += "#mzta-forcecomp-hint svg{position:absolute;bottom:-10px;right:-20px;width:40px;height:40px;}";
    style.textContent += "#thread-bottom-container{padding-bottom: 100px !important;box-sizing: border-box;}";



    // Add <style> to the page's <head>
    document.head.appendChild(style);

    // Fixed div
    var fixedDiv = document.createElement('div');
    fixedDiv.classList.add('mzta-header-fixed');
    fixedDiv.textContent = '';

    // Model warning div
    var modelWarnDiv = document.createElement('div');
    modelWarnDiv.id = 'mzta-model_warn';
    modelWarnDiv.textContent = browser.i18n.getMessage("chatgpt_win_model_warning");
    fixedDiv.appendChild(modelWarnDiv);

    // GPT Model Button
    var btn_model = document.createElement('button');
    btn_model.id="mzta-btn_model";
    btn_model.textContent = browser.i18n.getMessage("chatgpt_btn_model");
    btn_model.onclick = async function() {
        force_go = true;
        document.getElementById('mzta-model_warn').style.display = 'none';
    };
    modelWarnDiv.appendChild(btn_model);
    
    //prompt name
    var prompt_name_div = document.createElement('div');
    prompt_name_div.id = 'mzta-prompt-name';
    prompt_name_div.title= browser.i18n.getMessage("currently_used_prompt");
    prompt_name_div.textContent = mztaPromptName;
    fixedDiv.appendChild(prompt_name_div);

    //status page
    var status_page_div = document.createElement('div');
    status_page_div.id = 'mzta-status-page';
    status_page_div.innerHTML = '<a href="https://micz.it/thunderbird-addon-thunderai/status/">'+ mztaStatusPageDesc +'</a>';
    fixedDiv.appendChild(status_page_div);

    //force completion
    var force_completion_div = document.createElement('div');
    force_completion_div.id = 'mzta-force-completion';
    force_completion_div.textContent = mztaForceCompletionDesc;
    force_completion_div.title = mztaForceCompletionTitle;
    force_completion_div.addEventListener('click', function() {
        do_force_completion = true;
    });
    fixedDiv.appendChild(force_completion_div);

    // span for the text
    var curr_msg = document.createElement('span');
    curr_msg.id='mzta-curr_msg';
    curr_msg.textContent = browser.i18n.getMessage("chatgpt_win_working");
    curr_msg.style.display = 'block';
    fixedDiv.appendChild(curr_msg);

    // loading gif
    var loading = document.createElement('img');
    loading.src = browser.runtime.getURL("/images/loading.gif");
    loading.id = "mzta-loading";
    fixedDiv.appendChild(loading);

    // OK button
    var btn_ok = document.createElement('button');
    btn_ok.id="mzta-btn_ok";
    btn_ok.classList.add('mzta-btn');
    // Prevent the button from stealing focus and clearing the selection
    btn_ok.addEventListener('mousedown', function(e) {
        e.preventDefault();
    });
    //console.log('>>>>>>>>>>>>>>> default: '+prompt_action)
    switch(String(prompt_action)){ 
        default:
        case "0":     // close window
            btn_ok.textContent = browser.i18n.getMessage("chatgpt_win_close");
            btn_ok.onclick = async function() {
                browser.runtime.sendMessage({command: "chatgpt_close", window_id: mztaWinId});
            };
            fixedDiv.appendChild(btn_ok);
            break;
        case "1":     // do reply
            disableButton(btn_ok);
            const btn_ok_line1 = document.createElement('span');
            btn_ok_line1.textContent = browser.i18n.getMessage("chatgpt_win_get_answer");
            const btn_ok_line2 = document.createElement('span');
            btn_ok_line2.classList.add('action_btn_info');
            btn_ok_line2.textContent = mztaReplyType == 'reply_all' ? browser.i18n.getMessage("prefs_OptionText_reply_all") : browser.i18n.getMessage("prefs_OptionText_reply_sender");
            btn_ok_line1.style.display = 'block';
            btn_ok_line1.style.margin = '0';
            btn_ok_line1.style.padding = '0';
            btn_ok_line1.style.lineHeight = '1';
            btn_ok_line2.style.display = 'block';
            btn_ok_line2.style.margin = '0';
            btn_ok_line2.style.padding = '0';
            btn_ok_line2.style.lineHeight = '1';
            btn_ok_line2.style.fontSize = '0.7em';
            btn_ok.appendChild(btn_ok_line1);
            //btn_ok.appendChild(document.createElement('br'));
            btn_ok.appendChild(btn_ok_line2);
            btn_ok.setAttribute('data-reply-type', mztaReplyType);
            btn_ok.addEventListener('click', async function() {
                const response = getSelectedHtml();
                const currentReplyType = btn_ok.getAttribute('data-reply-type');
                // console.log(">>>>>>>>>> btn_ok reply type: " + JSON.stringify(currentReplyType));
                await browser.runtime.sendMessage({command: "chatgpt_replyMessage", text: response, tabId: tabId, mailMessageId: mailMessageId, replyType: currentReplyType});
                browser.runtime.sendMessage({command: "chatgpt_close", window_id: mztaWinId});
            });
            btn_ok.classList.add('mzta-btn_reply');
            // change reply type button
            var btn_change_reply_type = document.createElement('button');
            disableButton(btn_change_reply_type);
            btn_change_reply_type.id = 'mzta-btn_change_reply_type';
            // Prevent the button from stealing focus and clearing the selection
            btn_change_reply_type.addEventListener('mousedown', function(e) {
                e.preventDefault();
            });
            btn_change_reply_type.classList.add('mzta-btn');
            btn_change_reply_type.title = browser.i18n.getMessage("chatgpt_win_change_reply_type");
            // Create SVG element
            let currentIcon = createReplyToAllIcon();
            // Append SVG to button
            btn_change_reply_type.appendChild(currentIcon);
            btn_change_reply_type.addEventListener('click', function() {
                // console.log('>>>>>>>>>> change reply type clicked');
                btn_change_reply_type.removeChild(currentIcon);
                if(mztaReplyType == 'reply_all'){
                    mztaReplyType = 'reply_sender';
                    currentIcon = createReplyToSenderIcon();
                    btn_change_reply_type.appendChild(currentIcon);
                    btn_ok_line2.textContent = browser.i18n.getMessage("prefs_OptionText_reply_sender");
                }else{
                    mztaReplyType = 'reply_all';
                    currentIcon = createReplyToAllIcon();
                    btn_change_reply_type.appendChild(currentIcon);
                    btn_ok_line2.textContent = browser.i18n.getMessage("prefs_OptionText_reply_all");
                }
                btn_ok.setAttribute('data-reply-type', mztaReplyType);
            });
            fixedDiv.appendChild(btn_ok);
            fixedDiv.appendChild(btn_change_reply_type);
            btn_change_reply_type.style.display = 'none';
            break;
        case "2":     // replace text
            disableButton(btn_ok);
            btn_ok.textContent = browser.i18n.getMessage("chatgpt_win_get_answer");
            btn_ok.onclick = async function() {
                const response = getSelectedHtml();
                //console.log('replace text: '+tabId)
                await browser.runtime.sendMessage({command: "chatgpt_replaceSelectedText", text: response, tabId: tabId, mailMessageId: mailMessageId});
                browser.runtime.sendMessage({command: "chatgpt_close", window_id: mztaWinId});
            };
            fixedDiv.appendChild(btn_ok);
            break;
    }
    btn_ok.style.display = 'none';

    if(mztaUseDiffViewer == '1'){
        // diff overlay div
        var diffOverlay = document.createElement('div');
        diffOverlay.id = 'mzta-diff-overlay';
        diffOverlay.style.display = 'none';
        // diff div
        var diffDiv = document.createElement('div');
        diffDiv.id = 'mzta-diff';
        // diff div title
        var diffTitle = document.createElement('div');
        diffTitle.id = 'mzta-diff_title';
        diffTitle.textContent = browser.i18n.getMessage('chatgpt_win_diff_title');
        diffDiv.appendChild(diffTitle);
        // diff div content
        var diffContent = document.createElement('div');
        diffContent.id = 'mzta-diff_content';
        diffDiv.appendChild(diffContent);
        // diff div close button
        var btn_close_diff = document.createElement('button');
        btn_close_diff.id = 'mzta-btn_close_diff';
        btn_close_diff.classList.add('mzta-btn');
        btn_close_diff.textContent = browser.i18n.getMessage('chatgpt_win_close');
        btn_close_diff.onclick = function() {
            document.getElementById('mzta-diff-overlay').style.display = 'none';
        };
        diffDiv.appendChild(btn_close_diff);
        diffOverlay.appendChild(diffDiv);
        fixedDiv.appendChild(diffOverlay);

        // diff viewer button
        var btn_diff = document.createElement('button');
        btn_diff.id='mzta-btn_diff';
        btn_diff.classList.add('mzta-btn');
        // Prevent the button from stealing focus and clearing the selection
        btn_diff.addEventListener('mousedown', function(e) {
            e.preventDefault();
        });
        btn_diff.textContent = browser.i18n.getMessage('btn_show_differences');
        btn_diff.style.display = 'none';
        disableButton(btn_diff);
        btn_diff.onclick = async function() {
            diffContent.innerHTML = '';
            const response = getSelectedHtml();
            const wordDiff = Diff.diffWords(mztaOriginalText, response.replace(/<\\/?[^>]+(>|$)/g, ''));
            wordDiff.forEach(part => {
                // Split part.value by <br> (handling <br>, <br/>, <br />)
                const brRegex = /(<br\s*\\/?>)/gi;
                const segments = part.value.split(brRegex);

                segments.forEach(segment => {
                if (segment.match(brRegex)) {
                    // It's a <br>, add a real <br> element
                    diffContent.appendChild(document.createElement("br"));
                } else if (segment.length > 0) {
                    const diffElement = document.createElement("span");
                    if (part.added) {
                    diffElement.className = "added";
                    diffElement.textContent = segment;
                    } else if (part.removed) {
                    diffElement.className = "removed";
                    diffElement.textContent = segment;
                    } else {
                    diffElement.textContent = segment;
                    }
                    diffContent.appendChild(diffElement);
                }
                });
            });
            diffOverlay.style.display = 'block';
        };
        fixedDiv.appendChild(btn_diff);
    }

    //div per custom text
    let customDiv = document.createElement('div');
    customDiv.id = 'mzta-custom_text';
    let customInfo = document.createElement('div');
    customInfo.id = 'mzta-custom_info';
    customInfo.textContent = browser.i18n.getMessage("chatgpt_win_custom_text");
    customDiv.appendChild(customInfo);
    let customTextArea = document.createElement('textarea');
    customTextArea.id = 'mzta-custom_textarea';
    customTextArea.rows = 5;
    customDiv.appendChild(customTextArea);
    let customLoading = document.createElement('img');
    customLoading.src = browser.runtime.getURL("/images/loading.gif");
    customLoading.id = "mzta-custom_loading";
    customDiv.appendChild(customLoading);
    let customBtn = document.createElement('button');
    customBtn.id = 'mzta-custom_btn';
    customBtn.textContent = browser.i18n.getMessage("chatgpt_win_send");
    customBtn.classList.add('mzta-btn');
    customBtn.addEventListener("click", () => { customTextBtnClick({customBtn:customBtn,customLoading:customLoading,customDiv:customDiv}) });
    customTextArea.addEventListener("keydown", (event) => { if(event.code == "Enter" && event.ctrlKey) customTextBtnClick({customBtn:customBtn,customLoading:customLoading,customDiv:customDiv}) });
    customDiv.appendChild(customBtn);
    let customStep = document.createElement('div');
    customStep.id = 'mzta-custom_step';
    customDiv.appendChild(customStep);
    fixedDiv.appendChild(customDiv);

    // light background hint with diagonal thick arrow
    let forcecompletionHint_div = document.createElement('div');
    forcecompletionHint_div.id = 'mzta-forcecomp-hint';
    forcecompletionHint_div.addEventListener("click", () => {
       do_force_completion = true;
    });

    let arrowLabel = document.createElement('span');
    arrowLabel.className = 'label';
    forcecompletionHint_div.style.cursor = 'default';
    arrowLabel.textContent = browser.i18n.getMessage("chatgpt_click_force_completion");
    forcecompletionHint_div.appendChild(arrowLabel);
    arrowLabel.addEventListener("click", () => {
       do_force_completion = true;
    });

    // SVG solid arrow (triangle)
    const svgNS = 'http://www.w3.org/2000/svg';
    let hintArrow = document.createElementNS(svgNS, 'svg');
    hintArrow.setAttribute('viewBox', '0 0 40 40');
    hintArrow.addEventListener("click", () => {
       do_force_completion = true;
    });

    let arrowPolygon = document.createElementNS(svgNS, 'polygon');
    arrowPolygon.setAttribute('points', '0,0 40,40 0,28');
    arrowPolygon.setAttribute('fill', '#c3cefeff');

    hintArrow.appendChild(arrowPolygon);
    forcecompletionHint_div.appendChild(hintArrow);

    fixedDiv.appendChild(forcecompletionHint_div);

    document.body.insertBefore(fixedDiv, document.body.firstChild);
}

// Create SVG icons as functions
function createReplyToSenderIcon() {
  const svgNS = 'http://www.w3.org/2000/svg';
  const svg = document.createElementNS(svgNS, 'svg');
  svg.setAttribute('xmlns', svgNS);
  svg.setAttribute('viewBox', '0 0 24 24');
  svg.setAttribute('width', '24');
  svg.setAttribute('height', '24');
  svg.setAttribute('fill', 'currentColor');
  const path1 = document.createElementNS(svgNS, 'path');
  path1.setAttribute('d', 'M0 0h24v24H0z');
  path1.setAttribute('fill', 'none');
  const path2 = document.createElementNS(svgNS, 'path');
  path2.setAttribute('d', 'M10 9V5l-7 7 7 7v-4.1c5 0 8.5 1.6 11 5.1-1-5-4-10-11-11z');
  svg.appendChild(path1);
  svg.appendChild(path2);
  return svg;
}


function createReplyToAllIcon() {
  const svgNS = 'http://www.w3.org/2000/svg';
  const svg = document.createElementNS(svgNS, 'svg');
  svg.setAttribute('xmlns', svgNS);
  svg.setAttribute('viewBox', '0 0 24 24');
  svg.setAttribute('width', '24');
  svg.setAttribute('height', '24');
  svg.setAttribute('fill', 'currentColor');
  const path1 = document.createElementNS(svgNS, 'path');
  path1.setAttribute('d', 'M0 0h24v24H0z');
  path1.setAttribute('fill', 'none');
  const path2 = document.createElementNS(svgNS, 'path');
  path2.setAttribute('d', 'M7 8V5l-7 7 7 7v-3l-4-4 4-4zm6 1V5l-7 7 7 7v-4.1c5 0 8.5 1.6 11 5.1-1-5-4-10-11-11z');
  svg.appendChild(path1);
  svg.appendChild(path2);
  return svg;
}

function renderCustomTextStep() {
    const currentItem = _customTextArray[_currentCustomTextIndex];
    const infoDiv = document.getElementById('mzta-custom_info');
    const customTextArea = document.getElementById('mzta-custom_textarea');
    const customStep = document.getElementById('mzta-custom_step');
    
    customTextArea.value = "";
    infoDiv.textContent = browser.i18n.getMessage("chatgpt_win_custom_text");
    
    if (currentItem.info && currentItem.info.trim() !== "") {
        infoDiv.appendChild(document.createElement("br"));
        const infoSpan = document.createElement("span");
        infoSpan.textContent = "[" + browser.i18n.getMessage("customPrompts_form_label_ID") + ": " + currentItem.info + "]";
        infoDiv.appendChild(infoSpan);
    }

    if(_customTextArray.length > 1) {
        customStep.textContent = (_currentCustomTextIndex + 1) + "/" + _customTextArray.length;
        customStep.style.display = 'block';
    } else {
        customStep.style.display = 'none';
    }
    
    customTextArea.focus();
}

function customTextBtnClick(args) {
    const customText = document.getElementById('mzta-custom_textarea').value;
    
    if (_customTextArray[_currentCustomTextIndex]) {
        _customTextArray[_currentCustomTextIndex].custom_text = customText;
    }

    _currentCustomTextIndex++;

    if (_currentCustomTextIndex < _customTextArray.length) {
        renderCustomTextStep();
    } else {
        args.customBtn.disabled = true;
        args.customBtn.classList.add('disabled');
        args.customLoading.style.display = 'inline-block';
        args.customLoading.style.display = 'none';
        doProceed(current_message, _customTextArray);
        args.customDiv.style.display = 'none';
        
        args.customBtn.disabled = false;
        args.customBtn.classList.remove('disabled');
    }
}

function checkGPTModel(model) {
    if(model == '') return Promise.resolve();
    doLog("checkGPTModel model: " + model);
  return new Promise((resolve, reject) => {
    // Set up an interval that shows the warning after 2 seconds
    const intervalId2 = setTimeout(() => {
        let modelWarn = document.getElementById('mzta-model_warn');
        let btnModel = document.getElementById('mzta-btn_model');
        if (modelWarn) modelWarn.style.display = 'inline-block';
        if (btnModel) btnModel.style.display = 'inline';
    }, 2000);
    // Set up an interval that checks the value every 100 milliseconds
    const intervalId = setInterval(() => {
      // Get the '.text-token-text-secondary' element
     // const element = document.querySelector('div#radix-\\\\:ri2\\\\: > div > span.text-token-text-secondary');
     const elements = document.querySelectorAll('[id*=radix] span')

     // If there are no elements, we are using a free account, so go on immediately
     if (elements.length === 0) {
        doLog("checkGPTModel no model found in DOM, we are using free account.");
        clearInterval(intervalId);
        clearTimeout(intervalId2);
        resolve('free');
        return;
      }

     // Loop through the elements to find the one with the specified text content
     for(let element of elements){
      // Check if the element exists and its content is '4' or '4o'
      doLog("checkGPTModel model found in DOM: " + element.textContent);
      if ((element && element.textContent === model)||(force_go)) {
        doLog("The GPT Model is now " + model);
        clearInterval(intervalId);
        clearTimeout(intervalId2);
        resolve(model);
        break;
      } else if (!element) {
        console.error("[ThunderAI | ChatGPT Web] Model string element not found! [" + model + "]");
        clearInterval(intervalId);
        reject("Model string element not found: " + model);
      }
     }
    }, 200);
  });
}

function operation_done(){
    let curr_msg = document.getElementById('mzta-curr_msg');
    curr_msg.textContent = browser.i18n.getMessage("chatgpt_win_job_completed");
    if(current_action != '0'){
        curr_msg.textContent += " " + browser.i18n.getMessage("chatgpt_win_job_completed_select"); 
    }
    curr_msg.style.display = 'block';
    document.getElementById('mzta-btn_ok').style.display = 'inline';
    if(current_action == '1'){
        document.getElementById('mzta-btn_change_reply_type').style.display = 'inline';
    }
    if(mztaUseDiffViewer == '1'){
        document.getElementById('mzta-btn_diff').style.display = 'inline';
    }
    document.getElementById('mzta-loading').style.display = 'none';
    document.getElementById('mzta-force-completion').style.display = 'none';
    document.getElementById('mzta-forcecomp-hint').style.display = 'none';
    chatpgt_scrollToBottom();
}

function checkLoggedIn(){
    return !window.location.href.startsWith('https://chatgpt.com/auth/') && document.querySelector('button[data-testid*=login]') === null;
}

function showCustomTextField(){
    let rawArray = current_message.prompt_info?.custom_text_array;
    _customTextArray = Array.isArray(rawArray) ? rawArray : [];
    if (_customTextArray.length === 0) {
            _customTextArray.push({ placeholder: "{%additional_text%}", info: "" });
    }
    _currentCustomTextIndex = 0;
    document.getElementById('mzta-custom_text').style.display = 'block';
    renderCustomTextStep();
}

async function doProceed(message, customText = ''){
    let _gpt_model = mztaGPTModel;
    doLog("doProceed _gpt_model: " + JSON.stringify(_gpt_model));
    if(_gpt_model != ''){
        await checkGPTModel(_gpt_model);
    }
    let final_prompt = message.prompt;

    if (Array.isArray(customText)) {
        let anyReplaced = false;
        customText.forEach(obj => {
            if (final_prompt.includes(obj.placeholder)) {
                anyReplaced = true;
                let escapedPH = obj.placeholder.replace(/[.*+?^{$}()|[\\]\\\\]/g, '\\\\$&');
                let regex = new RegExp(escapedPH, 'g');
                final_prompt = final_prompt.replace(regex, obj.custom_text);
            }
        });
        if (!anyReplaced) {
            const inputText = customText.map(obj => obj.custom_text).join(' ');
            if (inputText !== '') {
                final_prompt += ' ' + inputText;
            }
        }
    } else {
        //check if there is the additional_text placeholder
        if(final_prompt.includes('{%additional_text%}')){
            final_prompt = final_prompt.replace('{%additional_text%}', customText || (mztaPhDefVal == '1'?'':'{%additional_text%}'));
        }else{
            if(customText != ''){
                final_prompt += ' '+customText;
            }
        }
    }

    let send_result = await chatgpt_sendMsg(final_prompt,'click');
    //console.log(">>>>>>>>>>> send_result: " + send_result);
    switch(send_result){
        case -1:        // send button not found
            let curr_msg = document.getElementById('mzta-curr_msg');
            curr_msg.style.display = 'block';
            curr_msg.textContent = browser.i18n.getMessage("chatgpt_sendbutton_not_found_error");
            break;
        case -2:    // textarea not found
            let curr_model_warn = document.getElementById('mzta-model_warn');
            curr_model_warn.textContent = browser.i18n.getMessage("chatgpt_textarea_not_found_error");
            curr_model_warn.style.display = 'inline-block';
            document.getElementById('mzta-curr_msg').textContent = "";
            document.getElementById('mzta-loading').style.display = 'none';
            let btn_retry = document.createElement('button');
            btn_retry.id="mzta-btn_retry";
            btn_retry.classList.add('mzta-btn');
            btn_retry.style.position = 'absolute';
            btn_retry.style.top = '5px';
            btn_retry.style.right = '5px';
            btn_retry.textContent = browser.i18n.getMessage("chatgpt_btn_retry");
            btn_retry.addEventListener('click', function() {
                doRetry();
            });
            curr_model_warn.insertAdjacentElement('afterend', btn_retry);
            break;
    }
    let forcecompletionHintTimeout;
    if(send_result == 0){
            forcecompletionHintTimeout = setTimeout(() => {
            document.getElementById('mzta-forcecomp-hint').style.display = 'block';
        }, delay_wait_completion);
    }
    await chatgpt_isIdle();
    if(send_result == 0){
        clearTimeout(forcecompletionHintTimeout);
    }
    operation_done();
}

function doRetry(){
    document.getElementById('mzta-model_warn').style.display = 'none';
    document.getElementById('mzta-btn_retry').remove();
    document.getElementById('mzta-loading').style.display = 'inline-block';
    document.getElementById('mzta-curr_msg').textContent = browser.i18n.getMessage("chatgpt_win_working");
    doProceed(current_message);
}

async function showForceCompletionHint(){
    const forcecompletionHintTimeout = setTimeout(() => {
        document.getElementById('mzta-forcecomp-hint').style.display = 'block';
    }, delay_wait_completion);
    await chatgpt_isIdle();
    clearTimeout(forcecompletionHintTimeout);
}

function removeTagsAndReturnHTML(rootElement, removeTags, preserveTags) {
    const fragment = document.createDocumentFragment();

    function handleElement(element) {
        let child = element.firstChild;
        while (child) {
            const nextSibling = child.nextSibling;
            if (preserveTags.includes(child.nodeName.toLowerCase())) {
                //console.log(">>>>>>>>>>>> preserve child: " + child.tagName.toLowerCase());
                fragment.appendChild(child);
            } else if (child.nodeType === Node.ELEMENT_NODE) {
                //console.log(">>>>>>>>>>>> handleElement(child): " + child.tagName.toLowerCase());
                handleElement(child);
            }
            child = nextSibling;
        }
    }

    removeTags.forEach(tag => {
        const elements = Array.from(rootElement.getElementsByTagName(tag));
        elements.forEach(element => {
            handleElement(element);
            element.parentNode.insertBefore(fragment.cloneNode(true), element);
            element.parentNode.removeChild(element);
            //console.log(">>>>>>>>>>>> removeChild: " + element.tagName.toLowerCase());
        });
    });

    replaceNewlinesWithBr(rootElement);
    //console.log(">>>>>>>>>>>> rootElement.innerHTML: " + rootElement.innerHTML);
    // Return the updated HTML as a string
    return rootElement.innerHTML;
}

// Replace newline characters with <br> tags
function replaceNewlinesWithBr(node) {
    for (let child of Array.from(node.childNodes)) {
        if (child.nodeType === Node.TEXT_NODE) {
            const parts = child.textContent.split('\\n');
            if (parts.length > 1) {
                const fragment = document.createDocumentFragment();
                parts.forEach((part, index) => {
                    fragment.appendChild(document.createTextNode(part));
                    if (index < parts.length - 1) {
                        fragment.appendChild(document.createElement('br'));
                    }
                });
                child.parentNode.replaceChild(fragment, child);
            }
        } else if (child.nodeType === Node.ELEMENT_NODE) {
            replaceNewlinesWithBr(child);
        }
    }
}

function getSelectedHtml() {
    // Get the Selection object
    var selection = window.getSelection();
    // console.log(">>>>>>>>>>>>> getSelectedHtml selection.rangeCount: " + selection.rangeCount);
    if (selection.rangeCount > 0) {
        // Get the first selected range
        var range = selection.getRangeAt(0);
        
        // Create a new temporary div
        var tempDiv = document.createElement("div");
        
        // Clone the contents of the range into the temporary div
        tempDiv.appendChild(range.cloneContents());
        // console.log(">>>>>>>>>>>>>>>>>> tempDiv.innerHTML: " + tempDiv.innerHTML);
        // Return the HTML of the selected content
        return tempDiv.innerHTML.replace(/^<p>&quot;/, '<p>').replace(/&quot;<\\/p>$/, '</p>'); // strip quotation marks;
    }
    if (lastSelectedHtml) {
        // console.log(">>>>>>>>>>>>> getSelectedHtml using cached selection lastSelectedHtml: "+ lastSelectedHtml);
        return lastSelectedHtml;
    }
    return "";
}

function isSomethingSelected() {
    // Get the Selection object
    var selection = window.getSelection();
    
    // Check if the selection range count is greater than 0 and the selection is not empty
    return selection.rangeCount > 0 && !selection.isCollapsed;
}

document.addEventListener("selectionchange", function() {
     // Clear any previous timeout to reset the delay
     clearTimeout(selectionChangeTimeout);
     if(current_action === '0'){
         return;
     }
     // Set a timeout to delay the execution of the callback
     selectionChangeTimeout = setTimeout(function() {
        let btn_ok = document.getElementById('mzta-btn_ok');
        let btn_diff = document.getElementById('mzta-btn_diff');
        if (isSomethingSelected()) {
            // Cache the selection
            var selection = window.getSelection();
            if (selection.rangeCount > 0) {
                var range = selection.getRangeAt(0);
                var tempDiv = document.createElement("div");
                tempDiv.appendChild(range.cloneContents());
                lastSelectedHtml = tempDiv.innerHTML.replace(/^<p>&quot;/, '<p>').replace(/&quot;<\\/p>$/, '</p>');
            }
            enableButton(btn_ok);
            if(current_action == '1'){
                let btn_reply_type = document.getElementById('mzta-btn_change_reply_type');
                enableButton(btn_reply_type);
            }
            if(mztaUseDiffViewer == '1'){
                enableButton(btn_diff);
            }
        } else {
            disableButton(btn_ok);
            if(current_action == '1'){
                let btn_reply_type = document.getElementById('mzta-btn_change_reply_type');
                disableButton(btn_reply_type);
            }
            if(mztaUseDiffViewer == '1'){
                disableButton(btn_diff);
            }
        }
     }, 300); // Delay in milliseconds
});

function enableButton(btn){
    btn.disabled = false;
    btn.classList.remove('btn_disabled');
}

function disableButton(btn){
    btn.disabled = true;
    btn.classList.add('btn_disabled');
}

function selectContentOnMouseDown(event) {
    // Reset the dragging flag when the mouse is pressed down
    isDragging = false;
}

function selectContentOnMouseMove(event) {
    // Set the dragging flag to true if the mouse moves
    isDragging = true;
}

function selectContentOnMouseUp(event) {
    var excludedArea = document.querySelector('.mzta-header-fixed');

    if (excludedArea && excludedArea.contains(event.target)) {
        // If the click was inside the excluded area, do nothing
        return;
    }
    // console.log(">>>>>>>>>>>>> selectContentOnMouseUp isDragging: " + isDragging);
    // Reset the dragging flag when the mouse is released")
    if ((!isDragging)&&(!isSomethingSelected())) {
        // If no dragging has occurred, execute the selection code
        selectContentOnClick(event);
    }
    // Remove the event listeners to prevent future executions
    // document.removeEventListener('mousedown', selectContentOnMouseDown);
    // document.removeEventListener('mousemove', selectContentOnMouseMove);
    // document.removeEventListener('mouseup', selectContentOnMouseUp);
}

function selectContentOnClick(event) {
    if(current_action === '0'){
        return;
    }

    // Prevent the default behavior of the click
    event.preventDefault();

    // Get the element that was clicked
    var clickedElement = event.target;

    // Traverse the DOM upwards to find the nearest parent div
    var parentDiv = clickedElement.closest('div');

    // console.log(">>>>>>>>>>>>> parentDiv: " + parentDiv.classList);

    if (parentDiv) {
        // Create a range object
        var range = document.createRange();

        // Select the contents of the div
        range.selectNodeContents(parentDiv);

        // Get the selection object
        var selection = window.getSelection();

        // Clear any existing selections
        selection.removeAllRanges();

        // Add the new range to the selection
        selection.addRange(range);
        // console.log(">>>>>>>>>>>>> selectContentOnClick selection.rangeCount: " + selection.rangeCount);
    }
}

function doLog(msg){
    if(mztaDoDebug == 1){
        console.log("[ThunderAI | ChatGPT Web] " + msg);
    }
}

function run(checkTab = null) {
    if(!checkLoggedIn()){
        // User not logged in
        if(checkTab){
            clearInterval(checkTab);
        }
        doLog("User not logged in, showing warning message.");
        alert(browser.i18n.getMessage("chatgpt_user_not_logged_in"));
    }else{
        addCustomDiv(current_action,current_tabId,current_mailMessageId);
        (async () => {
            if(mztaDoCustomText === "1"){
                showCustomTextField();
            } else {
                await doProceed(current_message);
            }
            // Add an event listener to the document to detect clicks
            document.addEventListener('mousedown', selectContentOnMouseDown);
            document.addEventListener('mousemove', selectContentOnMouseMove);
            document.addEventListener('mouseup', selectContentOnMouseUp);
        })();
    }
}

// In the content script
browser.runtime.onMessage.addListener((message, sender, sendResponse) => {
    //console.log(">>>>>>>>>>>>> content.js onMessage: " + JSON.stringify(message));
    switch(message.command) {
        case "chatgpt_send":
            current_action = message.action;
            current_message = message;
            current_tabId = message.tabId;
            current_mailMessageId = message.mailMessageId;
            if((current_mailMessageId == -1) && (current_action == '1')) {    // we are using the reply from the compose window!
                current_action = '2'; // replace text
            }
            run(checkTab);
            break;
        case "chatgpt_alive":
            sendResponse({isAlive: true});
            break;
    }
});

let checkTab = setInterval(() => {
    let customDiv = document.getElementById('mzta-custom_text');
    if(customDiv){
        clearInterval(checkTab);
    }else{
        run(checkTab);
    }
}, 1000);

`