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

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

src/actions/bulk-users.test.ts

@@ -1,7 +1,6 @@
 // @ts-nocheck
 import { beforeEach, describe, expect, it, vi } from "vitest";
 import { requirePermission } from "@/lib/admin/guard";
-import { prisma } from "@/lib/prisma";
 import { rcon } from "@/lib/services/rcon";
 import {
 	bulkBan,
@@ -10,14 +9,78 @@ import {
 	bulkUnban,
 } from "./bulk-users";
 
+const {
+	deleteWhere,
+	insertValues,
+	updateWhere,
+	selectLimit,
+	selectWhereResolved,
+	onDuplicateKeyUpdate,
+} = vi.hoisted(() => {
+	const deleteWhere = vi.fn().mockResolvedValue([{ affectedRows: 3 }]);
+	const onDuplicateKeyUpdate = vi.fn().mockResolvedValue([{ affectedRows: 1 }]);
+	const insertValues = vi.fn(() => ({
+		onDuplicateKeyUpdate,
+		// biome-ignore lint/suspicious/noThenProperty: Drizzle query thenable mock
+		then(resolve, reject) {
+			return Promise.resolve([{ insertId: 1 }]).then(resolve, reject);
+		},
+	}));
+	const updateWhere = vi.fn().mockResolvedValue([{ affectedRows: 1 }]);
+	const selectLimit = vi.fn().mockResolvedValue([]);
+	/** Rows returned when a select chain is awaited without `.limit()`. */
+	const selectWhereResolved = vi.fn().mockResolvedValue([]);
+	return {
+		deleteWhere,
+		insertValues,
+		updateWhere,
+		selectLimit,
+		selectWhereResolved,
+		onDuplicateKeyUpdate,
+	};
+});
+
 vi.mock("@/lib/admin/guard", () => ({ requirePermission: vi.fn() }));
 vi.mock("@/lib/permissions", () => ({ PERMS: { USERS_EDIT: "users.edit" } }));
-vi.mock("@/lib/prisma", () => ({
-	prisma: {
-		ban: { deleteMany: vi.fn(), create: vi.fn() },
-		user: { update: vi.fn() },
-		usersCurrency: { upsert: vi.fn() },
-		usersBadges: { findFirst: vi.fn(), aggregate: vi.fn(), create: vi.fn() },
+vi.mock("@/lib/db", () => ({
+	db: {
+		delete: vi.fn(() => ({ where: deleteWhere })),
+		insert: vi.fn(() => ({ values: insertValues })),
+		update: vi.fn(() => ({
+			set: vi.fn(() => ({ where: updateWhere })),
+		})),
+		select: vi.fn(() => ({
+			from: vi.fn(() => ({
+				where: vi.fn(() => ({
+					limit: selectLimit,
+					// biome-ignore lint/suspicious/noThenProperty: Drizzle query thenable mock
+					then(resolve, reject) {
+						return selectWhereResolved().then(resolve, reject);
+					},
+				})),
+			})),
+		})),
+		transaction: vi.fn(),
+	},
+	Ban: { userId: "userId", id: "id" },
+	User: {
+		id: "id",
+		credits: "credits",
+		username: "username",
+		online: "online",
+	},
+	UsersCurrency: { userId: "userId", type: "type", amount: "amount" },
+	UsersBadges: {
+		id: "id",
+		userId: "userId",
+		badgeCode: "badgeCode",
+		slotId: "slotId",
+	},
+	Sanctions: { id: "id", habboId: "habboId" },
+	UsersSettings: {
+		userId: "userId",
+		canTrade: "canTrade",
+		tradelockAmount: "tradelockAmount",
 	},
 }));
 vi.mock("@/lib/services/rcon", () => ({
@@ -35,11 +98,22 @@ const staff = { id: 1, rank: 7, username: "admin" };
 beforeEach(() => {
 	vi.clearAllMocks();
 	vi.mocked(requirePermission).mockResolvedValue(staff as never);
+	deleteWhere.mockResolvedValue([{ affectedRows: 3 }]);
+	insertValues.mockImplementation(() => ({
+		onDuplicateKeyUpdate,
+		// biome-ignore lint/suspicious/noThenProperty: Drizzle query thenable mock
+		then(resolve, reject) {
+			return Promise.resolve([{ insertId: 1 }]).then(resolve, reject);
+		},
+	}));
+	onDuplicateKeyUpdate.mockResolvedValue([{ affectedRows: 1 }]);
+	updateWhere.mockResolvedValue([{ affectedRows: 1 }]);
+	selectLimit.mockResolvedValue([]);
+	selectWhereResolved.mockResolvedValue([]);
 });
 
 describe("bulkUnban", () => {
 	it("unbans users", async () => {
-		vi.mocked(prisma.ban.deleteMany).mockResolvedValue({ count: 3 } as never);
 		const r = await bulkUnban({ userIds: [1, 2, 3] });
 		expect(r.ok).toBe(true);
 		expect(r.data).toEqual({ unbanned: 3, total: 3 });
@@ -48,7 +122,6 @@ describe("bulkUnban", () => {
 
 describe("bulkBan", () => {
 	it("bans users", async () => {
-		vi.mocked(prisma.ban.create).mockResolvedValue({} as never);
 		const r = await bulkBan({
 			userIds: [1, 2],
 			reason: "Spam",
@@ -56,12 +129,12 @@ describe("bulkBan", () => {
 		});
 		expect(r.ok).toBe(true);
 		expect(r.data.banned).toBe(2);
+		expect(insertValues).toHaveBeenCalledTimes(2);
 	});
 });
 
 describe("bulkGiveCurrency", () => {
 	it("gives credits", async () => {
-		vi.mocked(prisma.user.update).mockResolvedValue({} as never);
 		const r = await bulkGiveCurrency({
 			userIds: [1],
 			amount: 100,
@@ -69,10 +142,10 @@ describe("bulkGiveCurrency", () => {
 		});
 		expect(r.data.given).toBe(1);
 		expect(rcon.giveCredits).toHaveBeenCalledWith(1, 100);
+		expect(updateWhere).toHaveBeenCalled();
 	});
 
 	it("gives pixels", async () => {
-		vi.mocked(prisma.usersCurrency.upsert).mockResolvedValue({} as never);
 		const r = await bulkGiveCurrency({
 			userIds: [2],
 			amount: 50,
@@ -80,10 +153,10 @@ describe("bulkGiveCurrency", () => {
 		});
 		expect(r.data.given).toBe(1);
 		expect(rcon.giveDuckets).toHaveBeenCalledWith(2, 50);
+		expect(onDuplicateKeyUpdate).toHaveBeenCalled();
 	});
 
 	it("gives points", async () => {
-		vi.mocked(prisma.usersCurrency.upsert).mockResolvedValue({} as never);
 		const r = await bulkGiveCurrency({
 			userIds: [3],
 			amount: 25,
@@ -91,17 +164,17 @@ describe("bulkGiveCurrency", () => {
 		});
 		expect(r.data.given).toBe(1);
 		expect(rcon.givePointsGotw).toHaveBeenCalledWith(3, 25);
+		expect(onDuplicateKeyUpdate).toHaveBeenCalled();
 	});
 });
 
 describe("bulkGiveBadge", () => {
 	it("gives badge to user", async () => {
-		vi.mocked(prisma.usersBadges.findFirst).mockResolvedValue(null);
-		vi.mocked(prisma.usersBadges.aggregate).mockResolvedValue({
-			_max: { slotId: 5 },
-		} as never);
-		vi.mocked(prisma.usersBadges.create).mockResolvedValue({} as never);
+		selectLimit.mockResolvedValueOnce([]);
+		selectWhereResolved.mockResolvedValueOnce([{ maxSlot: 5 }]);
 		const r = await bulkGiveBadge({ userIds: [1], badgeCode: "ADM" });
 		expect(r.data.given).toBe(1);
+		expect(insertValues).toHaveBeenCalled();
+		expect(rcon.giveBadge).toHaveBeenCalledWith(1, "ADM");
 	});
 });