119 lines
3.0 KiB
TypeScript
119 lines
3.0 KiB
TypeScript
import { ipcMain } from "electron";
|
|
import { db } from "../../db";
|
|
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 { createLoggedHandler } from "./safe_handle";
|
|
|
|
import log from "electron-log";
|
|
import { getDyadAppPath } from "../../paths/paths";
|
|
import { UpdateChatParams } from "../ipc_types";
|
|
|
|
const logger = log.scope("chat_handlers");
|
|
const handle = createLoggedHandler(logger);
|
|
|
|
export function registerChatHandlers() {
|
|
handle("create-chat", async (_, appId: number): Promise<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;
|
|
});
|
|
|
|
ipcMain.handle("get-chat", async (_, chatId: number) => {
|
|
const chat = await db.query.chats.findFirst({
|
|
where: eq(chats.id, chatId),
|
|
with: {
|
|
messages: {
|
|
orderBy: (messages, { asc }) => [asc(messages.createdAt)],
|
|
},
|
|
},
|
|
});
|
|
|
|
if (!chat) {
|
|
throw new Error("Chat not found");
|
|
}
|
|
|
|
return chat;
|
|
});
|
|
|
|
handle("get-chats", async (_, appId?: number): Promise<ChatSummary[]> => {
|
|
// If appId is provided, filter chats for that app
|
|
const query = appId
|
|
? db.query.chats.findMany({
|
|
where: eq(chats.appId, appId),
|
|
columns: {
|
|
id: true,
|
|
title: true,
|
|
createdAt: true,
|
|
appId: true,
|
|
},
|
|
orderBy: [desc(chats.createdAt)],
|
|
})
|
|
: db.query.chats.findMany({
|
|
columns: {
|
|
id: true,
|
|
title: true,
|
|
createdAt: true,
|
|
appId: true,
|
|
},
|
|
orderBy: [desc(chats.createdAt)],
|
|
});
|
|
|
|
const allChats = await query;
|
|
return allChats;
|
|
});
|
|
|
|
handle("delete-chat", async (_, chatId: number): Promise<void> => {
|
|
await db.delete(chats).where(eq(chats.id, chatId));
|
|
});
|
|
|
|
handle("update-chat", async (_, { chatId, title }: UpdateChatParams) => {
|
|
await db.update(chats).set({ title }).where(eq(chats.id, chatId));
|
|
});
|
|
|
|
handle("delete-messages", async (_, chatId: number): Promise<void> => {
|
|
await db.delete(messages).where(eq(messages.chatId, chatId));
|
|
});
|
|
}
|