Remove all connections button for the popups

9a79c6b1a4138c04864d2664a0f7078b76b9994a

Wolfsblvt <wolfsblvt@gmail.com>

Parents: 3fc4207cdb3240dc648025fc8924dd0f01b3fdff

public/script.js+2 -0 Viewed

     updatePersonaConnectionsAvatarList,
     getConnectedPersonas,
     askForPersonaSelection,
 +    getCurrentConnectionObj,
 } from './scripts/personas.js';
 import { getBackgrounds, initBackgrounds, loadBackgroundSettings, background_settings } from './scripts/backgrounds.js';
 import { hideLoader, showLoader } from './scripts/loader.js';

         const selectedPersona = await askForPersonaSelection(t`Persona Connections`, message, connections, {
             okButton: t`Ok`,
             highlightPersonas: true,
 +            targetedChar: getCurrentConnectionObj(),
             shiftClickHandler: (element, ev) => {
                 const personaId = $(element).attr('data-pid');

public/scripts/personas.js+48 -6 Viewed

  * @param {string} [options.okButton='None'] - The label for the OK button
  * @param {(element: HTMLElement, ev: MouseEvent) => any} [options.shiftClickHandler] - A function to handle shift-click
  * @param {boolean|string[]} [options.highlightPersonas=false] - Whether to highlight personas - either by providing a list of persona keys, or true to highlight all present in current chat
 + * @param {PersonaConnection} [options.targetedChar] - The targeted character or gorup for this persona selection
  * @returns {Promise<string?>} - A promise that resolves to the selected persona id or null if no selection was made
  */
 export async function askForPersonaSelection(title, text, personas, { okButton = 'None', shiftClickHandler = undefined, highlightPersonas = false, targetedChar = undefined } = {}) {
     const content = document.createElement('div');
     const titleElement = document.createElement('h3');
     titleElement.textContent = title;

     if (personas.length > 0)
         buildPersonaAvatarList(personaListBlock, personas, { interactable: true });
     else
         personaListBlock.textContent = 't`[NoCurrently no personas connected]'`;
     const personasToHighlight = highlightPersonas instanceof Array ? highlightPersonas : (highlightPersonas ? getPersonasOfCurrentChat() : []);

+        }
     });
 -    const popup = new Popup(content, POPUP_TYPE.TEXT, '', { okButton: okButton });
 +    /** @type {import('./popup.js').CustomPopupButton[]} */
 +    const customButtons = [];
 +    if (targetedChar) {
 +        customButtons.push({
 +            text: t`Remove All Connections`,
 +            result: 2,
 +            action: () => {
 +                for (const [personaId, description] of Object.entries(power_user.persona_descriptions)) {
 +                    /** @type {PersonaConnection[]} */
 +                    const connections = description.connections;
 +                    if (connections) {
 +                        power_user.persona_descriptions[personaId].connections = connections.filter(c => {
 +                            if (targetedChar.type == c.type && targetedChar.id == c.id) return false;
 +                            return true;
 +                        });
 +                    }
 +                }
++
 +                saveSettingsDebounced();
 +                updatePersonaConnectionsAvatarList();
 +                if (power_user.persona_show_notifications) {
 +                    const name = targetedChar.type == 'character' ? characters[targetedChar.id]?.name : groups[targetedChar.id]?.name;
 +                    toastr.info(t`All connections to ${name} have been removed.`, t`Personas unlocked`);
 +                }
 +            },
 +        });
 +    }
++
 +    const popup = new Popup(content, POPUP_TYPE.TEXT, '', { okButton: okButton, customButtons: customButtons });
     const result = await popup.show();
     return Number(result) > 100 ? personas[Number(result) - 100] : null;
+}

             break;
+        }
         case 'character': {
 -            const newConnection = menu_type === 'character_edit' ? { type: 'character', id: characters[this_chid]?.avatar } :
 +            const newConnection = getCurrentConnectionObj();
 -                menu_type === 'group_edit' ? { type: 'group', id: selected_group } : null;
             /** @type {PersonaConnection[]} */
             const connections = power_user.persona_descriptions[user_avatar].connections?.filter(c => !isPersonaConnectionLocked(c)) ?? [];
             if (newConnection && newConnection.id) {

@@ -1249,7 +1277,7 @@ async function loadPersonaForCurrentChat() {
1249	1277	} else {
1250	1278	chatPersona = await askForPersonaSelection(t`Select Persona`,
1251	1279	t`Multiple personas are connected to this character.\nSelect a persona to use for this chat.`,
1252	1280	connectedPersonas, { highlightPersonas: true, targetedChar: getCurrentConnectionObj() });
1253	1281	}
1254	1282	}
1255	1283

     return connectedPersonas;
+}
 +/**
 + * Retrieves the current connection object based on whether the current chat is with a char or a group.
 + *
 + * @returns {PersonaConnection} An object representing the current connection
 + */
++
 +export function getCurrentConnectionObj() {
 +    if (menu_type === 'group_edit')
 +        return { type: 'group', id: selected_group };
 +    if (menu_type == 'character_edit')
 +        return { type: 'character', id: characters[this_chid]?.avatar };
 +    return null;
 +}
++
 function onBackupPersonas() {
     const timestamp = new Date().toISOString().split('T')[0].replace(/-/g, '');
     const filename = `personas_${timestamp}.json`;