From 583b93c51923b2d12b21d345d4e8b8e61051a64d Mon Sep 17 00:00:00 2001
From: Adrien <adrien.bouvais.pro@gmail.com>
Date: Thu, 23 May 2024 20:26:23 +0200
Subject: [PATCH] Working edit btn

---
 Chat.go                               | 26 ++++++++++++++++++++++++++
 main.go                               |  1 +
 views/partials/message-bot.html       |  2 +-
 views/partials/message-edit-form.html | 11 +++++++++--
 4 files changed, 37 insertions(+), 3 deletions(-)

diff --git a/Chat.go b/Chat.go
index a95f8e3..d5264a1 100644
--- a/Chat.go
+++ b/Chat.go
@@ -374,6 +374,32 @@ func RedoMessageHandler(c *fiber.Ctx) error {
 	return c.SendString(GeneratePlaceholderHTML(message.Content, selectedLLMIds, false))
 }
 
+func EditMessageHandler(c *fiber.Ctx) error {
+	messageId := c.FormValue("id")
+	message := c.FormValue("message")
+	messageUUID, _ := edgedb.ParseUUID(messageId)
+
+	var selectedLLMIds []string
+	err := json.Unmarshal([]byte(c.FormValue("selectedLLMIds")), &selectedLLMIds)
+	if err != nil {
+		// Handle the error
+		panic(err)
+	}
+
+	// Delete messages
+	err = edgeClient.Execute(edgeCtx, `
+	WITH
+		messageArea := (SELECT Message FILTER .id = <uuid>$0).area
+	DELETE Area
+		FILTER .position >= messageArea.position AND .conversation = messageArea.conversation AND .conversation.user = global currentUser;
+	`, messageUUID)
+	if err != nil {
+		panic(err)
+	}
+
+	return c.SendString(GeneratePlaceholderHTML(message, selectedLLMIds, true))
+}
+
 func ClearChatHandler(c *fiber.Ctx) error {
 	// Delete the default conversation
 	err := edgeClient.Execute(edgeCtx, `
diff --git a/main.go b/main.go
index 962f418..f622a42 100644
--- a/main.go
+++ b/main.go
@@ -75,6 +75,7 @@ func main() {
 	app.Post("/redoMessage", RedoMessageHandler)
 	app.Post("/clearChat", ClearChatHandler)
 	app.Get("/userMessage", GetUserMessageHandler)
+	app.Post("/editMessage", EditMessageHandler)
 
 	// Settings routes
 	app.Post("/addKeys", addKeys)
diff --git a/views/partials/message-bot.html b/views/partials/message-bot.html
index 96e0b12..b5cca71 100644
--- a/views/partials/message-bot.html
+++ b/views/partials/message-bot.html
@@ -1,4 +1,4 @@
-<div class="message-bot mt-3">
+<div class="message-bot mt-3" id="msg-{{ ConversationAreaId }}">
     <div class="columns is-mobile">
         <div class="column is-narrow" id="icon-column">
             <!-- Left column with the icon -->
diff --git a/views/partials/message-edit-form.html b/views/partials/message-edit-form.html
index 1fc58f8..28b1379 100644
--- a/views/partials/message-edit-form.html
+++ b/views/partials/message-edit-form.html
@@ -20,7 +20,8 @@
                                 <div class="control" style="width: 100%;">
                                     <textarea class="textarea is-small has-fixed-size mt-2"
                                         placeholder="Enter your message here" rows="{{ Rows }}"
-                                        style="background-color: transparent; width: 100%;">{{ Content }}</textarea>
+                                        style="background-color: transparent; width: 100%;"
+                                        name="message">{{ Content }}</textarea>
                                 </div>
                             </div>
                             <div class="field is-grouped is-flex is-justify-content-flex-end mb-3">
@@ -33,7 +34,13 @@
                                     </button>
                                 </div>
                                 <div class="control">
-                                    <button hx-get="/test" class="button is-success is-outlined is-small">
+                                    <div style="display: none;" hx-trigger="click from:#edit-button-{{ ID }}"
+                                        hx-target="next .message-bot" hx-swap="outerHTML" hx-get="/empty"></div>
+                                    <button class="button is-success is-outlined is-small"
+                                        hx-post="/editMessage?id={{ ID }}" hx-target="closest .message-user"
+                                        hx-include="[name='message']"
+                                        hx-vals="js:{selectedLLMIds: getSelectedModelsIDs()}" hx-swap="outerHTML"
+                                        id="edit-button-{{ ID }}" hx-trigger="click">
                                         <span class="icon">
                                             <i class="fa-solid fa-check"></i>
                                         </span>