Fix undo and redo by using initial commit hash for chat (#94)

drizzle/0004_flawless_jigsaw.sql

+ALTER TABLE `chats` ADD `initial_commit_hash` text;
\ No newline at end of file

drizzle/meta/0004_snapshot.json

+{
+  "version": "6",
+  "dialect": "sqlite",
+  "id": "ceedb797-6aa3-4a50-b42f-bc85ee08b3df",
+  "prevId": "859942b1-88b8-4a16-b2d0-77c9ece76693",
+  "tables": {
+    "apps": {
+      "name": "apps",
+      "columns": {
+        "id": {
+          "name": "id",
+          "type": "integer",
+          "primaryKey": true,
+          "notNull": true,
+          "autoincrement": true
+        },
+        "name": {
+          "name": "name",
+          "type": "text",
+          "primaryKey": false,
+          "notNull": true,
+          "autoincrement": false
+        },
+        "path": {
+          "name": "path",
+          "type": "text",
+          "primaryKey": false,
+          "notNull": true,
+          "autoincrement": false
+        },
+        "created_at": {
+          "name": "created_at",
+          "type": "integer",
+          "primaryKey": false,
+          "notNull": true,
+          "autoincrement": false,
+          "default": "(unixepoch())"
+        },
+        "updated_at": {
+          "name": "updated_at",
+          "type": "integer",
+          "primaryKey": false,
+          "notNull": true,
+          "autoincrement": false,
+          "default": "(unixepoch())"
+        },
+        "github_org": {
+          "name": "github_org",
+          "type": "text",
+          "primaryKey": false,
+          "notNull": false,
+          "autoincrement": false
+        },
+        "github_repo": {
+          "name": "github_repo",
+          "type": "text",
+          "primaryKey": false,
+          "notNull": false,
+          "autoincrement": false
+        },
+        "supabase_project_id": {
+          "name": "supabase_project_id",
+          "type": "text",
+          "primaryKey": false,
+          "notNull": false,
+          "autoincrement": false
+        }
+      },
+      "indexes": {},
+      "foreignKeys": {},
+      "compositePrimaryKeys": {},
+      "uniqueConstraints": {},
+      "checkConstraints": {}
+    },
+    "chats": {
+      "name": "chats",
+      "columns": {
+        "id": {
+          "name": "id",
+          "type": "integer",
+          "primaryKey": true,
+          "notNull": true,
+          "autoincrement": true
+        },
+        "app_id": {
+          "name": "app_id",
+          "type": "integer",
+          "primaryKey": false,
+          "notNull": true,
+          "autoincrement": false
+        },
+        "title": {
+          "name": "title",
+          "type": "text",
+          "primaryKey": false,
+          "notNull": false,
+          "autoincrement": false
+        },
+        "initial_commit_hash": {
+          "name": "initial_commit_hash",
+          "type": "text",
+          "primaryKey": false,
+          "notNull": false,
+          "autoincrement": false
+        },
+        "created_at": {
+          "name": "created_at",
+          "type": "integer",
+          "primaryKey": false,
+          "notNull": true,
+          "autoincrement": false,
+          "default": "(unixepoch())"
+        }
+      },
+      "indexes": {},
+      "foreignKeys": {
+        "chats_app_id_apps_id_fk": {
+          "name": "chats_app_id_apps_id_fk",
+          "tableFrom": "chats",
+          "tableTo": "apps",
+          "columnsFrom": [
+            "app_id"
+          ],
+          "columnsTo": [
+            "id"
+          ],
+          "onDelete": "cascade",
+          "onUpdate": "no action"
+        }
+      },
+      "compositePrimaryKeys": {},
+      "uniqueConstraints": {},
+      "checkConstraints": {}
+    },
+    "messages": {
+      "name": "messages",
+      "columns": {
+        "id": {
+          "name": "id",
+          "type": "integer",
+          "primaryKey": true,
+          "notNull": true,
+          "autoincrement": true
+        },
+        "chat_id": {
+          "name": "chat_id",
+          "type": "integer",
+          "primaryKey": false,
+          "notNull": true,
+          "autoincrement": false
+        },
+        "role": {
+          "name": "role",
+          "type": "text",
+          "primaryKey": false,
+          "notNull": true,
+          "autoincrement": false
+        },
+        "content": {
+          "name": "content",
+          "type": "text",
+          "primaryKey": false,
+          "notNull": true,
+          "autoincrement": false
+        },
+        "approval_state": {
+          "name": "approval_state",
+          "type": "text",
+          "primaryKey": false,
+          "notNull": false,
+          "autoincrement": false
+        },
+        "commit_hash": {
+          "name": "commit_hash",
+          "type": "text",
+          "primaryKey": false,
+          "notNull": false,
+          "autoincrement": false
+        },
+        "created_at": {
+          "name": "created_at",
+          "type": "integer",
+          "primaryKey": false,
+          "notNull": true,
+          "autoincrement": false,
+          "default": "(unixepoch())"
+        }
+      },
+      "indexes": {},
+      "foreignKeys": {
+        "messages_chat_id_chats_id_fk": {
+          "name": "messages_chat_id_chats_id_fk",
+          "tableFrom": "messages",
+          "tableTo": "chats",
+          "columnsFrom": [
+            "chat_id"
+          ],
+          "columnsTo": [
+            "id"
+          ],
+          "onDelete": "cascade",
+          "onUpdate": "no action"
+        }
+      },
+      "compositePrimaryKeys": {},
+      "uniqueConstraints": {},
+      "checkConstraints": {}
+    }
+  },
+  "views": {},
+  "enums": {},
+  "_meta": {
+    "schemas": {},
+    "tables": {},
+    "columns": {}
+  },
+  "internal": {
+    "indexes": {}
+  }
+}
\ No newline at end of file

drizzle/meta/_journal.json

@@ -29,6 +29,13 @@
       "when": 1746209201530,
       "tag": "0003_open_bucky",
       "breakpoints": true
+    },
+    {
+      "idx": 4,
+      "version": "6",
+      "when": 1746556241557,
+      "tag": "0004_flawless_jigsaw",
+      "breakpoints": true
     }
   ]
 }
\ No newline at end of file

src/db/schema.ts

@@ -23,6 +23,7 @@ export const chats = sqliteTable("chats", {
     .notNull()
     .references(() => apps.id, { onDelete: "cascade" }),
   title: text("title"),
+  initialCommitHash: text("initial_commit_hash"),
   createdAt: integer("created_at", { mode: "timestamp" })
     .notNull()
     .default(sql`(unixepoch())`),

src/ipc/handlers/app_handlers.ts

@@ -189,12 +189,20 @@ export function registerAppHandlers() {
       });
 
       // Create initial commit
-      await git.commit({
+      const commitHash = await git.commit({
         fs: fs,
         dir: fullAppPath,
         message: "Init from react vite template",
         author: await getGitAuthor(),
       });
+
+      // Update chat with initial commit hash
+      await db
+        .update(chats)
+        .set({
+          initialCommitHash: commitHash,
+        })
+        .where(eq(chats.id, chat.id));
     } catch (error) {
       logger.error("Error in background app initialization:", error);
     }

src/ipc/handlers/chat_handlers.ts

 import { ipcMain } from "electron";
 import { db } from "../../db";
-import { chats } from "../../db/schema";
+import { apps, chats, messages } from "../../db/schema";
 import { desc, eq } from "drizzle-orm";
 import type { ChatSummary } from "../../lib/schemas";
+import * as git from "isomorphic-git";
+import * as fs from "fs";
+import * as path from "path";
+import log from "electron-log";
+import { getDyadAppPath } from "../../paths/paths";
+
+const logger = log.scope("chat_handlers");
 
 export function registerChatHandlers() {
   ipcMain.handle("create-chat", async (_, appId: number) => {
+    // Get the app's path first
+    const app = await db.query.apps.findFirst({
+      where: eq(apps.id, appId),
+      columns: {
+        path: true,
+      },
+    });
+
+    if (!app) {
+      throw new Error("App not found");
+    }
+
+    let initialCommitHash = null;
+    try {
+      // Get the current git revision of main branch
+      initialCommitHash = await git.resolveRef({
+        fs,
+        dir: getDyadAppPath(app.path),
+        ref: "main",
+      });
+    } catch (error) {
+      logger.error("Error getting git revision:", error);
+      // Continue without the git revision
+    }
+
     // Create a new chat
     const [chat] = await db
       .insert(chats)
       .values({
         appId,
+        initialCommitHash,
       })
       .returning();
-
+    logger.info(
+      "Created chat:",
+      chat.id,
+      "for app:",
+      appId,
+      "with initial commit hash:",
+      initialCommitHash
+    );
     return chat.id;
   });
 
@@ -70,7 +110,17 @@ export function registerChatHandlers() {
       await db.delete(chats).where(eq(chats.id, chatId));
       return { success: true };
     } catch (error) {
-      console.error("Error deleting chat:", error);
+      logger.error("Error deleting chat:", error);
+      return { success: false, error: (error as Error).message };
+    }
+  });
+
+  ipcMain.handle("delete-messages", async (_, chatId: number) => {
+    try {
+      await db.delete(messages).where(eq(messages.chatId, chatId));
+      return { success: true };
+    } catch (error) {
+      logger.error("Error deleting messages:", error);
       return { success: false, error: (error as Error).message };
     }
   });

src/ipc/ipc_client.ts

@@ -330,12 +330,24 @@ export class IpcClient {
     }
   }
 
-  public async deleteChat(chatId: number): Promise<{ success: boolean }> {
+  public async deleteChat(
+    chatId: number
+  ): Promise<{ success: boolean; error?: string }> {
     try {
-      const result = (await this.ipcRenderer.invoke("delete-chat", chatId)) as {
-        success: boolean;
-      };
-      return result;
+      const result = await this.ipcRenderer.invoke("delete-chat", chatId);
+      return result as { success: boolean; error?: string };
+    } catch (error) {
+      showError(error);
+      throw error;
+    }
+  }
+
+  public async deleteMessages(
+    chatId: number
+  ): Promise<{ success: boolean; error?: string }> {
+    try {
+      const result = await this.ipcRenderer.invoke("delete-messages", chatId);
+      return result as { success: boolean; error?: string };
     } catch (error) {
       showError(error);
       throw error;

src/ipc/ipc_types.ts

@@ -54,6 +54,7 @@ export interface Chat {
   id: number;
   title: string;
   messages: Message[];
+  initialCommitHash?: string | null;
 }
 
 export interface App {

src/preload.ts

@@ -58,6 +58,8 @@ const validInvokeChannels = [
   "window:get-platform",
   "upload-to-signed-url",
   "delete-chat",
+  "delete-messages",
+  "start-chat-stream",
 ] as const;
 
 // Add valid receive channels