1539 lines
54 KiB
JavaScript
1539 lines
54 KiB
JavaScript
// Move chat functions here from script.js (eventually)
|
|
|
|
import css from '../lib/css-parser.mjs';
|
|
import {
|
|
addCopyToCodeBlocks,
|
|
appendMediaToMessage,
|
|
characters,
|
|
chat,
|
|
eventSource,
|
|
event_types,
|
|
getCurrentChatId,
|
|
getRequestHeaders,
|
|
hideSwipeButtons,
|
|
name2,
|
|
reloadCurrentChat,
|
|
saveChatDebounced,
|
|
saveSettingsDebounced,
|
|
showSwipeButtons,
|
|
this_chid,
|
|
saveChatConditional,
|
|
chat_metadata,
|
|
} from '../script.js';
|
|
import { selected_group } from './group-chats.js';
|
|
import { power_user } from './power-user.js';
|
|
import {
|
|
extractTextFromHTML,
|
|
extractTextFromMarkdown,
|
|
extractTextFromPDF,
|
|
extractTextFromEpub,
|
|
getBase64Async,
|
|
getStringHash,
|
|
humanFileSize,
|
|
saveBase64AsFile,
|
|
extractTextFromOffice,
|
|
} from './utils.js';
|
|
import { extension_settings, renderExtensionTemplateAsync, saveMetadataDebounced } from './extensions.js';
|
|
import { POPUP_RESULT, POPUP_TYPE, Popup, callGenericPopup } from './popup.js';
|
|
import { ScraperManager } from './scrapers.js';
|
|
import { DragAndDropHandler } from './dragdrop.js';
|
|
import { renderTemplateAsync } from './templates.js';
|
|
|
|
/**
|
|
* @typedef {Object} FileAttachment
|
|
* @property {string} url File URL
|
|
* @property {number} size File size
|
|
* @property {string} name File name
|
|
* @property {number} created Timestamp
|
|
* @property {string} [text] File text
|
|
*/
|
|
|
|
/**
|
|
* @typedef {function} ConverterFunction
|
|
* @param {File} file File object
|
|
* @returns {Promise<string>} Converted file text
|
|
*/
|
|
|
|
const fileSizeLimit = 1024 * 1024 * 100; // 100 MB
|
|
const ATTACHMENT_SOURCE = {
|
|
GLOBAL: 'global',
|
|
CHARACTER: 'character',
|
|
CHAT: 'chat',
|
|
};
|
|
|
|
/**
|
|
* @type {Record<string, ConverterFunction>} File converters
|
|
*/
|
|
const converters = {
|
|
'application/pdf': extractTextFromPDF,
|
|
'text/html': extractTextFromHTML,
|
|
'text/markdown': extractTextFromMarkdown,
|
|
'application/epub+zip': extractTextFromEpub,
|
|
'application/vnd.openxmlformats-officedocument.wordprocessingml.document': extractTextFromOffice,
|
|
'application/vnd.openxmlformats-officedocument.spreadsheetml.sheet': extractTextFromOffice,
|
|
'application/vnd.openxmlformats-officedocument.presentationml.presentation': extractTextFromOffice,
|
|
'application/vnd.oasis.opendocument.text': extractTextFromOffice,
|
|
'application/vnd.oasis.opendocument.presentation': extractTextFromOffice,
|
|
'application/vnd.oasis.opendocument.spreadsheet': extractTextFromOffice,
|
|
};
|
|
|
|
/**
|
|
* Finds a matching key in the converters object.
|
|
* @param {string} type MIME type
|
|
* @returns {string} Matching key
|
|
*/
|
|
function findConverterKey(type) {
|
|
return Object.keys(converters).find((key) => {
|
|
// Match exact type
|
|
if (type === key) {
|
|
return true;
|
|
}
|
|
|
|
// Match wildcards
|
|
if (key.endsWith('*')) {
|
|
return type.startsWith(key.substring(0, key.length - 1));
|
|
}
|
|
|
|
return false;
|
|
});
|
|
}
|
|
|
|
/**
|
|
* Determines if the file type has a converter function.
|
|
* @param {string} type MIME type
|
|
* @returns {boolean} True if the file type is convertible, false otherwise.
|
|
*/
|
|
function isConvertible(type) {
|
|
return Boolean(findConverterKey(type));
|
|
}
|
|
|
|
/**
|
|
* Gets the converter function for a file type.
|
|
* @param {string} type MIME type
|
|
* @returns {ConverterFunction} Converter function
|
|
*/
|
|
function getConverter(type) {
|
|
const key = findConverterKey(type);
|
|
return key && converters[key];
|
|
}
|
|
|
|
/**
|
|
* Mark a range of messages as hidden ("is_system") or not.
|
|
* @param {number} start Starting message ID
|
|
* @param {number} end Ending message ID (inclusive)
|
|
* @param {boolean} unhide If true, unhide the messages instead.
|
|
* @returns {Promise<void>}
|
|
*/
|
|
export async function hideChatMessageRange(start, end, unhide) {
|
|
if (!getCurrentChatId()) return;
|
|
|
|
if (isNaN(start)) return;
|
|
if (!end) end = start;
|
|
const hide = !unhide;
|
|
|
|
for (let messageId = start; messageId <= end; messageId++) {
|
|
const message = chat[messageId];
|
|
if (!message) continue;
|
|
|
|
const messageBlock = $(`.mes[mesid="${messageId}"]`);
|
|
if (!messageBlock.length) continue;
|
|
|
|
message.is_system = hide;
|
|
messageBlock.attr('is_system', String(hide));
|
|
}
|
|
|
|
// Reload swipes. Useful when a last message is hidden.
|
|
hideSwipeButtons();
|
|
showSwipeButtons();
|
|
|
|
saveChatDebounced();
|
|
}
|
|
|
|
/**
|
|
* Mark message as hidden (system message).
|
|
* @deprecated Use hideChatMessageRange.
|
|
* @param {number} messageId Message ID
|
|
* @param {JQuery<Element>} _messageBlock Unused
|
|
* @returns {Promise<void>}
|
|
*/
|
|
export async function hideChatMessage(messageId, _messageBlock) {
|
|
return hideChatMessageRange(messageId, messageId, false);
|
|
}
|
|
|
|
/**
|
|
* Mark message as visible (non-system message).
|
|
* @deprecated Use hideChatMessageRange.
|
|
* @param {number} messageId Message ID
|
|
* @param {JQuery<Element>} _messageBlock Unused
|
|
* @returns {Promise<void>}
|
|
*/
|
|
export async function unhideChatMessage(messageId, _messageBlock) {
|
|
return hideChatMessageRange(messageId, messageId, true);
|
|
}
|
|
|
|
/**
|
|
* Adds a file attachment to the message.
|
|
* @param {object} message Message object
|
|
* @returns {Promise<void>} A promise that resolves when file is uploaded.
|
|
*/
|
|
export async function populateFileAttachment(message, inputId = 'file_form_input') {
|
|
try {
|
|
if (!message) return;
|
|
if (!message.extra) message.extra = {};
|
|
const fileInput = document.getElementById(inputId);
|
|
if (!(fileInput instanceof HTMLInputElement)) return;
|
|
const file = fileInput.files[0];
|
|
if (!file) return;
|
|
|
|
const slug = getStringHash(file.name);
|
|
const fileNamePrefix = `${Date.now()}_${slug}`;
|
|
const fileBase64 = await getBase64Async(file);
|
|
let base64Data = fileBase64.split(',')[1];
|
|
|
|
// If file is image
|
|
if (file.type.startsWith('image/')) {
|
|
const extension = file.type.split('/')[1];
|
|
const imageUrl = await saveBase64AsFile(base64Data, name2, fileNamePrefix, extension);
|
|
message.extra.image = imageUrl;
|
|
message.extra.inline_image = true;
|
|
} else {
|
|
const uniqueFileName = `${fileNamePrefix}.txt`;
|
|
|
|
if (isConvertible(file.type)) {
|
|
try {
|
|
const converter = getConverter(file.type);
|
|
const fileText = await converter(file);
|
|
base64Data = window.btoa(unescape(encodeURIComponent(fileText)));
|
|
} catch (error) {
|
|
toastr.error(String(error), 'Could not convert file');
|
|
console.error('Could not convert file', error);
|
|
}
|
|
}
|
|
|
|
const fileUrl = await uploadFileAttachment(uniqueFileName, base64Data);
|
|
|
|
if (!fileUrl) {
|
|
return;
|
|
}
|
|
|
|
message.extra.file = {
|
|
url: fileUrl,
|
|
size: file.size,
|
|
name: file.name,
|
|
created: Date.now(),
|
|
};
|
|
}
|
|
|
|
} catch (error) {
|
|
console.error('Could not upload file', error);
|
|
} finally {
|
|
$('#file_form').trigger('reset');
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Uploads file to the server.
|
|
* @param {string} fileName
|
|
* @param {string} base64Data
|
|
* @returns {Promise<string>} File URL
|
|
*/
|
|
export async function uploadFileAttachment(fileName, base64Data) {
|
|
try {
|
|
const result = await fetch('/api/files/upload', {
|
|
method: 'POST',
|
|
headers: getRequestHeaders(),
|
|
body: JSON.stringify({
|
|
name: fileName,
|
|
data: base64Data,
|
|
}),
|
|
});
|
|
|
|
if (!result.ok) {
|
|
const error = await result.text();
|
|
throw new Error(error);
|
|
}
|
|
|
|
const responseData = await result.json();
|
|
return responseData.path;
|
|
} catch (error) {
|
|
toastr.error(String(error), 'Could not upload file');
|
|
console.error('Could not upload file', error);
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Downloads file from the server.
|
|
* @param {string} url File URL
|
|
* @returns {Promise<string>} File text
|
|
*/
|
|
export async function getFileAttachment(url) {
|
|
try {
|
|
const result = await fetch(url, {
|
|
method: 'GET',
|
|
cache: 'force-cache',
|
|
headers: getRequestHeaders(),
|
|
});
|
|
|
|
if (!result.ok) {
|
|
const error = await result.text();
|
|
throw new Error(error);
|
|
}
|
|
|
|
const text = await result.text();
|
|
return text;
|
|
} catch (error) {
|
|
toastr.error(error, 'Could not download file');
|
|
console.error('Could not download file', error);
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Validates file to make sure it is not binary or not image.
|
|
* @param {File} file File object
|
|
* @returns {Promise<boolean>} True if file is valid, false otherwise.
|
|
*/
|
|
async function validateFile(file) {
|
|
const fileText = await file.text();
|
|
const isImage = file.type.startsWith('image/');
|
|
const isBinary = /^[\x00-\x08\x0E-\x1F\x7F-\xFF]*$/.test(fileText);
|
|
|
|
if (!isImage && file.size > fileSizeLimit) {
|
|
toastr.error(`File is too big. Maximum size is ${humanFileSize(fileSizeLimit)}.`);
|
|
return false;
|
|
}
|
|
|
|
// If file is binary
|
|
if (isBinary && !isImage && !isConvertible(file.type)) {
|
|
toastr.error('Binary files are not supported. Select a text file or image.');
|
|
return false;
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
export function hasPendingFileAttachment() {
|
|
const fileInput = document.getElementById('file_form_input');
|
|
if (!(fileInput instanceof HTMLInputElement)) return false;
|
|
const file = fileInput.files[0];
|
|
return !!file;
|
|
}
|
|
|
|
/**
|
|
* Displays file information in the message sending form.
|
|
* @param {File} file File object
|
|
* @returns {Promise<void>}
|
|
*/
|
|
async function onFileAttach(file) {
|
|
if (!file) return;
|
|
|
|
const isValid = await validateFile(file);
|
|
|
|
// If file is binary
|
|
if (!isValid) {
|
|
$('#file_form').trigger('reset');
|
|
return;
|
|
}
|
|
|
|
$('#file_form .file_name').text(file.name);
|
|
$('#file_form .file_size').text(humanFileSize(file.size));
|
|
$('#file_form').removeClass('displayNone');
|
|
|
|
// Reset form on chat change
|
|
eventSource.once(event_types.CHAT_CHANGED, () => {
|
|
$('#file_form').trigger('reset');
|
|
});
|
|
}
|
|
|
|
/**
|
|
* Deletes file from message.
|
|
* @param {number} messageId Message ID
|
|
*/
|
|
async function deleteMessageFile(messageId) {
|
|
const confirm = await callGenericPopup('Are you sure you want to delete this file?', POPUP_TYPE.CONFIRM);
|
|
|
|
if (confirm !== POPUP_RESULT.AFFIRMATIVE) {
|
|
console.debug('Delete file cancelled');
|
|
return;
|
|
}
|
|
|
|
const message = chat[messageId];
|
|
|
|
if (!message?.extra?.file) {
|
|
console.debug('Message has no file');
|
|
return;
|
|
}
|
|
|
|
const url = message.extra.file.url;
|
|
|
|
delete message.extra.file;
|
|
$(`.mes[mesid="${messageId}"] .mes_file_container`).remove();
|
|
await saveChatConditional();
|
|
await deleteFileFromServer(url);
|
|
}
|
|
|
|
|
|
/**
|
|
* Opens file from message in a modal.
|
|
* @param {number} messageId Message ID
|
|
*/
|
|
async function viewMessageFile(messageId) {
|
|
const messageFile = chat[messageId]?.extra?.file;
|
|
|
|
if (!messageFile) {
|
|
console.debug('Message has no file or it is empty');
|
|
return;
|
|
}
|
|
|
|
await openFilePopup(messageFile);
|
|
}
|
|
|
|
/**
|
|
* Inserts a file embed into the message.
|
|
* @param {number} messageId
|
|
* @param {JQuery<HTMLElement>} messageBlock
|
|
* @returns {Promise<void>}
|
|
*/
|
|
function embedMessageFile(messageId, messageBlock) {
|
|
const message = chat[messageId];
|
|
|
|
if (!message) {
|
|
console.warn('Failed to find message with id', messageId);
|
|
return;
|
|
}
|
|
|
|
$('#embed_file_input')
|
|
.off('change')
|
|
.on('change', parseAndUploadEmbed)
|
|
.trigger('click');
|
|
|
|
async function parseAndUploadEmbed(e) {
|
|
const file = e.target.files[0];
|
|
if (!file) return;
|
|
|
|
const isValid = await validateFile(file);
|
|
|
|
if (!isValid) {
|
|
$('#file_form').trigger('reset');
|
|
return;
|
|
}
|
|
|
|
await populateFileAttachment(message, 'embed_file_input');
|
|
await eventSource.emit(event_types.MESSAGE_FILE_EMBEDDED, messageId);
|
|
appendMediaToMessage(message, messageBlock);
|
|
await saveChatConditional();
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Appends file content to the message text.
|
|
* @param {object} message Message object
|
|
* @param {string} messageText Message text
|
|
* @returns {Promise<string>} Message text with file content appended.
|
|
*/
|
|
export async function appendFileContent(message, messageText) {
|
|
if (message.extra?.file) {
|
|
const fileText = message.extra.file.text || (await getFileAttachment(message.extra.file.url));
|
|
|
|
if (fileText) {
|
|
const fileWrapped = `${fileText}\n\n`;
|
|
message.extra.fileLength = fileWrapped.length;
|
|
messageText = fileWrapped + messageText;
|
|
}
|
|
}
|
|
return messageText;
|
|
}
|
|
|
|
/**
|
|
* Replaces style tags in the message text with custom tags with encoded content.
|
|
* @param {string} text
|
|
* @returns {string} Encoded message text
|
|
* @copyright https://github.com/kwaroran/risuAI
|
|
*/
|
|
export function encodeStyleTags(text) {
|
|
const styleRegex = /<style>(.+?)<\/style>/gms;
|
|
return text.replaceAll(styleRegex, (_, match) => {
|
|
return `<custom-style>${escape(match)}</custom-style>`;
|
|
});
|
|
}
|
|
|
|
/**
|
|
* Sanitizes custom style tags in the message text to prevent DOM pollution.
|
|
* @param {string} text Message text
|
|
* @returns {string} Sanitized message text
|
|
* @copyright https://github.com/kwaroran/risuAI
|
|
*/
|
|
export function decodeStyleTags(text) {
|
|
const styleDecodeRegex = /<custom-style>(.+?)<\/custom-style>/gms;
|
|
const mediaAllowed = isExternalMediaAllowed();
|
|
|
|
function sanitizeRule(rule) {
|
|
if (Array.isArray(rule.selectors)) {
|
|
for (let i = 0; i < rule.selectors.length; i++) {
|
|
const selector = rule.selectors[i];
|
|
if (selector) {
|
|
const selectors = (selector.split(' ') ?? []).map((v) => {
|
|
if (v.startsWith('.')) {
|
|
return '.custom-' + v.substring(1);
|
|
}
|
|
return v;
|
|
}).join(' ');
|
|
|
|
rule.selectors[i] = '.mes_text ' + selectors;
|
|
}
|
|
}
|
|
}
|
|
if (!mediaAllowed && Array.isArray(rule.declarations) && rule.declarations.length > 0) {
|
|
rule.declarations = rule.declarations.filter(declaration => !declaration.value.includes('://'));
|
|
}
|
|
}
|
|
|
|
function sanitizeRuleSet(ruleSet) {
|
|
if (Array.isArray(ruleSet.selectors) || Array.isArray(ruleSet.declarations)) {
|
|
sanitizeRule(ruleSet);
|
|
}
|
|
|
|
if (Array.isArray(ruleSet.rules)) {
|
|
ruleSet.rules = ruleSet.rules.filter(rule => rule.type !== 'import');
|
|
|
|
for (const mediaRule of ruleSet.rules) {
|
|
sanitizeRuleSet(mediaRule);
|
|
}
|
|
}
|
|
}
|
|
|
|
return text.replaceAll(styleDecodeRegex, (_, style) => {
|
|
try {
|
|
let styleCleaned = unescape(style).replaceAll(/<br\/>/g, '');
|
|
const ast = css.parse(styleCleaned);
|
|
const sheet = ast?.stylesheet;
|
|
if (sheet) {
|
|
sanitizeRuleSet(ast.stylesheet);
|
|
}
|
|
return `<style>${css.stringify(ast)}</style>`;
|
|
} catch (error) {
|
|
return `CSS ERROR: ${error}`;
|
|
}
|
|
});
|
|
}
|
|
|
|
async function openExternalMediaOverridesDialog() {
|
|
const entityId = getCurrentEntityId();
|
|
|
|
if (!entityId) {
|
|
toastr.info('No character or group selected');
|
|
return;
|
|
}
|
|
|
|
const template = $(await renderTemplateAsync('forbidMedia'));
|
|
template.find('.forbid_media_global_state_forbidden').toggle(power_user.forbid_external_media);
|
|
template.find('.forbid_media_global_state_allowed').toggle(!power_user.forbid_external_media);
|
|
|
|
if (power_user.external_media_allowed_overrides.includes(entityId)) {
|
|
template.find('#forbid_media_override_allowed').prop('checked', true);
|
|
}
|
|
else if (power_user.external_media_forbidden_overrides.includes(entityId)) {
|
|
template.find('#forbid_media_override_forbidden').prop('checked', true);
|
|
}
|
|
else {
|
|
template.find('#forbid_media_override_global').prop('checked', true);
|
|
}
|
|
|
|
callGenericPopup(template, POPUP_TYPE.TEXT, '', { wide: false, large: false });
|
|
}
|
|
|
|
export function getCurrentEntityId() {
|
|
if (selected_group) {
|
|
return String(selected_group);
|
|
}
|
|
|
|
return characters[this_chid]?.avatar ?? null;
|
|
}
|
|
|
|
export function isExternalMediaAllowed() {
|
|
const entityId = getCurrentEntityId();
|
|
if (!entityId) {
|
|
return !power_user.forbid_external_media;
|
|
}
|
|
|
|
if (power_user.external_media_allowed_overrides.includes(entityId)) {
|
|
return true;
|
|
}
|
|
|
|
if (power_user.external_media_forbidden_overrides.includes(entityId)) {
|
|
return false;
|
|
}
|
|
|
|
return !power_user.forbid_external_media;
|
|
}
|
|
|
|
async function enlargeMessageImage() {
|
|
const mesBlock = $(this).closest('.mes');
|
|
const mesId = mesBlock.attr('mesid');
|
|
const message = chat[mesId];
|
|
const imgSrc = message?.extra?.image;
|
|
const title = message?.extra?.title;
|
|
|
|
if (!imgSrc) {
|
|
return;
|
|
}
|
|
|
|
const img = document.createElement('img');
|
|
img.classList.add('img_enlarged');
|
|
img.src = imgSrc;
|
|
const imgHolder = document.createElement('div');
|
|
imgHolder.classList.add('img_enlarged_holder');
|
|
imgHolder.append(img);
|
|
const imgContainer = $('<div><pre><code></code></pre></div>');
|
|
imgContainer.prepend(imgHolder);
|
|
imgContainer.addClass('img_enlarged_container');
|
|
imgContainer.find('code').addClass('txt').text(title);
|
|
const titleEmpty = !title || title.trim().length === 0;
|
|
imgContainer.find('pre').toggle(!titleEmpty);
|
|
addCopyToCodeBlocks(imgContainer);
|
|
|
|
const popup = new Popup(imgContainer, POPUP_TYPE.DISPLAY, '', { large: true, transparent: true });
|
|
|
|
popup.dlg.style.width = 'unset';
|
|
popup.dlg.style.height = 'unset';
|
|
|
|
img.addEventListener('click', () => {
|
|
const shouldZoom = !img.classList.contains('zoomed');
|
|
img.classList.toggle('zoomed', shouldZoom);
|
|
});
|
|
|
|
await popup.show();
|
|
}
|
|
|
|
async function deleteMessageImage() {
|
|
const value = await callGenericPopup('<h3>Delete image from message?<br>This action can\'t be undone.</h3>', POPUP_TYPE.CONFIRM);
|
|
|
|
if (value !== POPUP_RESULT.AFFIRMATIVE) {
|
|
return;
|
|
}
|
|
|
|
const mesBlock = $(this).closest('.mes');
|
|
const mesId = mesBlock.attr('mesid');
|
|
const message = chat[mesId];
|
|
delete message.extra.image;
|
|
delete message.extra.inline_image;
|
|
delete message.extra.title;
|
|
delete message.extra.append_title;
|
|
mesBlock.find('.mes_img_container').removeClass('img_extra');
|
|
mesBlock.find('.mes_img').attr('src', '');
|
|
await saveChatConditional();
|
|
}
|
|
|
|
/**
|
|
* Deletes file from the server.
|
|
* @param {string} url Path to the file on the server
|
|
* @param {boolean} [silent=false] If true, do not show error messages
|
|
* @returns {Promise<boolean>} True if file was deleted, false otherwise.
|
|
*/
|
|
async function deleteFileFromServer(url, silent = false) {
|
|
try {
|
|
const result = await fetch('/api/files/delete', {
|
|
method: 'POST',
|
|
headers: getRequestHeaders(),
|
|
body: JSON.stringify({ path: url }),
|
|
});
|
|
|
|
if (!result.ok && !silent) {
|
|
const error = await result.text();
|
|
throw new Error(error);
|
|
}
|
|
|
|
await eventSource.emit(event_types.FILE_ATTACHMENT_DELETED, url);
|
|
return true;
|
|
} catch (error) {
|
|
toastr.error(String(error), 'Could not delete file');
|
|
console.error('Could not delete file', error);
|
|
return false;
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Opens file attachment in a modal.
|
|
* @param {FileAttachment} attachment File attachment
|
|
*/
|
|
async function openFilePopup(attachment) {
|
|
const fileText = attachment.text || (await getFileAttachment(attachment.url));
|
|
|
|
const modalTemplate = $('<div><pre><code></code></pre></div>');
|
|
modalTemplate.find('code').addClass('txt').text(fileText);
|
|
modalTemplate.addClass('file_modal').addClass('textarea_compact').addClass('fontsize90p');
|
|
addCopyToCodeBlocks(modalTemplate);
|
|
|
|
callGenericPopup(modalTemplate, POPUP_TYPE.TEXT, '', { wide: true, large: true });
|
|
}
|
|
|
|
/**
|
|
* Edit a file attachment in a notepad-like modal.
|
|
* @param {FileAttachment} attachment Attachment to edit
|
|
* @param {string} source Attachment source
|
|
* @param {function} callback Callback function
|
|
*/
|
|
async function editAttachment(attachment, source, callback) {
|
|
const originalFileText = attachment.text || (await getFileAttachment(attachment.url));
|
|
const template = $(await renderExtensionTemplateAsync('attachments', 'notepad'));
|
|
|
|
let editedFileText = originalFileText;
|
|
template.find('[name="notepadFileContent"]').val(editedFileText).on('input', function () {
|
|
editedFileText = String($(this).val());
|
|
});
|
|
|
|
let editedFileName = attachment.name;
|
|
template.find('[name="notepadFileName"]').val(editedFileName).on('input', function () {
|
|
editedFileName = String($(this).val());
|
|
});
|
|
|
|
const result = await callGenericPopup(template, POPUP_TYPE.CONFIRM, '', { wide: true, large: true, okButton: 'Save', cancelButton: 'Cancel' });
|
|
|
|
if (result !== POPUP_RESULT.AFFIRMATIVE) {
|
|
return;
|
|
}
|
|
|
|
if (editedFileText === originalFileText && editedFileName === attachment.name) {
|
|
return;
|
|
}
|
|
|
|
const nullCallback = () => { };
|
|
await deleteAttachment(attachment, source, nullCallback, false);
|
|
const file = new File([editedFileText], editedFileName, { type: 'text/plain' });
|
|
await uploadFileAttachmentToServer(file, source);
|
|
|
|
callback();
|
|
}
|
|
|
|
/**
|
|
* Downloads an attachment to the user's device.
|
|
* @param {FileAttachment} attachment Attachment to download
|
|
*/
|
|
async function downloadAttachment(attachment) {
|
|
const fileText = attachment.text || (await getFileAttachment(attachment.url));
|
|
const blob = new Blob([fileText], { type: 'text/plain' });
|
|
const url = URL.createObjectURL(blob);
|
|
const a = document.createElement('a');
|
|
a.href = url;
|
|
a.download = attachment.name;
|
|
a.click();
|
|
URL.revokeObjectURL(url);
|
|
}
|
|
|
|
/**
|
|
* Removes an attachment from the disabled list.
|
|
* @param {FileAttachment} attachment Attachment to enable
|
|
* @param {function} callback Success callback
|
|
*/
|
|
function enableAttachment(attachment, callback) {
|
|
ensureAttachmentsExist();
|
|
extension_settings.disabled_attachments = extension_settings.disabled_attachments.filter(url => url !== attachment.url);
|
|
saveSettingsDebounced();
|
|
callback();
|
|
}
|
|
|
|
/**
|
|
* Adds an attachment to the disabled list.
|
|
* @param {FileAttachment} attachment Attachment to disable
|
|
* @param {function} callback Success callback
|
|
*/
|
|
function disableAttachment(attachment, callback) {
|
|
ensureAttachmentsExist();
|
|
extension_settings.disabled_attachments.push(attachment.url);
|
|
saveSettingsDebounced();
|
|
callback();
|
|
}
|
|
|
|
/**
|
|
* Moves a file attachment to a different source.
|
|
* @param {FileAttachment} attachment Attachment to moves
|
|
* @param {string} source Source of the attachment
|
|
* @param {function} callback Success callback
|
|
* @returns {Promise<void>} A promise that resolves when the attachment is moved.
|
|
*/
|
|
async function moveAttachment(attachment, source, callback) {
|
|
let selectedTarget = source;
|
|
const targets = getAvailableTargets();
|
|
const template = $(await renderExtensionTemplateAsync('attachments', 'move-attachment', { name: attachment.name, targets }));
|
|
template.find('.moveAttachmentTarget').val(source).on('input', function () {
|
|
selectedTarget = String($(this).val());
|
|
});
|
|
|
|
const result = await callGenericPopup(template, POPUP_TYPE.CONFIRM, '', { wide: false, large: false, okButton: 'Move', cancelButton: 'Cancel' });
|
|
|
|
if (result !== POPUP_RESULT.AFFIRMATIVE) {
|
|
console.debug('Move attachment cancelled');
|
|
return;
|
|
}
|
|
|
|
if (selectedTarget === source) {
|
|
console.debug('Move attachment cancelled: same source and target');
|
|
return;
|
|
}
|
|
|
|
const content = await getFileAttachment(attachment.url);
|
|
const file = new File([content], attachment.name, { type: 'text/plain' });
|
|
await deleteAttachment(attachment, source, () => { }, false);
|
|
await uploadFileAttachmentToServer(file, selectedTarget);
|
|
callback();
|
|
}
|
|
|
|
/**
|
|
* Deletes an attachment from the server and the chat.
|
|
* @param {FileAttachment} attachment Attachment to delete
|
|
* @param {string} source Source of the attachment
|
|
* @param {function} callback Callback function
|
|
* @param {boolean} [confirm=true] If true, show a confirmation dialog
|
|
* @returns {Promise<void>} A promise that resolves when the attachment is deleted.
|
|
*/
|
|
export async function deleteAttachment(attachment, source, callback, confirm = true) {
|
|
if (confirm) {
|
|
const result = await callGenericPopup('Are you sure you want to delete this attachment?', POPUP_TYPE.CONFIRM);
|
|
|
|
if (result !== POPUP_RESULT.AFFIRMATIVE) {
|
|
return;
|
|
}
|
|
}
|
|
|
|
ensureAttachmentsExist();
|
|
|
|
switch (source) {
|
|
case 'global':
|
|
extension_settings.attachments = extension_settings.attachments.filter((a) => a.url !== attachment.url);
|
|
saveSettingsDebounced();
|
|
break;
|
|
case 'chat':
|
|
chat_metadata.attachments = chat_metadata.attachments.filter((a) => a.url !== attachment.url);
|
|
saveMetadataDebounced();
|
|
break;
|
|
case 'character':
|
|
extension_settings.character_attachments[characters[this_chid]?.avatar] = extension_settings.character_attachments[characters[this_chid]?.avatar].filter((a) => a.url !== attachment.url);
|
|
break;
|
|
}
|
|
|
|
if (Array.isArray(extension_settings.disabled_attachments) && extension_settings.disabled_attachments.includes(attachment.url)) {
|
|
extension_settings.disabled_attachments = extension_settings.disabled_attachments.filter(url => url !== attachment.url);
|
|
saveSettingsDebounced();
|
|
}
|
|
|
|
const silent = confirm === false;
|
|
await deleteFileFromServer(attachment.url, silent);
|
|
callback();
|
|
}
|
|
|
|
/**
|
|
* Determines if the attachment is disabled.
|
|
* @param {FileAttachment} attachment Attachment to check
|
|
* @returns {boolean} True if attachment is disabled, false otherwise.
|
|
*/
|
|
function isAttachmentDisabled(attachment) {
|
|
return extension_settings.disabled_attachments.some(url => url === attachment?.url);
|
|
}
|
|
|
|
/**
|
|
* Opens the attachment manager.
|
|
*/
|
|
async function openAttachmentManager() {
|
|
/**
|
|
* Renders a list of attachments.
|
|
* @param {FileAttachment[]} attachments List of attachments
|
|
* @param {string} source Source of the attachments
|
|
*/
|
|
async function renderList(attachments, source) {
|
|
/**
|
|
* Sorts attachments by sortField and sortOrder.
|
|
* @param {FileAttachment} a First attachment
|
|
* @param {FileAttachment} b Second attachment
|
|
* @returns {number} Sort order
|
|
*/
|
|
function sortFn(a, b) {
|
|
const sortValueA = a[sortField];
|
|
const sortValueB = b[sortField];
|
|
if (typeof sortValueA === 'string' && typeof sortValueB === 'string') {
|
|
return sortValueA.localeCompare(sortValueB) * (sortOrder === 'asc' ? 1 : -1);
|
|
}
|
|
return (sortValueA - sortValueB) * (sortOrder === 'asc' ? 1 : -1);
|
|
}
|
|
|
|
/**
|
|
* Filters attachments by name.
|
|
* @param {FileAttachment} a Attachment
|
|
* @returns {boolean} True if attachment matches the filter, false otherwise.
|
|
*/
|
|
function filterFn(a) {
|
|
if (!filterString) {
|
|
return true;
|
|
}
|
|
|
|
return a.name.toLowerCase().includes(filterString.toLowerCase());
|
|
}
|
|
const sources = {
|
|
[ATTACHMENT_SOURCE.GLOBAL]: '.globalAttachmentsList',
|
|
[ATTACHMENT_SOURCE.CHARACTER]: '.characterAttachmentsList',
|
|
[ATTACHMENT_SOURCE.CHAT]: '.chatAttachmentsList',
|
|
};
|
|
|
|
const selected = template
|
|
.find(sources[source])
|
|
.find('.attachmentListItemCheckbox:checked')
|
|
.map((_, el) => $(el).closest('.attachmentListItem').attr('data-attachment-url'))
|
|
.get();
|
|
|
|
template.find(sources[source]).empty();
|
|
|
|
// Sort attachments by sortField and sortOrder, and apply filter
|
|
const sortedAttachmentList = attachments.slice().filter(filterFn).sort(sortFn);
|
|
|
|
for (const attachment of sortedAttachmentList) {
|
|
const isDisabled = isAttachmentDisabled(attachment);
|
|
const attachmentTemplate = template.find('.attachmentListItemTemplate .attachmentListItem').clone();
|
|
attachmentTemplate.toggleClass('disabled', isDisabled);
|
|
attachmentTemplate.attr('data-attachment-url', attachment.url);
|
|
attachmentTemplate.attr('data-attachment-source', source);
|
|
attachmentTemplate.find('.attachmentFileIcon').attr('title', attachment.url);
|
|
attachmentTemplate.find('.attachmentListItemName').text(attachment.name);
|
|
attachmentTemplate.find('.attachmentListItemSize').text(humanFileSize(attachment.size));
|
|
attachmentTemplate.find('.attachmentListItemCreated').text(new Date(attachment.created).toLocaleString());
|
|
attachmentTemplate.find('.viewAttachmentButton').on('click', () => openFilePopup(attachment));
|
|
attachmentTemplate.find('.editAttachmentButton').on('click', () => editAttachment(attachment, source, renderAttachments));
|
|
attachmentTemplate.find('.deleteAttachmentButton').on('click', () => deleteAttachment(attachment, source, renderAttachments));
|
|
attachmentTemplate.find('.downloadAttachmentButton').on('click', () => downloadAttachment(attachment));
|
|
attachmentTemplate.find('.moveAttachmentButton').on('click', () => moveAttachment(attachment, source, renderAttachments));
|
|
attachmentTemplate.find('.enableAttachmentButton').toggle(isDisabled).on('click', () => enableAttachment(attachment, renderAttachments));
|
|
attachmentTemplate.find('.disableAttachmentButton').toggle(!isDisabled).on('click', () => disableAttachment(attachment, renderAttachments));
|
|
template.find(sources[source]).append(attachmentTemplate);
|
|
|
|
if (selected.includes(attachment.url)) {
|
|
attachmentTemplate.find('.attachmentListItemCheckbox').prop('checked', true);
|
|
}
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Renders buttons for the attachment manager.
|
|
*/
|
|
async function renderButtons() {
|
|
const sources = {
|
|
[ATTACHMENT_SOURCE.GLOBAL]: '.globalAttachmentsTitle',
|
|
[ATTACHMENT_SOURCE.CHARACTER]: '.characterAttachmentsTitle',
|
|
[ATTACHMENT_SOURCE.CHAT]: '.chatAttachmentsTitle',
|
|
};
|
|
|
|
const modal = template.find('.actionButtonsModal').hide();
|
|
const scrapers = ScraperManager.getDataBankScrapers();
|
|
|
|
for (const scraper of scrapers) {
|
|
const isAvailable = await ScraperManager.isScraperAvailable(scraper.id);
|
|
if (!isAvailable) {
|
|
continue;
|
|
}
|
|
|
|
const buttonTemplate = template.find('.actionButtonTemplate .actionButton').clone();
|
|
if (scraper.iconAvailable) {
|
|
buttonTemplate.find('.actionButtonIcon').addClass(scraper.iconClass);
|
|
buttonTemplate.find('.actionButtonImg').remove();
|
|
} else {
|
|
buttonTemplate.find('.actionButtonImg').attr('src', scraper.iconClass);
|
|
buttonTemplate.find('.actionButtonIcon').remove();
|
|
}
|
|
buttonTemplate.find('.actionButtonText').text(scraper.name);
|
|
buttonTemplate.attr('title', scraper.description);
|
|
buttonTemplate.on('click', () => {
|
|
const target = modal.attr('data-attachment-manager-target');
|
|
runScraper(scraper.id, target, renderAttachments);
|
|
});
|
|
modal.append(buttonTemplate);
|
|
}
|
|
|
|
const modalButtonData = Object.entries(sources).map(entry => {
|
|
const [source, selector] = entry;
|
|
const button = template.find(selector).find('.openActionModalButton').get(0);
|
|
|
|
if (!button) {
|
|
return;
|
|
}
|
|
|
|
const bodyListener = (e) => {
|
|
if (modal.is(':visible') && (!$(e.target).closest('.openActionModalButton').length)) {
|
|
modal.hide();
|
|
}
|
|
|
|
// Replay a click if the modal was already open by another button
|
|
if ($(e.target).closest('.openActionModalButton').length && !modal.is(':visible')) {
|
|
modal.show();
|
|
}
|
|
};
|
|
document.body.addEventListener('click', bodyListener);
|
|
|
|
const popper = Popper.createPopper(button, modal.get(0), { placement: 'bottom-end' });
|
|
button.addEventListener('click', () => {
|
|
modal.attr('data-attachment-manager-target', source);
|
|
modal.toggle();
|
|
popper.update();
|
|
});
|
|
|
|
return [popper, bodyListener];
|
|
}).filter(Boolean);
|
|
|
|
return () => {
|
|
modalButtonData.forEach(p => {
|
|
const [popper, bodyListener] = p;
|
|
popper.destroy();
|
|
document.body.removeEventListener('click', bodyListener);
|
|
});
|
|
modal.remove();
|
|
};
|
|
}
|
|
|
|
async function renderAttachments() {
|
|
/** @type {FileAttachment[]} */
|
|
const globalAttachments = extension_settings.attachments ?? [];
|
|
/** @type {FileAttachment[]} */
|
|
const chatAttachments = chat_metadata.attachments ?? [];
|
|
/** @type {FileAttachment[]} */
|
|
const characterAttachments = extension_settings.character_attachments?.[characters[this_chid]?.avatar] ?? [];
|
|
|
|
await renderList(globalAttachments, ATTACHMENT_SOURCE.GLOBAL);
|
|
await renderList(chatAttachments, ATTACHMENT_SOURCE.CHAT);
|
|
await renderList(characterAttachments, ATTACHMENT_SOURCE.CHARACTER);
|
|
|
|
const isNotCharacter = this_chid === undefined || selected_group;
|
|
const isNotInChat = getCurrentChatId() === undefined;
|
|
template.find('.characterAttachmentsBlock').toggle(!isNotCharacter);
|
|
template.find('.chatAttachmentsBlock').toggle(!isNotInChat);
|
|
|
|
const characterName = characters[this_chid]?.name || 'Anonymous';
|
|
template.find('.characterAttachmentsName').text(characterName);
|
|
|
|
const chatName = getCurrentChatId() || 'Unnamed chat';
|
|
template.find('.chatAttachmentsName').text(chatName);
|
|
}
|
|
|
|
const dragDropHandler = new DragAndDropHandler('.popup', async (files, event) => {
|
|
let selectedTarget = ATTACHMENT_SOURCE.GLOBAL;
|
|
const targets = getAvailableTargets();
|
|
|
|
const targetSelectTemplate = $(await renderExtensionTemplateAsync('attachments', 'files-dropped', { count: files.length, targets: targets }));
|
|
targetSelectTemplate.find('.droppedFilesTarget').on('input', function () {
|
|
selectedTarget = String($(this).val());
|
|
});
|
|
const result = await callGenericPopup(targetSelectTemplate, POPUP_TYPE.CONFIRM, '', { wide: false, large: false, okButton: 'Upload', cancelButton: 'Cancel' });
|
|
if (result !== POPUP_RESULT.AFFIRMATIVE) {
|
|
console.log('File upload cancelled');
|
|
return;
|
|
}
|
|
for (const file of files) {
|
|
await uploadFileAttachmentToServer(file, selectedTarget);
|
|
}
|
|
renderAttachments();
|
|
});
|
|
|
|
let sortField = localStorage.getItem('DataBank_sortField') || 'created';
|
|
let sortOrder = localStorage.getItem('DataBank_sortOrder') || 'desc';
|
|
let filterString = '';
|
|
|
|
const template = $(await renderExtensionTemplateAsync('attachments', 'manager', {}));
|
|
|
|
template.find('.attachmentSearch').on('input', function () {
|
|
filterString = String($(this).val());
|
|
renderAttachments();
|
|
});
|
|
template.find('.attachmentSort').on('change', function () {
|
|
if (!(this instanceof HTMLSelectElement) || this.selectedOptions.length === 0) {
|
|
return;
|
|
}
|
|
|
|
sortField = this.selectedOptions[0].dataset.sortField;
|
|
sortOrder = this.selectedOptions[0].dataset.sortOrder;
|
|
localStorage.setItem('DataBank_sortField', sortField);
|
|
localStorage.setItem('DataBank_sortOrder', sortOrder);
|
|
renderAttachments();
|
|
});
|
|
function handleBulkAction(action) {
|
|
return async () => {
|
|
const selectedAttachments = document.querySelectorAll('.attachmentListItemCheckboxContainer .attachmentListItemCheckbox:checked');
|
|
|
|
if (selectedAttachments.length === 0) {
|
|
toastr.info('No attachments selected.', 'Data Bank');
|
|
return;
|
|
}
|
|
|
|
if (action.confirmMessage) {
|
|
const confirm = await callGenericPopup(action.confirmMessage, POPUP_TYPE.CONFIRM);
|
|
if (confirm !== POPUP_RESULT.AFFIRMATIVE) {
|
|
return;
|
|
}
|
|
}
|
|
|
|
const includeDisabled = true;
|
|
const attachments = getDataBankAttachments(includeDisabled);
|
|
selectedAttachments.forEach(async (checkbox) => {
|
|
const listItem = checkbox.closest('.attachmentListItem');
|
|
if (!(listItem instanceof HTMLElement)) {
|
|
return;
|
|
}
|
|
const url = listItem.dataset.attachmentUrl;
|
|
const source = listItem.dataset.attachmentSource;
|
|
const attachment = attachments.find(a => a.url === url);
|
|
if (!attachment) {
|
|
return;
|
|
}
|
|
await action.perform(attachment, source);
|
|
});
|
|
|
|
document.querySelectorAll('.attachmentListItemCheckbox, .attachmentsBulkEditCheckbox').forEach(checkbox => {
|
|
if (checkbox instanceof HTMLInputElement) {
|
|
checkbox.checked = false;
|
|
}
|
|
});
|
|
|
|
await renderAttachments();
|
|
};
|
|
}
|
|
|
|
template.find('.bulkActionDisable').on('click', handleBulkAction({
|
|
perform: (attachment) => disableAttachment(attachment, () => { }),
|
|
}));
|
|
|
|
template.find('.bulkActionEnable').on('click', handleBulkAction({
|
|
perform: (attachment) => enableAttachment(attachment, () => { }),
|
|
}));
|
|
|
|
template.find('.bulkActionDelete').on('click', handleBulkAction({
|
|
confirmMessage: 'Are you sure you want to delete the selected attachments?',
|
|
perform: async (attachment, source) => await deleteAttachment(attachment, source, () => { }, false),
|
|
}));
|
|
|
|
template.find('.bulkActionSelectAll').on('click', () => {
|
|
$('.attachmentListItemCheckbox:visible').each((_, checkbox) => {
|
|
if (checkbox instanceof HTMLInputElement) {
|
|
checkbox.checked = true;
|
|
}
|
|
});
|
|
});
|
|
template.find('.bulkActionSelectNone').on('click', () => {
|
|
$('.attachmentListItemCheckbox:visible').each((_, checkbox) => {
|
|
if (checkbox instanceof HTMLInputElement) {
|
|
checkbox.checked = false;
|
|
}
|
|
});
|
|
});
|
|
|
|
const cleanupFn = await renderButtons();
|
|
await verifyAttachments();
|
|
await renderAttachments();
|
|
await callGenericPopup(template, POPUP_TYPE.TEXT, '', { wide: true, large: true, okButton: 'Close', allowVerticalScrolling: true });
|
|
|
|
cleanupFn();
|
|
dragDropHandler.destroy();
|
|
}
|
|
|
|
/**
|
|
* Gets a list of available targets for attachments.
|
|
* @returns {string[]} List of available targets
|
|
*/
|
|
function getAvailableTargets() {
|
|
const targets = Object.values(ATTACHMENT_SOURCE);
|
|
|
|
const isNotCharacter = this_chid === undefined || selected_group;
|
|
const isNotInChat = getCurrentChatId() === undefined;
|
|
|
|
if (isNotCharacter) {
|
|
targets.splice(targets.indexOf(ATTACHMENT_SOURCE.CHARACTER), 1);
|
|
}
|
|
|
|
if (isNotInChat) {
|
|
targets.splice(targets.indexOf(ATTACHMENT_SOURCE.CHAT), 1);
|
|
}
|
|
|
|
return targets;
|
|
}
|
|
|
|
/**
|
|
* Runs a known scraper on a source and saves the result as an attachment.
|
|
* @param {string} scraperId Id of the scraper
|
|
* @param {string} target Target for the attachment
|
|
* @param {function} callback Callback function
|
|
* @returns {Promise<void>} A promise that resolves when the source is scraped.
|
|
*/
|
|
async function runScraper(scraperId, target, callback) {
|
|
try {
|
|
console.log(`Running scraper ${scraperId} for ${target}`);
|
|
const files = await ScraperManager.runDataBankScraper(scraperId);
|
|
|
|
if (!Array.isArray(files)) {
|
|
console.warn('Scraping returned nothing');
|
|
return;
|
|
}
|
|
|
|
if (files.length === 0) {
|
|
console.warn('Scraping returned no files');
|
|
toastr.info('No files were scraped.', 'Data Bank');
|
|
return;
|
|
}
|
|
|
|
for (const file of files) {
|
|
await uploadFileAttachmentToServer(file, target);
|
|
}
|
|
|
|
toastr.success(`Scraped ${files.length} files from ${scraperId} to ${target}.`, 'Data Bank');
|
|
callback();
|
|
}
|
|
catch (error) {
|
|
console.error('Scraping failed', error);
|
|
toastr.error('Check browser console for details.', 'Scraping failed');
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Uploads a file attachment to the server.
|
|
* @param {File} file File to upload
|
|
* @param {string} target Target for the attachment
|
|
* @returns {Promise<string>} Path to the uploaded file
|
|
*/
|
|
export async function uploadFileAttachmentToServer(file, target) {
|
|
const isValid = await validateFile(file);
|
|
|
|
if (!isValid) {
|
|
return;
|
|
}
|
|
|
|
let base64Data = await getBase64Async(file);
|
|
const slug = getStringHash(file.name);
|
|
const uniqueFileName = `${Date.now()}_${slug}.txt`;
|
|
|
|
if (isConvertible(file.type)) {
|
|
try {
|
|
const converter = getConverter(file.type);
|
|
const fileText = await converter(file);
|
|
base64Data = window.btoa(unescape(encodeURIComponent(fileText)));
|
|
} catch (error) {
|
|
toastr.error(String(error), 'Could not convert file');
|
|
console.error('Could not convert file', error);
|
|
}
|
|
} else {
|
|
const fileText = await file.text();
|
|
base64Data = window.btoa(unescape(encodeURIComponent(fileText)));
|
|
}
|
|
|
|
const fileUrl = await uploadFileAttachment(uniqueFileName, base64Data);
|
|
const convertedSize = Math.round(base64Data.length * 0.75);
|
|
|
|
if (!fileUrl) {
|
|
return;
|
|
}
|
|
|
|
const attachment = {
|
|
url: fileUrl,
|
|
size: convertedSize,
|
|
name: file.name,
|
|
created: Date.now(),
|
|
};
|
|
|
|
ensureAttachmentsExist();
|
|
|
|
switch (target) {
|
|
case ATTACHMENT_SOURCE.GLOBAL:
|
|
extension_settings.attachments.push(attachment);
|
|
saveSettingsDebounced();
|
|
break;
|
|
case ATTACHMENT_SOURCE.CHAT:
|
|
chat_metadata.attachments.push(attachment);
|
|
saveMetadataDebounced();
|
|
break;
|
|
case ATTACHMENT_SOURCE.CHARACTER:
|
|
extension_settings.character_attachments[characters[this_chid]?.avatar].push(attachment);
|
|
saveSettingsDebounced();
|
|
break;
|
|
}
|
|
|
|
return fileUrl;
|
|
}
|
|
|
|
function ensureAttachmentsExist() {
|
|
if (!Array.isArray(extension_settings.disabled_attachments)) {
|
|
extension_settings.disabled_attachments = [];
|
|
}
|
|
|
|
if (!Array.isArray(extension_settings.attachments)) {
|
|
extension_settings.attachments = [];
|
|
}
|
|
|
|
if (!Array.isArray(chat_metadata.attachments)) {
|
|
chat_metadata.attachments = [];
|
|
}
|
|
|
|
if (this_chid !== undefined && characters[this_chid]) {
|
|
if (!extension_settings.character_attachments) {
|
|
extension_settings.character_attachments = {};
|
|
}
|
|
|
|
if (!Array.isArray(extension_settings.character_attachments[characters[this_chid].avatar])) {
|
|
extension_settings.character_attachments[characters[this_chid].avatar] = [];
|
|
}
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Gets all currently available attachments. Ignores disabled attachments by default.
|
|
* @param {boolean} [includeDisabled=false] If true, include disabled attachments
|
|
* @returns {FileAttachment[]} List of attachments
|
|
*/
|
|
export function getDataBankAttachments(includeDisabled = false) {
|
|
ensureAttachmentsExist();
|
|
const globalAttachments = extension_settings.attachments ?? [];
|
|
const chatAttachments = chat_metadata.attachments ?? [];
|
|
const characterAttachments = extension_settings.character_attachments?.[characters[this_chid]?.avatar] ?? [];
|
|
|
|
return [...globalAttachments, ...chatAttachments, ...characterAttachments].filter(x => includeDisabled || !isAttachmentDisabled(x));
|
|
}
|
|
|
|
/**
|
|
* Gets all attachments for a specific source. Includes disabled attachments by default.
|
|
* @param {string} source Attachment source
|
|
* @param {boolean} [includeDisabled=true] If true, include disabled attachments
|
|
* @returns {FileAttachment[]} List of attachments
|
|
*/
|
|
export function getDataBankAttachmentsForSource(source, includeDisabled = true) {
|
|
ensureAttachmentsExist();
|
|
|
|
function getBySource() {
|
|
switch (source) {
|
|
case ATTACHMENT_SOURCE.GLOBAL:
|
|
return extension_settings.attachments ?? [];
|
|
case ATTACHMENT_SOURCE.CHAT:
|
|
return chat_metadata.attachments ?? [];
|
|
case ATTACHMENT_SOURCE.CHARACTER:
|
|
return extension_settings.character_attachments?.[characters[this_chid]?.avatar] ?? [];
|
|
}
|
|
|
|
return [];
|
|
}
|
|
|
|
return getBySource().filter(x => includeDisabled || !isAttachmentDisabled(x));
|
|
}
|
|
|
|
/**
|
|
* Verifies all attachments in the Data Bank.
|
|
* @returns {Promise<void>} A promise that resolves when attachments are verified.
|
|
*/
|
|
async function verifyAttachments() {
|
|
for (const source of Object.values(ATTACHMENT_SOURCE)) {
|
|
await verifyAttachmentsForSource(source);
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Verifies all attachments for a specific source.
|
|
* @param {string} source Attachment source
|
|
* @returns {Promise<void>} A promise that resolves when attachments are verified.
|
|
*/
|
|
async function verifyAttachmentsForSource(source) {
|
|
try {
|
|
const attachments = getDataBankAttachmentsForSource(source);
|
|
const urls = attachments.map(a => a.url);
|
|
const response = await fetch('/api/files/verify', {
|
|
method: 'POST',
|
|
headers: getRequestHeaders(),
|
|
body: JSON.stringify({ urls }),
|
|
});
|
|
|
|
if (!response.ok) {
|
|
const error = await response.text();
|
|
throw new Error(error);
|
|
}
|
|
|
|
const verifiedUrls = await response.json();
|
|
for (const attachment of attachments) {
|
|
if (verifiedUrls[attachment.url] === false) {
|
|
console.log('Deleting orphaned attachment', attachment);
|
|
await deleteAttachment(attachment, source, () => { }, false);
|
|
}
|
|
}
|
|
} catch (error) {
|
|
console.error('Attachment verification failed', error);
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Registers a file converter function.
|
|
* @param {string} mimeType MIME type
|
|
* @param {ConverterFunction} converter Function to convert file
|
|
* @returns {void}
|
|
*/
|
|
export function registerFileConverter(mimeType, converter) {
|
|
if (typeof mimeType !== 'string' || typeof converter !== 'function') {
|
|
console.error('Invalid converter registration');
|
|
return;
|
|
}
|
|
|
|
if (Object.keys(converters).includes(mimeType)) {
|
|
console.error('Converter already registered');
|
|
return;
|
|
}
|
|
|
|
converters[mimeType] = converter;
|
|
}
|
|
|
|
jQuery(function () {
|
|
$(document).on('click', '.mes_hide', async function () {
|
|
const messageBlock = $(this).closest('.mes');
|
|
const messageId = Number(messageBlock.attr('mesid'));
|
|
await hideChatMessageRange(messageId, messageId, false);
|
|
});
|
|
|
|
$(document).on('click', '.mes_unhide', async function () {
|
|
const messageBlock = $(this).closest('.mes');
|
|
const messageId = Number(messageBlock.attr('mesid'));
|
|
await hideChatMessageRange(messageId, messageId, true);
|
|
});
|
|
|
|
$(document).on('click', '.mes_file_delete', async function () {
|
|
const messageBlock = $(this).closest('.mes');
|
|
const messageId = Number(messageBlock.attr('mesid'));
|
|
await deleteMessageFile(messageId);
|
|
});
|
|
|
|
$(document).on('click', '.mes_file_open', async function () {
|
|
const messageBlock = $(this).closest('.mes');
|
|
const messageId = Number(messageBlock.attr('mesid'));
|
|
await viewMessageFile(messageId);
|
|
});
|
|
|
|
// Do not change. #attachFile is added by extension.
|
|
$(document).on('click', '#attachFile', function () {
|
|
$('#file_form_input').trigger('click');
|
|
});
|
|
|
|
// Do not change. #manageAttachments is added by extension.
|
|
$(document).on('click', '#manageAttachments', function () {
|
|
openAttachmentManager();
|
|
});
|
|
|
|
$(document).on('click', '.mes_embed', function () {
|
|
const messageBlock = $(this).closest('.mes');
|
|
const messageId = Number(messageBlock.attr('mesid'));
|
|
embedMessageFile(messageId, messageBlock);
|
|
});
|
|
|
|
$(document).on('click', '.editor_maximize', function () {
|
|
const broId = $(this).attr('data-for');
|
|
const bro = $(`#${broId}`);
|
|
const withTab = $(this).attr('data-tab');
|
|
|
|
if (!bro.length) {
|
|
console.error('Could not find editor with id', broId);
|
|
return;
|
|
}
|
|
|
|
const wrapper = document.createElement('div');
|
|
wrapper.classList.add('height100p', 'wide100p', 'flex-container');
|
|
wrapper.classList.add('flexFlowColumn', 'justifyCenter', 'alignitemscenter');
|
|
const textarea = document.createElement('textarea');
|
|
textarea.value = String(bro.val());
|
|
textarea.classList.add('height100p', 'wide100p', 'maximized_textarea');
|
|
bro.hasClass('monospace') && textarea.classList.add('monospace');
|
|
textarea.addEventListener('input', function () {
|
|
bro.val(textarea.value).trigger('input');
|
|
});
|
|
wrapper.appendChild(textarea);
|
|
|
|
if (withTab) {
|
|
textarea.addEventListener('keydown', (evt) => {
|
|
if (evt.key == 'Tab' && !evt.shiftKey && !evt.ctrlKey && !evt.altKey) {
|
|
evt.preventDefault();
|
|
const start = textarea.selectionStart;
|
|
const end = textarea.selectionEnd;
|
|
if (end - start > 0 && textarea.value.substring(start, end).includes('\n')) {
|
|
const lineStart = textarea.value.lastIndexOf('\n', start);
|
|
const count = textarea.value.substring(lineStart, end).split('\n').length - 1;
|
|
textarea.value = `${textarea.value.substring(0, lineStart)}${textarea.value.substring(lineStart, end).replace(/\n/g, '\n\t')}${textarea.value.substring(end)}`;
|
|
textarea.selectionStart = start + 1;
|
|
textarea.selectionEnd = end + count;
|
|
} else {
|
|
textarea.value = `${textarea.value.substring(0, start)}\t${textarea.value.substring(end)}`;
|
|
textarea.selectionStart = start + 1;
|
|
textarea.selectionEnd = end + 1;
|
|
}
|
|
} else if (evt.key == 'Tab' && evt.shiftKey && !evt.ctrlKey && !evt.altKey) {
|
|
evt.preventDefault();
|
|
const start = textarea.selectionStart;
|
|
const end = textarea.selectionEnd;
|
|
const lineStart = textarea.value.lastIndexOf('\n', start);
|
|
const count = textarea.value.substring(lineStart, end).split('\n\t').length - 1;
|
|
textarea.value = `${textarea.value.substring(0, lineStart)}${textarea.value.substring(lineStart, end).replace(/\n\t/g, '\n')}${textarea.value.substring(end)}`;
|
|
textarea.selectionStart = start - 1;
|
|
textarea.selectionEnd = end - count;
|
|
}
|
|
});
|
|
}
|
|
|
|
callGenericPopup(wrapper, POPUP_TYPE.TEXT, '', { wide: true, large: true });
|
|
});
|
|
|
|
$(document).on('click', 'body.documentstyle .mes .mes_text', function () {
|
|
if (window.getSelection().toString()) return;
|
|
if ($('.edit_textarea').length) return;
|
|
$(this).closest('.mes').find('.mes_edit').trigger('click');
|
|
});
|
|
|
|
$(document).on('click', '.open_media_overrides', openExternalMediaOverridesDialog);
|
|
$(document).on('input', '#forbid_media_override_allowed', function () {
|
|
const entityId = getCurrentEntityId();
|
|
if (!entityId) return;
|
|
power_user.external_media_allowed_overrides.push(entityId);
|
|
power_user.external_media_forbidden_overrides = power_user.external_media_forbidden_overrides.filter((v) => v !== entityId);
|
|
saveSettingsDebounced();
|
|
reloadCurrentChat();
|
|
});
|
|
$(document).on('input', '#forbid_media_override_forbidden', function () {
|
|
const entityId = getCurrentEntityId();
|
|
if (!entityId) return;
|
|
power_user.external_media_forbidden_overrides.push(entityId);
|
|
power_user.external_media_allowed_overrides = power_user.external_media_allowed_overrides.filter((v) => v !== entityId);
|
|
saveSettingsDebounced();
|
|
reloadCurrentChat();
|
|
});
|
|
$(document).on('input', '#forbid_media_override_global', function () {
|
|
const entityId = getCurrentEntityId();
|
|
if (!entityId) return;
|
|
power_user.external_media_allowed_overrides = power_user.external_media_allowed_overrides.filter((v) => v !== entityId);
|
|
power_user.external_media_forbidden_overrides = power_user.external_media_forbidden_overrides.filter((v) => v !== entityId);
|
|
saveSettingsDebounced();
|
|
reloadCurrentChat();
|
|
});
|
|
|
|
$(document).on('click', '.mes_img_enlarge', enlargeMessageImage);
|
|
$(document).on('click', '.mes_img_delete', deleteMessageImage);
|
|
|
|
$('#file_form_input').on('change', async () => {
|
|
const fileInput = document.getElementById('file_form_input');
|
|
if (!(fileInput instanceof HTMLInputElement)) return;
|
|
const file = fileInput.files[0];
|
|
await onFileAttach(file);
|
|
});
|
|
$('#file_form').on('reset', function () {
|
|
$('#file_form').addClass('displayNone');
|
|
});
|
|
|
|
document.getElementById('send_textarea').addEventListener('paste', async function (event) {
|
|
if (event.clipboardData.files.length === 0) {
|
|
return;
|
|
}
|
|
|
|
event.preventDefault();
|
|
event.stopPropagation();
|
|
|
|
const fileInput = document.getElementById('file_form_input');
|
|
if (!(fileInput instanceof HTMLInputElement)) return;
|
|
|
|
// Workaround for Firefox: Use a DataTransfer object to indirectly set fileInput.files
|
|
const dataTransfer = new DataTransfer();
|
|
for (let i = 0; i < event.clipboardData.files.length; i++) {
|
|
dataTransfer.items.add(event.clipboardData.files[i]);
|
|
}
|
|
|
|
fileInput.files = dataTransfer.files;
|
|
await onFileAttach(fileInput.files[0]);
|
|
});
|
|
});
|