feat: ditch triplit, move to postgresql + drizzle orm

2026-04-08 17:03:07 -05:00
parent c341c96798
commit e2e3ac6e86
121 changed files with 6680 additions and 4373 deletions
@@ -0,0 +1,34 @@
+import { and, eq } from "drizzle-orm";
+import { agents } from "~~/drizzle/schema";
+import { db } from "~~/server/lib/db";
+import { userEvents } from "~~/server/utils/events";
+
+export default defineEventHandler(async (event) => {
+    await protectRoute(event);
+    const userId = event.context.user!.id;
+
+    const agentId = getRouterParam(event, 'id')!;
+
+    const res = await db.delete(agents).where(
+        and(
+            eq(agents.id, agentId),
+            eq(agents.userId, userId),
+        )
+    );
+
+    if (res.rowCount === 0) {
+        throw createError({
+            statusCode: 400,
+            statusMessage: 'Invalid agent',
+        });
+    }
+
+    userEvents.emit(userId, 'agents', {
+        op: 'delete',
+        payload: {
+            id: agentId,
+        },
+    });
+
+    return { ok: true };
+});
@@ -0,0 +1,66 @@
+import { and, eq } from 'drizzle-orm';
+import * as z from 'zod';
+import { agents } from '~~/drizzle/schema';
+import { db } from '~~/server/lib/db';
+import { userEvents } from '~~/server/utils/events';
+
+export default defineEventHandler(async (event) => {
+    await protectRoute(event);
+    const userId = event.context.user!.id;
+
+    const result = await readValidatedBody(event, (body) =>
+        z
+            .object({
+                name: z.string().optional(),
+                systemPrompt: z.string().nullable().optional(),
+                imageUrl: z.string().optional(),
+                defaultModelId: z.string().optional(),
+            })
+            .safeParse(body),
+    );
+
+    if (!result.success) {
+        throw createError({
+            statusCode: 400,
+            message: result.error.issues[0]!.message,
+        });
+    }
+
+    const agentId = getRouterParam(event, 'id')!;
+
+    const { name, systemPrompt, imageUrl, defaultModelId } = result.data;
+
+    const res = await db.update(agents)
+        .set({
+            name,
+            systemPrompt,
+            imageUrl,
+            defaultModelId,
+        })
+        .where(
+            and(
+                eq(agents.id, agentId),
+                eq(agents.userId, userId),
+            )
+        );
+
+    if (res.rowCount === 0) {
+        throw createError({
+            statusCode: 400,
+            statusMessage: 'Invalid agent',
+        });
+    }
+
+    userEvents.emit(userId, 'agents', {
+        op: 'update',
+        payload: {
+            id: agentId,
+            name,
+            systemPrompt,
+            imageUrl,
+            defaultModelId,
+        },
+    });
+
+    return { ok: true };
+});
@@ -0,0 +1,44 @@
+import { createInsertSchema } from 'drizzle-orm/zod';
+import { db } from '~~/server/lib/db';
+import { agents } from '~~/drizzle/schema';
+import { userEvents } from '~~/server/utils/events';
+
+export default defineEventHandler(async (event) => {
+    await protectRoute(event);
+
+    const userId = event.context.user!.id;
+
+    const result = await readValidatedBody(event,
+        createInsertSchema(agents).safeParse
+    );
+
+    if (!result.success) {
+        throw createError({
+            statusCode: 400,
+            message: result.error.issues[0]!.message,
+        });
+    }
+
+    if (!result.data.userId) {
+        result.data.userId = userId;
+    }
+
+    if (result.data.userId !== userId) {
+        throw createError({
+            statusCode: 403,
+            statusMessage: 'Forbidden',
+            data: {
+                code: 'FORBIDDEN',
+                ok: false,
+            }
+        });
+    }
+
+    await db.insert(agents).values(result.data).onConflictDoNothing();
+    userEvents.emit(userId, 'agents', {
+        op: 'create',
+        payload: result.data,
+    });
+
+    return { ok: true };
+})