refactor(db): migrate staff and app actions from Prisma facade to Drizzle

Co-authored-by: Cursor <[email protected]>

src/actions/admin-ip.test.ts

@@ -2,7 +2,6 @@
 import { revalidatePath } from "next/cache";
 import { beforeEach, describe, expect, it, vi } from "vitest";
 import { requirePermission } from "@/lib/admin/guard";
-import { prisma } from "@/lib/prisma";
 import {
 	addBlacklist,
 	addWhitelist,
@@ -10,15 +9,23 @@ import {
 	deleteWhitelist,
 } from "./admin-ip";
 
+const { insertValues, deleteWhere } = vi.hoisted(() => {
+	const insertValues = vi.fn().mockResolvedValue([{ insertId: 1 }]);
+	const deleteWhere = vi.fn().mockResolvedValue([{ affectedRows: 1 }]);
+	return { insertValues, deleteWhere };
+});
+
 vi.mock("@/lib/admin/guard", () => ({ requirePermission: vi.fn() }));
 vi.mock("@/lib/permissions", () => ({
 	PERMS: { SETTINGS_EDIT: "settings.edit" },
 }));
-vi.mock("@/lib/prisma", () => ({
-	prisma: {
-		websiteIpWhitelist: { create: vi.fn(), delete: vi.fn() },
-		websiteIpBlacklist: { create: vi.fn(), delete: vi.fn() },
+vi.mock("@/lib/db", () => ({
+	db: {
+		insert: vi.fn(() => ({ values: insertValues })),
+		delete: vi.fn(() => ({ where: deleteWhere })),
 	},
+	WebsiteIpWhitelist: { id: "id" },
+	WebsiteIpBlacklist: { id: "id" },
 }));
 vi.mock("next/cache", () => ({ revalidatePath: vi.fn() }));
 
@@ -30,6 +37,8 @@ const fakeForm = (data: Record<string, string>) => ({
 beforeEach(() => {
 	vi.clearAllMocks();
 	vi.mocked(requirePermission).mockResolvedValue(staff as never);
+	insertValues.mockResolvedValue([{ insertId: 1 }]);
+	deleteWhere.mockResolvedValue([{ affectedRows: 1 }]);
 });
 
 describe("addWhitelist", () => {
@@ -37,24 +46,24 @@ describe("addWhitelist", () => {
 		await addWhitelist(
 			fakeForm({ ipAddress: "192.168.1.1" }) as unknown as FormData,
 		);
-		expect(prisma.websiteIpWhitelist.create).toHaveBeenCalledWith({
-			data: { ipAddress: "192.168.1.1", asn: null, whitelistAsn: false },
+		expect(insertValues).toHaveBeenCalledWith({
+			ipAddress: "192.168.1.1",
+			asn: null,
+			whitelistAsn: false,
 		});
 		expect(revalidatePath).toHaveBeenCalledWith("/admin/ip");
 	});
 
 	it("returns early when ip is empty", async () => {
 		await addWhitelist(fakeForm({ ipAddress: "" }) as unknown as FormData);
-		expect(prisma.websiteIpWhitelist.create).not.toHaveBeenCalled();
+		expect(insertValues).not.toHaveBeenCalled();
 	});
 });
 
 describe("deleteWhitelist", () => {
 	it("deletes whitelist entry", async () => {
 		await deleteWhitelist(fakeForm({ id: "42" }) as unknown as FormData);
-		expect(prisma.websiteIpWhitelist.delete).toHaveBeenCalledWith({
-			where: { id: BigInt(42) },
-		});
+		expect(deleteWhere).toHaveBeenCalled();
 	});
 });
 
@@ -63,8 +72,10 @@ describe("addBlacklist", () => {
 		await addBlacklist(
 			fakeForm({ ipAddress: "203.0.113.1" }) as unknown as FormData,
 		);
-		expect(prisma.websiteIpBlacklist.create).toHaveBeenCalledWith({
-			data: { ipAddress: "203.0.113.1", asn: null, blacklistAsn: false },
+		expect(insertValues).toHaveBeenCalledWith({
+			ipAddress: "203.0.113.1",
+			asn: null,
+			blacklistAsn: false,
 		});
 	});
 });
@@ -72,8 +83,6 @@ describe("addBlacklist", () => {
 describe("deleteBlacklist", () => {
 	it("deletes blacklist entry", async () => {
 		await deleteBlacklist(fakeForm({ id: "99" }) as unknown as FormData);
-		expect(prisma.websiteIpBlacklist.delete).toHaveBeenCalledWith({
-			where: { id: BigInt(99) },
-		});
+		expect(deleteWhere).toHaveBeenCalled();
 	});
 });