From afbe93200293c722d806406790b4271048c115a6 Mon Sep 17 00:00:00 2001
From: Webifi <john@webifi.com>
Date: Mon, 29 May 2023 08:37:38 -0500
Subject: [PATCH] Allow deletion of last summarization

---
 src/lib/EditMessage.svelte | 57 ++++++++++++++++++++++++++++----------
 src/lib/Storage.svelte     | 25 +++++++++++++++++
 2 files changed, 67 insertions(+), 15 deletions(-)

diff --git a/src/lib/EditMessage.svelte b/src/lib/EditMessage.svelte
index 4b6554d..029ba5f 100644
--- a/src/lib/EditMessage.svelte
+++ b/src/lib/EditMessage.svelte
@@ -1,7 +1,7 @@
 <script lang="ts">
   import Code from './Code.svelte'
   import { createEventDispatcher, onMount } from 'svelte'
-  import { deleteMessage, chatsStorage } from './Storage.svelte'
+  import { deleteMessage, chatsStorage, deleteSummaryMessage } from './Storage.svelte'
   import { getPrice } from './Stats.svelte'
   import SvelteMarkdown from 'svelte-markdown'
   import type { Message, Model, Chat } from './Types.svelte'
@@ -43,6 +43,31 @@
     }, 0)
   }
 
+  const checkDelete = () => {
+    if (message.summarized) {
+      // is in a summary, so we're summarized
+      window.alert(`Sorry, you can't delete a summarized message`)
+      return
+    }
+    if (message.summary) {
+      // We're linked to messages we're a summary of
+      if (window.confirm("Are you sure you want to delete this summary?\nYour session may be too long to submit again after you do.")) {
+        try {
+          deleteSummaryMessage(chatId, message.uuid)
+        } catch(e) {
+          alert('Unable to delete summary:\n'+e.message)
+        }
+      }
+      return
+    }
+    try {
+      deleteMessage(chatId, message.uuid)
+    } catch(e) {
+      alert('Unable to delete:\n'+e.message)
+    }
+    
+  }
+
   let dbnc
   const update = () => {
     clearTimeout(dbnc)
@@ -67,6 +92,7 @@
       editing = false
     }
   }
+
   const scrollToMessage = (uuid:string | string[] | undefined) => {
     if (Array.isArray(uuid)) {
       uuid = uuid[0]
@@ -110,21 +136,10 @@
 >
   <div class="message-body content">
     <div class="greyscale is-pulled-right ml-2 button-pack">
-    {#if !message.summarized && !message.summary}
+    {#if message.summarized}
     <a
       href={'#'}
-      class=" delButton"
-      on:click|preventDefault={() => {
-        // messages.splice(i, 1)
-        deleteMessage(chatId, message.uuid)
-      }}
-    >
-    <Fa icon={faTrash} />
-    </a>
-    {:else if message.summarized}
-    <a
-      href={'#'}
-      class="delButton"
+      class="msg-summary-button"
       on:click|preventDefault={() => {
         scrollToMessage(message.summarized)
       }}
@@ -135,7 +150,7 @@
     {#if message.summary}
     <a
       href={'#'}
-      class="delButton"
+      class="msg-summarized-button"
       on:click|preventDefault={() => {
         scrollToMessage(message.summary)
       }}
@@ -143,6 +158,18 @@
     <Fa icon={faDiagramPredecessor} />
     </a>
     {/if}
+    {#if !message.summarized}
+    <a
+      href={'#'}
+      class=" msg-delete-button"
+      on:click|preventDefault={() => {
+        // messages.splice(i, 1)
+        checkDelete()
+      }}
+    >
+    <Fa icon={faTrash} />
+    </a>
+    {/if}
     </div>
     {#if editing && !noEdit}
       <form class="message-edit" on:submit|preventDefault={update} on:keydown={keydown}>
diff --git a/src/lib/Storage.svelte b/src/lib/Storage.svelte
index d46b65a..4dbb896 100644
--- a/src/lib/Storage.svelte
+++ b/src/lib/Storage.svelte
@@ -166,6 +166,10 @@
     return chat.messages
   }
 
+  const getMessage = (chat: Chat, uuid:string):Message|undefined => {
+    return chat.messages.find((m) => m.uuid === uuid)
+  }
+
   export const insertMessages = (chatId: number, insertAfter: Message, newMessages: Message[]) => {
     const chats = get(chatsStorage)
     const chat = chats.find((chat) => chat.id === chatId) as Chat
@@ -178,10 +182,31 @@
     chatsStorage.set(chats)
   }
 
+  export const deleteSummaryMessage = (chatId: number, uuid: string) => {
+    const chats = get(chatsStorage)
+    const chat = chats.find((chat) => chat.id === chatId) as Chat
+    const message = getMessage(chat, uuid)
+    if (message && message.summarized) throw new Error('Unable to delete summarized message')
+    if (message && message.summary) { // messages we summarized
+      message.summary.forEach(sid=>{
+        const m = getMessage(chat, sid)
+        if (m) {
+          delete m.summarized // unbind to this summary
+        }
+      })
+      delete message.summary
+    }
+    chatsStorage.set(chats)
+    deleteMessage(chatId, uuid)
+  }
+
   export const deleteMessage = (chatId: number, uuid: string) => {
     const chats = get(chatsStorage)
     const chat = chats.find((chat) => chat.id === chatId) as Chat
     const index = chat.messages.findIndex((m) => m.uuid === uuid)
+    const message = getMessage(chat, uuid)
+    if (message && message.summarized) throw new Error('Unable to delete summarized message')
+    if (message && message.summary) throw new Error('Unable to directly delete message summary')
     // const found = chat.messages.filter((m) => m.uuid === uuid)
     if (index < 0) {
       console.error(`Unable to find and delete message with ID: ${uuid}`)