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

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

src/actions/admin-bans.test.ts

@@ -2,17 +2,32 @@
 import { revalidatePath } from "next/cache";
 import { beforeEach, describe, expect, it, vi } from "vitest";
 import { requirePermissionRateLimited } from "@/lib/admin/guard";
-import { prisma } from "@/lib/prisma";
 import { rcon } from "@/lib/services/rcon";
 import { createBan, liftBan } from "./admin-bans";
 
+const { selectLimit, insertValues, deleteWhere } = vi.hoisted(() => {
+	const selectLimit = vi.fn();
+	const insertValues = vi.fn().mockResolvedValue([{ insertId: 1 }]);
+	const deleteWhere = vi.fn().mockResolvedValue([{ affectedRows: 1 }]);
+	return { selectLimit, insertValues, deleteWhere };
+});
+
 vi.mock("@/lib/admin/guard", () => ({ requirePermissionRateLimited: vi.fn() }));
 vi.mock("@/lib/permissions", () => ({ PERMS: { USERS_BAN: "users.ban" } }));
-vi.mock("@/lib/prisma", () => ({
-	prisma: {
-		user: { findUnique: vi.fn() },
-		ban: { create: vi.fn(), delete: vi.fn() },
+vi.mock("@/lib/db", () => ({
+	db: {
+		select: vi.fn(() => ({
+			from: vi.fn(() => ({
+				where: vi.fn(() => ({
+					limit: selectLimit,
+				})),
+			})),
+		})),
+		insert: vi.fn(() => ({ values: insertValues })),
+		delete: vi.fn(() => ({ where: deleteWhere })),
 	},
+	Ban: { id: "id", userId: "userId" },
+	User: { id: "id", username: "username" },
 }));
 vi.mock("@/lib/services/rcon", () => ({ rcon: { disconnectUser: vi.fn() } }));
 vi.mock("@/lib/services/staff-activity", () => ({ logStaffActivity: vi.fn() }));
@@ -26,13 +41,13 @@ const fakeForm = (data: Record<string, string>) => ({
 beforeEach(() => {
 	vi.clearAllMocks();
 	vi.mocked(requirePermissionRateLimited).mockResolvedValue(staff as never);
+	selectLimit.mockResolvedValue([{ username: "baduser" }]);
+	insertValues.mockResolvedValue([{ insertId: 1 }]);
+	deleteWhere.mockResolvedValue([{ affectedRows: 1 }]);
 });
 
 describe("createBan", () => {
 	it("creates a ban for valid inputs", async () => {
-		vi.mocked(prisma.user.findUnique).mockResolvedValue({
-			username: "baduser",
-		} as never);
 		await createBan(
 			fakeForm({
 				userId: "42",
@@ -41,9 +56,9 @@ describe("createBan", () => {
 				type: "account",
 			}) as unknown as FormData,
 		);
-		expect(prisma.ban.create).toHaveBeenCalledWith({
-			data: expect.objectContaining({ userId: 42, type: "account" }),
-		});
+		expect(insertValues).toHaveBeenCalledWith(
+			expect.objectContaining({ userId: 42, type: "account" }),
+		);
 		expect(rcon.disconnectUser).toHaveBeenCalledWith(42, "baduser");
 		expect(revalidatePath).toHaveBeenCalledWith("/admin/bans");
 	});
@@ -56,14 +71,14 @@ describe("createBan", () => {
 				type: "account",
 			}) as unknown as FormData,
 		);
-		expect(prisma.ban.create).not.toHaveBeenCalled();
+		expect(insertValues).not.toHaveBeenCalled();
 	});
 });
 
 describe("liftBan", () => {
 	it("deletes ban and revalidates", async () => {
 		await liftBan(fakeForm({ id: "42" }) as unknown as FormData);
-		expect(prisma.ban.delete).toHaveBeenCalledWith({ where: { id: 42 } });
+		expect(deleteWhere).toHaveBeenCalled();
 		expect(revalidatePath).toHaveBeenCalledWith("/admin/bans");
 	});
 });