feat: Make it so users can edit queued messages (up arrow) (#742)

Author: charlesvien

apps/twig/src/renderer/features/message-editor/tiptap/useTiptapEditor.ts

@@ -1,3 +1,4 @@
+import { getSessionActions } from "@features/sessions/stores/sessionStore";
 import { trpcVanilla } from "@renderer/trpc/client";
 import { toast } from "@renderer/utils/toast";
 import { useSettingsStore } from "@stores/settingsStore";
@@ -143,6 +144,18 @@ export function useTiptapEditor(options: UseTiptapEditorOptions) {
             const isAtEnd = from === view.state.doc.content.size - 1;
 
             if (event.key === "ArrowUp" && (isEmpty || isAtStart)) {
+              const queuedContent =
+                getSessionActions().popQueuedMessagesAsText(taskId);
+              if (queuedContent !== null && queuedContent !== undefined) {
+                event.preventDefault();
+                view.dispatch(
+                  view.state.tr
+                    .delete(1, view.state.doc.content.size - 1)
+                    .insertText(queuedContent, 1),
+                );
+                return true;
+              }
+
               const newText = historyActions.navigateUp(taskId, currentText);
               if (newText !== null) {
                 event.preventDefault();

apps/twig/src/renderer/features/sessions/stores/sessionStore.ts

@@ -127,6 +127,7 @@ interface SessionActions {
   clearSessionError: (taskId: string) => Promise<void>;
   removeQueuedMessage: (taskId: string, queueId: string) => void;
   popAllQueuedMessages: (taskId: string) => QueuedMessage[];
+  popQueuedMessagesAsText: (taskId: string) => string | null;
 }
 
 interface AuthCredentials {
@@ -1402,6 +1403,14 @@ const useStore = create<SessionStore>()(
 
           return messages;
         },
+
+        popQueuedMessagesAsText: (taskId: string): string | null => {
+          const messages = useStore
+            .getState()
+            .actions.popAllQueuedMessages(taskId);
+          if (messages.length === 0) return null;
+          return messages.map((msg) => msg.content).join("\n\n");
+        },
       },
     };
   }),