first

packages/database/drizzle/queries/users.ts

@@ -0,0 +1,112 @@
+import { eq } from "drizzle-orm";
+import type { z } from "zod";
+import { db } from "../client";
+import { account, user } from "../schema/postgres";
+import type { UserUpdateSchema } from "../zod";
+
+export async function getUsers({
+	limit,
+	offset,
+	query,
+}: {
+	limit: number;
+	offset: number;
+	query?: string;
+}) {
+	return await db.query.user.findMany({
+		where: (user, { like }) => like(user.name, `%${query}%`),
+		limit,
+		offset,
+	});
+}
+
+export async function countAllUsers() {
+	return db.$count(user);
+}
+
+export async function getUserById(id: string) {
+	return await db.query.user.findFirst({
+		where: (user, { eq }) => eq(user.id, id),
+	});
+}
+
+export async function getUserByEmail(email: string) {
+	return await db.query.user.findFirst({
+		where: (user, { eq }) => eq(user.email, email),
+	});
+}
+
+export async function createUser({
+	email,
+	name,
+	role,
+	emailVerified,
+	onboardingComplete,
+}: {
+	email: string;
+	name: string;
+	role: "admin" | "user";
+	emailVerified: boolean;
+	onboardingComplete: boolean;
+}) {
+	const [{ id }] = await db
+		.insert(user)
+		.values({
+			email,
+			name,
+			role,
+			emailVerified,
+			onboardingComplete,
+			createdAt: new Date(),
+			updatedAt: new Date(),
+		})
+		.returning({
+			id: user.id,
+		});
+
+	const newUser = await getUserById(id);
+
+	return newUser;
+}
+
+export async function getAccountById(id: string) {
+	return await db.query.account.findFirst({
+		where: (account, { eq }) => eq(account.id, id),
+	});
+}
+
+export async function createUserAccount({
+	userId,
+	providerId,
+	accountId,
+	hashedPassword,
+}: {
+	userId: string;
+	providerId: string;
+	accountId: string;
+	hashedPassword?: string;
+}) {
+	const [{ id }] = await db
+		.insert(account)
+		.values({
+			userId,
+			accountId,
+			providerId,
+			createdAt: new Date(),
+			updatedAt: new Date(),
+			password: hashedPassword,
+		})
+		.returning({
+			id: account.id,
+		});
+
+	const newAccount = await getAccountById(id);
+
+	return newAccount;
+}
+
+export async function updateUser(
+	updatedUser: z.infer<typeof UserUpdateSchema>,
+) {
+	return db.update(user).set(updatedUser).where(eq(user.id, updatedUser.id));
+}