Prevent ephemeral message extra changes vanish upon swiping

0c9b301a57362df7c824cebf99cb3053fd40db85

Cohee <18619528+Cohee1207@users.noreply.github.com>

public/script.js+19 -0 Viewed

     return { type, getMessage };
+}
 +export function syncCurrentSwipeInfoExtras() {
 +    if (!chat.length) {
 +        return;
 +    }
 +    const currentMessage = chat[chat.length - 1];
 +    if (currentMessage && Array.isArray(currentMessage.swipe_info) && typeof currentMessage.swipe_id === 'number') {
 +        const swipeInfo = currentMessage.swipe_info[currentMessage.swipe_id];
 +        if (swipeInfo && typeof swipeInfo === 'object') {
 +            swipeInfo.extra = structuredClone(currentMessage.extra);
 +        }
 +    }
 +}
++
 function saveImageToMessage(img, mes) {
     if (mes && img.image) {
         if (!mes.extra || typeof mes.extra !== 'object') {

         streamingProcessor.onStopStreaming();
+    }
 +    // Make sure ad-hoc changes to extras are saved before swiping away
 +    syncCurrentSwipeInfoExtras();
++
     const swipe_duration = 120;
     const swipe_range = '700px';
     chat[chat.length - 1]['swipe_id']--;

         return unblockGeneration();
+    }
 +    // Make sure ad-hoc changes to extras are saved before swiping away
 +    syncCurrentSwipeInfoExtras();
++
     const swipe_duration = 200;
     const swipe_range = 700;
     //console.log(swipe_range);

public/scripts/slash-commands.js+4 -0 Viewed

     showMoreMessages,
     stopGeneration,
     substituteParams,
 +    syncCurrentSwipeInfoExtras,
     system_avatar,
     system_message_types,
     this_chid,

     const newSwipeId = lastMessage.swipes.length - 1;
     if (isTrueBoolean(args.switch)) {
 +        // Make sure ad-hoc changes to extras are saved before swiping away
 +        syncCurrentSwipeInfoExtras();
         lastMessage.swipe_id = newSwipeId;
         lastMessage.mes = lastMessage.swipes[newSwipeId];
 +        lastMessage.extra = structuredClone(lastMessage.swipe_info?.[newSwipeId]?.extra ?? lastMessage.extra ?? {});
+    }
     await saveChatConditional();