import { saveSettingsDebounced, callPopup, getRequestHeaders, substituteParams, eventSource, event_types } from "../../../script.js"; import { getContext, extension_settings } from "../../extensions.js"; import { initScrollHeight, resetScrollHeight, getSortableDelay } from "../../utils.js"; import { executeSlashCommands, registerSlashCommand } from "../../slash-commands.js"; import { ContextMenu } from "./src/ContextMenu.js"; import { MenuItem } from "./src/MenuItem.js"; import { MenuHeader } from "./src/MenuHeader.js"; export { MODULE_NAME }; const MODULE_NAME = 'quick-reply'; const UPDATE_INTERVAL = 1000; let presets = []; let selected_preset = ''; const defaultSettings = { quickReplyEnabled: false, numberOfSlots: 5, quickReplySlots: [], placeBeforeInputEnabled: false, quickActionEnabled: false, AutoInputInject: true, } //method from worldinfo async function updateQuickReplyPresetList() { const result = await fetch("/getsettings", { method: "POST", headers: getRequestHeaders(), body: JSON.stringify({}), }); if (result.ok) { var data = await result.json(); presets = data.quickReplyPresets?.length ? data.quickReplyPresets : []; console.debug('Quick Reply presets', presets); $("#quickReplyPresets").find('option[value!=""]').remove(); if (presets !== undefined) { presets.forEach((item) => { const option = document.createElement('option'); option.value = item.name; option.innerText = item.name; option.selected = selected_preset.includes(item.name); $("#quickReplyPresets").append(option); }); } } } async function loadSettings(type) { if (type === 'init') { await updateQuickReplyPresetList() } if (Object.keys(extension_settings.quickReply).length === 0) { Object.assign(extension_settings.quickReply, defaultSettings); } if (extension_settings.quickReply.AutoInputInject === undefined) { extension_settings.quickReply.AutoInputInject = true; } // If the user has an old version of the extension, update it if (!Array.isArray(extension_settings.quickReply.quickReplySlots)) { extension_settings.quickReply.quickReplySlots = []; extension_settings.quickReply.numberOfSlots = defaultSettings.numberOfSlots; for (let i = 1; i <= extension_settings.quickReply.numberOfSlots; i++) { extension_settings.quickReply.quickReplySlots.push({ mes: extension_settings.quickReply[`quickReply${i}Mes`], label: extension_settings.quickReply[`quickReply${i}Label`], enabled: true, }); delete extension_settings.quickReply[`quickReply${i}Mes`]; delete extension_settings.quickReply[`quickReply${i}Label`]; } } initializeEmptySlots(extension_settings.quickReply.numberOfSlots); generateQuickReplyElements(); for (let i = 1; i <= extension_settings.quickReply.numberOfSlots; i++) { $(`#quickReply${i}Mes`).val(extension_settings.quickReply.quickReplySlots[i - 1]?.mes).trigger('input'); $(`#quickReply${i}Label`).val(extension_settings.quickReply.quickReplySlots[i - 1]?.label).trigger('input'); } $('#quickReplyEnabled').prop('checked', extension_settings.quickReply.quickReplyEnabled); $('#quickReplyNumberOfSlots').val(extension_settings.quickReply.numberOfSlots); $('#placeBeforeInputEnabled').prop('checked', extension_settings.quickReply.placeBeforeInputEnabled); $('#quickActionEnabled').prop('checked', extension_settings.quickReply.quickActionEnabled); $('#AutoInputInject').prop('checked', extension_settings.quickReply.AutoInputInject); } function onQuickReplyInput(id) { extension_settings.quickReply.quickReplySlots[id - 1].mes = $(`#quickReply${id}Mes`).val(); $(`#quickReply${id}`).attr('title', String($(`#quickReply${id}Mes`).val())); resetScrollHeight($(`#quickReply${id}Mes`)); saveSettingsDebounced(); } function onQuickReplyLabelInput(id) { extension_settings.quickReply.quickReplySlots[id - 1].label = $(`#quickReply${id}Label`).val(); addQuickReplyBar(); saveSettingsDebounced(); } async function onQuickReplyContextMenuChange(id) { extension_settings.quickReply.quickReplySlots[id - 1].contextMenu = JSON.parse($(`#quickReplyContainer > [data-order="${id}"]`).attr('data-contextMenu')) saveSettingsDebounced(); } async function onQuickReplyCtxButtonClick(id) { const editorHtml = $(await $.get('scripts/extensions/quick-reply/contextMenuEditor.html')); const popupResult = callPopup(editorHtml, "confirm", undefined, { okButton: "Save", wide: false, large: false, rows: 1 }); const qr = extension_settings.quickReply.quickReplySlots[id - 1]; if (!qr.contextMenu) { qr.contextMenu = []; } /**@type {HTMLTemplateElement}*/ const tpl = document.querySelector('#quickReply_contextMenuEditor_itemTemplate'); const fillPresetSelect = (select, item) => { [{ name: 'Select a preset', value: '' }, ...presets].forEach(preset => { const opt = document.createElement('option'); { opt.value = preset.value ?? preset.name; opt.textContent = preset.name; opt.selected = preset.name == item.preset; select.append(opt); } }); }; const addCtxItem = (item, idx) => { const dom = tpl.content.cloneNode(true); const ctxItem = dom.querySelector('.quickReplyContextMenuEditor_item'); ctxItem.setAttribute('data-order', idx); const select = ctxItem.querySelector('.quickReply_contextMenuEditor_preset'); fillPresetSelect(select, item); dom.querySelector('.quickReply_contextMenuEditor_chaining').checked = item.chain; $('.quickReply_contextMenuEditor_remove', ctxItem).on('click', () => ctxItem.remove()); document.querySelector('#quickReply_contextMenuEditor_content').append(ctxItem); } [...qr.contextMenu, {}].forEach((item, idx) => { addCtxItem(item, idx) }); $('#quickReply_contextMenuEditor_addPreset').on('click', () => { addCtxItem({}, document.querySelector('#quickReply_contextMenuEditor_content').children.length); }); $('#quickReply_contextMenuEditor_content').sortable({ delay: getSortableDelay(), stop: () => { }, }); $('#quickReply_autoExecute_userMessage').prop('checked', qr.autoExecute_userMessage ?? false); $('#quickReply_autoExecute_botMessage').prop('checked', qr.autoExecute_botMessage ?? false); $('#quickReply_autoExecute_chatLoad').prop('checked', qr.autoExecute_chatLoad ?? false); $('#quickReply_hidden').prop('checked', qr.hidden ?? false); $('#quickReply_hidden').on('input', () => { const state = !!$('#quickReply_hidden').prop('checked'); qr.hidden = state; saveSettingsDebounced(); }); $('#quickReply_autoExecute_userMessage').on('input', () => { const state = !!$('#quickReply_autoExecute_userMessage').prop('checked'); qr.autoExecute_userMessage = state; saveSettingsDebounced(); }); $('#quickReply_autoExecute_botMessage').on('input', () => { const state = !!$('#quickReply_autoExecute_botMessage').prop('checked'); qr.autoExecute_botMessage = state; saveSettingsDebounced(); }); $('#quickReply_autoExecute_chatLoad').on('input', () => { const state = !!$('#quickReply_autoExecute_chatLoad').prop('checked'); qr.autoExecute_chatLoad = state; saveSettingsDebounced(); }); if (await popupResult) { qr.contextMenu = Array.from(document.querySelectorAll('#quickReply_contextMenuEditor_content > .quickReplyContextMenuEditor_item')) .map(item => ({ preset: item.querySelector('.quickReply_contextMenuEditor_preset').value, chain: item.querySelector('.quickReply_contextMenuEditor_chaining').checked, })) .filter(item => item.preset); $(`#quickReplyContainer[data-order="${id}"]`).attr('data-contextMenu', JSON.stringify(qr.contextMenu)); updateQuickReplyPreset(); onQuickReplyLabelInput(id); } } async function onQuickReplyEnabledInput() { let isEnabled = $(this).prop('checked') extension_settings.quickReply.quickReplyEnabled = !!isEnabled; if (isEnabled === true) { $("#quickReplyBar").show(); } else { $("#quickReplyBar").hide(); } saveSettingsDebounced(); } // New function to handle input on quickActionEnabled async function onQuickActionEnabledInput() { extension_settings.quickReply.quickActionEnabled = !!$(this).prop('checked'); saveSettingsDebounced(); } async function onPlaceBeforeInputEnabledInput() { extension_settings.quickReply.placeBeforeInputEnabled = !!$(this).prop('checked'); saveSettingsDebounced(); } async function onAutoInputInject() { extension_settings.quickReply.AutoInputInject = !!$(this).prop('checked'); saveSettingsDebounced(); } async function sendQuickReply(index) { const prompt = extension_settings.quickReply.quickReplySlots[index]?.mes || ''; return await performQuickReply(prompt, index); } async function executeQuickReplyByName(name) { if (!extension_settings.quickReply.quickReplyEnabled) { throw new Error('Quick Reply is disabled'); } const qr = extension_settings.quickReply.quickReplySlots.find(x => x.label == name); if (!qr) { throw new Error(`Quick Reply "${name}" not found`); } return await performQuickReply(qr.mes); } window['executeQuickReplyByName'] = executeQuickReplyByName; async function performQuickReply(prompt, index) { if (!prompt) { console.warn(`Quick reply slot ${index} is empty! Aborting.`); return; } const existingText = $("#send_textarea").val(); let newText; if (existingText && extension_settings.quickReply.AutoInputInject) { if (extension_settings.quickReply.placeBeforeInputEnabled) { newText = `${prompt} ${existingText} `; } else { newText = `${existingText} ${prompt} `; } } else { // If no existing text and placeBeforeInputEnabled false, add prompt only (with a trailing space) newText = `${prompt} `; } // the prompt starts with '/' - execute slash commands natively if (prompt.startsWith('/')) { const result = await executeSlashCommands(newText); return result?.pipe; } newText = substituteParams(newText); $("#send_textarea").val(newText); // Set the focus back to the textarea $("#send_textarea").trigger('focus'); // Only trigger send button if quickActionEnabled is not checked or if (!extension_settings.quickReply.quickActionEnabled) { $("#send_but").trigger('click'); } } function buildContextMenu(qr, chainMes = null, hierarchy = [], labelHierarchy = []) { const tree = { label: qr.label, mes: (chainMes && qr.mes ? `${chainMes} | ` : '') + qr.mes, children: [], }; qr.contextMenu?.forEach(ctxItem => { let chain = ctxItem.chain; let subName = ctxItem.preset; const sub = presets.find(it => it.name == subName); if (sub) { // prevent circular references if (hierarchy.indexOf(sub.name) == -1) { const nextHierarchy = [...hierarchy, sub.name]; const nextLabelHierarchy = [...labelHierarchy, tree.label]; tree.children.push(new MenuHeader(sub.name)); sub.quickReplySlots.forEach(subQr => { const subInfo = buildContextMenu(subQr, chain ? tree.mes : null, nextHierarchy, nextLabelHierarchy); tree.children.push(new MenuItem( subInfo.label, subInfo.mes, (evt) => { evt.stopPropagation(); performQuickReply(subInfo.mes.replace(/%%parent(-\d+)?%%/g, (_, index) => { return nextLabelHierarchy.slice(parseInt(index ?? '-1'))[0]; })); }, subInfo.children, )); }); } } }); return tree; } function addQuickReplyBar() { $('#quickReplyBar').remove(); let quickReplyButtonHtml = ''; for (let i = 0; i < extension_settings.quickReply.numberOfSlots; i++) { const qr = extension_settings.quickReply.quickReplySlots[i]; const quickReplyMes = qr?.mes || ''; const quickReplyLabel = qr?.label || ''; const hidden = qr?.hidden ?? false; let expander = ''; if (extension_settings.quickReply.quickReplySlots[i]?.contextMenu?.length) { expander = ''; } quickReplyButtonHtml += `
${quickReplyLabel}${expander}
`; } const quickReplyBarFullHtml = `
${quickReplyButtonHtml}
`; $('#send_form').prepend(quickReplyBarFullHtml); $('.quickReplyButton').on('click', function () { let index = $(this).data('index'); sendQuickReply(index); }); $('.quickReplyButton > .ctx-expander').on('click', function (evt) { evt.stopPropagation(); let index = $(this.closest('.quickReplyButton')).data('index'); const qr = extension_settings.quickReply.quickReplySlots[index]; if (qr.contextMenu?.length) { evt.preventDefault(); const tree = buildContextMenu(qr); const menu = new ContextMenu(tree.children); menu.show(evt); } }) $('.quickReplyButton').on('contextmenu', function (evt) { let index = $(this).data('index'); const qr = extension_settings.quickReply.quickReplySlots[index]; if (qr.contextMenu?.length) { evt.preventDefault(); const tree = buildContextMenu(qr); const menu = new ContextMenu(tree.children); menu.show(evt); } }); } async function moduleWorker() { if (extension_settings.quickReply.quickReplyEnabled === true) { $('#quickReplyBar').toggle(getContext().onlineStatus !== 'no_connection'); } if (extension_settings.quickReply.selectedPreset) { selected_preset = extension_settings.quickReply.selectedPreset; } } async function saveQuickReplyPreset() { const name = await callPopup('Enter a name for the Quick Reply Preset:', 'input'); if (!name) { return; } const quickReplyPreset = { name: name, quickReplyEnabled: extension_settings.quickReply.quickReplyEnabled, quickReplySlots: extension_settings.quickReply.quickReplySlots, numberOfSlots: extension_settings.quickReply.numberOfSlots, AutoInputInject: extension_settings.quickReply.AutoInputInject, selectedPreset: name, } const response = await fetch('/savequickreply', { method: 'POST', headers: getRequestHeaders(), body: JSON.stringify(quickReplyPreset) }); if (response.ok) { const quickReplyPresetIndex = presets.findIndex(x => x.name == name); if (quickReplyPresetIndex == -1) { presets.push(quickReplyPreset); const option = document.createElement('option'); option.selected = true; option.value = name; option.innerText = name; $('#quickReplyPresets').append(option); } else { presets[quickReplyPresetIndex] = quickReplyPreset; $(`#quickReplyPresets option[value="${name}"]`).prop('selected', true); } saveSettingsDebounced(); } else { toastr.warning('Failed to save Quick Reply Preset.') } } //just a copy of save function with the name hardcoded to currently selected preset async function updateQuickReplyPreset() { const name = $("#quickReplyPresets").val() if (!name) { return; } const quickReplyPreset = { name: name, quickReplyEnabled: extension_settings.quickReply.quickReplyEnabled, quickReplySlots: extension_settings.quickReply.quickReplySlots, numberOfSlots: extension_settings.quickReply.numberOfSlots, AutoInputInject: extension_settings.quickReply.AutoInputInject, selectedPreset: name, } const response = await fetch('/savequickreply', { method: 'POST', headers: getRequestHeaders(), body: JSON.stringify(quickReplyPreset) }); if (response.ok) { const quickReplyPresetIndex = presets.findIndex(x => x.name == name); if (quickReplyPresetIndex == -1) { presets.push(quickReplyPreset); const option = document.createElement('option'); option.selected = true; option.value = name; option.innerText = name; $('#quickReplyPresets').append(option); } else { presets[quickReplyPresetIndex] = quickReplyPreset; $(`#quickReplyPresets option[value="${name}"]`).prop('selected', true); } saveSettingsDebounced(); } else { toastr.warning('Failed to save Quick Reply Preset.') } } async function onQuickReplyNumberOfSlotsInput() { const $input = $('#quickReplyNumberOfSlots'); let numberOfSlots = Number($input.val()); if (isNaN(numberOfSlots)) { numberOfSlots = defaultSettings.numberOfSlots; } // Clamp min and max values (from input attributes) if (numberOfSlots < Number($input.attr('min'))) { numberOfSlots = Number($input.attr('min')); } else if (numberOfSlots > Number($input.attr('max'))) { numberOfSlots = Number($input.attr('max')); } extension_settings.quickReply.numberOfSlots = numberOfSlots; extension_settings.quickReply.quickReplySlots.length = numberOfSlots; // Initialize new slots initializeEmptySlots(numberOfSlots); await loadSettings(); addQuickReplyBar(); moduleWorker(); saveSettingsDebounced(); } function initializeEmptySlots(numberOfSlots) { for (let i = 0; i < numberOfSlots; i++) { if (!extension_settings.quickReply.quickReplySlots[i]) { extension_settings.quickReply.quickReplySlots[i] = { mes: '', label: '', enabled: true, }; } } } function generateQuickReplyElements() { let quickReplyHtml = ''; for (let i = 1; i <= extension_settings.quickReply.numberOfSlots; i++) { let itemNumber = i + 1 quickReplyHtml += `
`; } $('#quickReplyContainer').empty().append(quickReplyHtml); for (let i = 1; i <= extension_settings.quickReply.numberOfSlots; i++) { $(`#quickReply${i}Mes`).on('input', function () { onQuickReplyInput(i); }); $(`#quickReply${i}Label`).on('input', function () { onQuickReplyLabelInput(i); }); $(`#quickReply${i}CtxButton`).on('click', function () { onQuickReplyCtxButtonClick(i); }); $(`#quickReplyContainer > [data-order="${i}"]`).attr('data-contextMenu', JSON.stringify(extension_settings.quickReply.quickReplySlots[i - 1]?.contextMenu ?? [])); } $('.quickReplySettings .inline-drawer-toggle').off('click').on('click', function () { for (let i = 1; i <= extension_settings.quickReply.numberOfSlots; i++) { initScrollHeight($(`#quickReply${i}Mes`)); } }); } async function applyQuickReplyPreset(name) { const quickReplyPreset = presets.find(x => x.name == name); if (!quickReplyPreset) { toastr.warning(`error, QR preset '${name}' not found. Confirm you are using proper case sensitivity!`) return; } extension_settings.quickReply = quickReplyPreset; extension_settings.quickReply.selectedPreset = name; saveSettingsDebounced() loadSettings('init') addQuickReplyBar(); moduleWorker(); $(`#quickReplyPresets option[value="${name}"]`).prop('selected', true); console.debug('QR Preset applied: ' + name); } async function doQRPresetSwitch(_, text) { text = String(text) applyQuickReplyPreset(text) } async function doQR(_, text) { if (!text) { toastr.warning('must specify which QR # to use') return } text = Number(text) //use scale starting with 0 //ex: user inputs "/qr 2" >> qr with data-index 1 (but 2nd item displayed) gets triggered let QRnum = Number(text - 1) if (QRnum <= 0) { QRnum = 0 } const whichQR = $("#quickReplies").find(`[data-index='${QRnum}']`); whichQR.trigger('click') } function saveQROrder() { //update html-level order data to match new sort let i = 1 $('#quickReplyContainer').children().each(function () { $(this).attr('data-order', i) $(this).find('input').attr('id', `quickReply${i}Label`) $(this).find('textarea').attr('id', `quickReply${i}Mes`) i++ }); //rebuild the extension_Settings array based on new order i = 1 $('#quickReplyContainer').children().each(function () { onQuickReplyContextMenuChange(i) onQuickReplyLabelInput(i) onQuickReplyInput(i) i++ }); } /** * Executes quick replies on message received. * @param {number} index New message index * @returns {Promise} */ async function onMessageReceived(index) { if (!extension_settings.quickReply.quickReplyEnabled) return; for (let i = 0; i < extension_settings.quickReply.numberOfSlots; i++) { const qr = extension_settings.quickReply.quickReplySlots[i]; if (qr?.autoExecute_botMessage) { const message = getContext().chat[index]; if (message?.mes && message?.mes !== '...') { await sendQuickReply(i); } } } } /** * Executes quick replies on message sent. * @param {number} index New message index * @returns {Promise} */ async function onMessageSent(index) { if (!extension_settings.quickReply.quickReplyEnabled) return; for (let i = 0; i < extension_settings.quickReply.numberOfSlots; i++) { const qr = extension_settings.quickReply.quickReplySlots[i]; if (qr?.autoExecute_userMessage) { const message = getContext().chat[index]; if (message?.mes && message?.mes !== '...') { await sendQuickReply(i); } } } } /** * Executes quick replies on chat changed. * @param {string} chatId New chat id * @returns {Promise} */ async function onChatChanged(chatId) { if (!extension_settings.quickReply.quickReplyEnabled) return; for (let i = 0; i < extension_settings.quickReply.numberOfSlots; i++) { const qr = extension_settings.quickReply.quickReplySlots[i]; if (qr?.autoExecute_chatLoad && chatId) { await sendQuickReply(i); } } } jQuery(async () => { moduleWorker(); setInterval(moduleWorker, UPDATE_INTERVAL); const settingsHtml = `
Quick Reply
Customize your Quick Replies:
`; $('#extensions_settings2').append(settingsHtml); // Add event handler for quickActionEnabled $('#quickActionEnabled').on('input', onQuickActionEnabledInput); $('#placeBeforeInputEnabled').on('input', onPlaceBeforeInputEnabledInput); $('#AutoInputInject').on('input', onAutoInputInject); $('#quickReplyEnabled').on('input', onQuickReplyEnabledInput); $('#quickReplyNumberOfSlotsApply').on('click', onQuickReplyNumberOfSlotsInput); $("#quickReplyPresetSaveButton").on('click', saveQuickReplyPreset); $("#quickReplyPresetUpdateButton").on('click', updateQuickReplyPreset); $('#quickReplyContainer').sortable({ delay: getSortableDelay(), stop: saveQROrder, }); $("#quickReplyPresets").on('change', async function () { const quickReplyPresetSelected = $(this).find(':selected').val(); extension_settings.quickReplyPreset = quickReplyPresetSelected; applyQuickReplyPreset(quickReplyPresetSelected); saveSettingsDebounced(); }); await loadSettings('init'); addQuickReplyBar(); eventSource.on(event_types.MESSAGE_RECEIVED, onMessageReceived); eventSource.on(event_types.MESSAGE_SENT, onMessageSent); eventSource.on(event_types.CHAT_CHANGED, onChatChanged); }); jQuery(() => { registerSlashCommand('qr', doQR, [], '(number) – activates the specified Quick Reply', true, true); registerSlashCommand('qrset', doQRPresetSwitch, [], '(name) – swaps to the specified Quick Reply Preset', true, true); })