113 lines
2 KiB
TypeScript
113 lines
2 KiB
TypeScript
|
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));
|
||
|
|
}
|