import { DrizzleClient } from "@lib/DrizzleClient";
import { users } from "@/db/schema";
import { eq } from "drizzle-orm";

export async function getUserBalance(userId: string) {
    const user = await DrizzleClient.query.users.findFirst({ where: eq(users.userId, userId) });
    return user?.balance ?? 0;
}

export async function setUserBalance(userId: string, balance: number) {
    await DrizzleClient.update(users).set({ balance }).where(eq(users.userId, userId));
}