Undo chat history (#74)

src/hooks/useStreamChat.ts

@@ -13,7 +13,7 @@ import type { ChatResponseEnd } from "@/ipc/ipc_types";
 import { useChats } from "./useChats";
 import { useLoadApp } from "./useLoadApp";
 import { selectedAppIdAtom } from "@/atoms/appAtoms";
-import { useLoadVersions } from "./useLoadVersions";
+import { useVersions } from "./useVersions";
 import { showError } from "@/lib/toast";
 import { useProposal } from "./useProposal";
 import { useSearch } from "@tanstack/react-router";
@@ -35,7 +35,7 @@ export function useStreamChat({
   const { refreshChats } = useChats(selectedAppId);
   const { refreshApp } = useLoadApp(selectedAppId);
   const setStreamCount = useSetAtom(chatStreamCountAtom);
-  const { refreshVersions } = useLoadVersions(selectedAppId);
+  const { refreshVersions } = useVersions(selectedAppId);
   const { refreshAppIframe } = useRunApp();
   const { countTokens } = useCountTokens();
 

src/hooks/useVersions.ts

@@ -1,13 +1,16 @@
 import { useState, useEffect, useCallback } from "react";
-import { useAtom } from "jotai";
+import { useAtom, useAtomValue } from "jotai";
 import { versionsListAtom } from "@/atoms/appAtoms";
 import { IpcClient } from "@/ipc/ipc_client";
+import { showError } from "@/lib/toast";
+import { chatMessagesAtom, selectedChatIdAtom } from "@/atoms/chatAtoms";
 
-export function useLoadVersions(appId: number | null) {
+export function useVersions(appId: number | null) {
   const [versions, setVersions] = useAtom(versionsListAtom);
   const [loading, setLoading] = useState(true);
   const [error, setError] = useState<Error | null>(null);
-
+  const selectedChatId = useAtomValue(selectedChatIdAtom);
+  const [messages, setMessages] = useAtom(chatMessagesAtom);
   useEffect(() => {
     const loadVersions = async () => {
       // If no app is selected, clear versions and return
@@ -50,5 +53,26 @@ export function useLoadVersions(appId: number | null) {
     }
   }, [appId, setVersions, setError]);
 
-  return { versions, loading, error, refreshVersions };
+  const revertVersion = useCallback(
+    async ({ versionId }: { versionId: string }) => {
+      if (appId === null) {
+        return;
+      }
+
+      try {
+        const ipcClient = IpcClient.getInstance();
+        await ipcClient.revertVersion({ appId, previousVersionId: versionId });
+        await refreshVersions();
+        if (selectedChatId) {
+          const chat = await IpcClient.getInstance().getChat(selectedChatId);
+          setMessages(chat.messages);
+        }
+      } catch (error) {
+        showError(error);
+      }
+    },
+    [appId, setVersions, setError, selectedChatId, setMessages]
+  );
+
+  return { versions, loading, error, refreshVersions, revertVersion };
 }