mirror of
https://github.com/SillyTavern/SillyTavern.git
synced 2024-12-14 02:16:14 +01:00
356 lines
12 KiB
JavaScript
356 lines
12 KiB
JavaScript
import { saveSettingsDebounced, callPopup, getRequestHeaders } from "../../../script.js";
|
|
import { getContext, extension_settings } from "../../extensions.js";
|
|
import { initScrollHeight, resetScrollHeight } from "../../utils.js";
|
|
import { executeSlashCommands, getSlashCommandsHelp, registerSlashCommand } from "../../slash-commands.js";
|
|
|
|
|
|
export { MODULE_NAME };
|
|
|
|
const MODULE_NAME = 'quick-reply';
|
|
const UPDATE_INTERVAL = 1000;
|
|
let presets = [];
|
|
let selected_preset = '';
|
|
|
|
const defaultSettings = {
|
|
quickReplyEnabled: true,
|
|
numberOfSlots: 5,
|
|
quickReplySlots: [],
|
|
}
|
|
|
|
//method from worldinfo
|
|
async function updateQuickReplyPresetList() {
|
|
var 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.log(presets)
|
|
$("#quickReplyPresets").find('option[value!=""]').remove();
|
|
|
|
|
|
if (presets !== undefined) {
|
|
presets.forEach((item, i) => {
|
|
$("#quickReplyPresets").append(`<option value='${item.name}'${selected_preset.includes(item.name) ? ' selected' : ''}>${item.name}</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 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);
|
|
}
|
|
|
|
function onQuickReplyInput(id) {
|
|
extension_settings.quickReply.quickReplySlots[id - 1].mes = $(`#quickReply${id}Mes`).val();
|
|
$(`#quickReply${id}`).attr('title', ($(`#quickReply${id}Mes`).val()));
|
|
resetScrollHeight($(`#quickReply${id}Mes`));
|
|
saveSettingsDebounced();
|
|
}
|
|
|
|
function onQuickReplyLabelInput(id) {
|
|
extension_settings.quickReply.quickReplySlots[id - 1].label = $(`#quickReply${id}Label`).val();
|
|
$(`#quickReply${id}`).text($(`#quickReply${id}Label`).val());
|
|
saveSettingsDebounced();
|
|
}
|
|
|
|
async function onQuickReplyEnabledInput() {
|
|
let isEnabled = $(this).prop('checked')
|
|
extension_settings.quickReply.quickReplyEnabled = !!isEnabled;
|
|
if (isEnabled === true) {
|
|
$("#quickReplyBar").show();
|
|
} else { $("#quickReplyBar").hide(); }
|
|
saveSettingsDebounced();
|
|
}
|
|
|
|
async function sendQuickReply(index) {
|
|
const prompt = extension_settings.quickReply.quickReplySlots[index]?.mes || '';
|
|
|
|
if (!prompt) {
|
|
console.warn(`Quick reply slot ${index} is empty! Aborting.`);
|
|
return;
|
|
}
|
|
|
|
$("#send_textarea").val(prompt);
|
|
$("#send_but").trigger('click');
|
|
}
|
|
|
|
function addQuickReplyBar() {
|
|
$('#quickReplyBar').remove();
|
|
let quickReplyButtonHtml = '';
|
|
|
|
for (let i = 0; i < extension_settings.quickReply.numberOfSlots; i++) {
|
|
let quickReplyMes = extension_settings.quickReply.quickReplySlots[i]?.mes || '';
|
|
let quickReplyLabel = extension_settings.quickReply.quickReplySlots[i]?.label || '';
|
|
quickReplyButtonHtml += `<div title="${quickReplyMes}" class="quickReplyButton" data-index="${i}" id="quickReply${i + 1}">${quickReplyLabel}</div>`;
|
|
}
|
|
|
|
const quickReplyBarFullHtml = `
|
|
<div id="quickReplyBar" class="flex-container flexGap5">
|
|
<div id="quickReplies">
|
|
${quickReplyButtonHtml}
|
|
</div>
|
|
</div>
|
|
`;
|
|
|
|
$('#send_form').prepend(quickReplyBarFullHtml);
|
|
|
|
$('.quickReplyButton').on('click', function () {
|
|
let index = $(this).data('index');
|
|
sendQuickReply(index);
|
|
});
|
|
}
|
|
|
|
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,
|
|
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}"]`).attr('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++) {
|
|
quickReplyHtml += `
|
|
<div class="flex-container alignitemsflexstart">
|
|
<input class="text_pole wide30p" id="quickReply${i}Label" placeholder="(Add a button label)">
|
|
<textarea id="quickReply${i}Mes" placeholder="(custom message here)" class="text_pole widthUnset flex1" rows="2"></textarea>
|
|
</div>
|
|
`;
|
|
}
|
|
|
|
$('#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); });
|
|
}
|
|
|
|
$('.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}"]`).attr('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')
|
|
}
|
|
|
|
jQuery(async () => {
|
|
|
|
moduleWorker();
|
|
setInterval(moduleWorker, UPDATE_INTERVAL);
|
|
const settingsHtml = `
|
|
<div class="quickReplySettings">
|
|
<div class="inline-drawer">
|
|
<div class="inline-drawer-toggle inline-drawer-header">
|
|
<b>Quick Reply</b>
|
|
<div class="inline-drawer-icon fa-solid fa-circle-chevron-down down"></div>
|
|
</div>
|
|
<div class="inline-drawer-content">
|
|
<div class="flex-container ">
|
|
<label class="checkbox_label marginBot10 wide100p flexnowrap">
|
|
<input id="quickReplyEnabled" type="checkbox" />
|
|
Enable Quick Replies
|
|
</label>
|
|
<div class="flex-container flexnowrap wide100p">
|
|
<select id="quickReplyPresets" name="quickreply-preset">
|
|
</select>
|
|
<i id="quickReplyPresetSaveButton" class="fa-solid fa-save"></i>
|
|
</div>
|
|
<label for="quickReplyNumberOfSlots">Number of slots:</label>
|
|
</div>
|
|
<div class="flex-container flexGap5 flexnowrap">
|
|
<input id="quickReplyNumberOfSlots" class="text_pole" type="number" min="1" max="100" value="" />
|
|
<div class="menu_button menu_button_icon" id="quickReplyNumberOfSlotsApply">
|
|
<div class="fa-solid fa-check"></div>
|
|
<span>Apply</span>
|
|
</div>
|
|
</div>
|
|
<small><i>Customize your Quick Replies:</i></small><br>
|
|
<div id="quickReplyContainer">
|
|
</div>
|
|
</div>
|
|
</div>`;
|
|
|
|
$('#extensions_settings2').append(settingsHtml);
|
|
|
|
$('#quickReplyEnabled').on('input', onQuickReplyEnabledInput);
|
|
$('#quickReplyNumberOfSlotsApply').on('click', onQuickReplyNumberOfSlotsInput);
|
|
$("#quickReplyPresetSaveButton").on('click', saveQuickReplyPreset);
|
|
|
|
$("#quickReplyPresets").on('change', async function () {
|
|
const quickReplyPresetSelected = $(this).find(':selected').val();
|
|
extension_settings.quickReplyPreset = quickReplyPresetSelected;
|
|
applyQuickReplyPreset(quickReplyPresetSelected);
|
|
saveSettingsDebounced();
|
|
|
|
});
|
|
|
|
await loadSettings('init');
|
|
addQuickReplyBar();
|
|
|
|
});
|
|
|
|
$(document).ready(() => {
|
|
registerSlashCommand('qr', doQR, [], "- requires number argument, activates the specified QuickReply", true, true);
|
|
registerSlashCommand('qrset', doQRPresetSwitch, [], "- arg: QuickReply Preset Name, swaps to that QR preset", true, true);
|
|
|
|
})
|