create db update message function

api/db.go

@@ -150,6 +150,54 @@ func dbAddMessage(message *Message) error {
 	return db.WriteDB("messages", messages)
 }
 
+func dbUpdateMessage(updatedMessage *Message) error {
+	currentData := db.ExecDB("messages")
+	if currentData == nil {
+		return fmt.Errorf("error reading messages database")
+	}
+
+	messages, ok := currentData["messages"].([]interface{})
+	if !ok {
+		return fmt.Errorf("messages data is in an unexpected format")
+	}
+
+	var updatedMessages []interface{}
+	found := false
+
+	for _, m := range messages {
+		message, ok := m.(map[string]interface{})
+		if !ok {
+			continue
+		}
+
+		if messageID, ok := message["ID"].(string); ok && messageID == updatedMessage.ID {
+			found = true
+
+			var edited interface{}
+			if updatedMessage.Edited.IsZero() {
+				edited = nil // Set to nil if Edited is the zero value
+			} else {
+				edited = updatedMessage.Edited.Format(time.RFC3339)
+			}
+
+			message = map[string]interface{}{
+				"ID":        updatedMessage.ID,
+				"UserID":    updatedMessage.UserID,
+				"Body":      updatedMessage.Body,
+				"Timestamp": updatedMessage.Timestamp.Format(time.RFC3339),
+				"Edited":    edited,
+			}
+		}
+		updatedMessages = append(updatedMessages, message)
+	}
+
+	if !found {
+		return fmt.Errorf("message with ID %s not found", updatedMessage.ID)
+	}
+
+	return db.WriteDB("messages", updatedMessages)
+}
+
 func dbDeleteMessage(id string) error {
 	currentData := db.ExecDB("messages")
 	if currentData == nil {

api/message.go

@@ -64,11 +64,19 @@ func EditMessage(w http.ResponseWriter, r *http.Request) {
 		http.Error(w, "Message body cannot be empty", http.StatusBadRequest)
 		return
 	}
+
 	message.Body = body
-	dbDeleteMessage(message.ID)
 	message.Edited = time.Now()
-	err = dbAddMessage(message)
+
+	err = dbUpdateMessage(message)
 	if err != nil {
+		render.Render(w, r, ErrRender(err))
+		return
+	}
+
+	if err := render.Render(w, r, NewMessageResponse(message)); err != nil {
+		render.Render(w, r, ErrRender(err))
+		return
 	}
 }