songtianlun/prmbr-image-mksaas
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

chore: update credit related functions (2)

Browse Source
This commit is contained in:
javayhu 2025-05-29 01:21:04 +08:00
parent dae7a3b0e8
commit b30355dfe5
1 changed files with 21 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
src/lib/credits.ts
View File

@ -28,6 +28,12 @@ async function logCreditTransaction(params: {
paymentId?: string; paymentId?: string;
expirationDate?: Date; expirationDate?: Date;
}) { }) {
if (!params.userId || !params.type || !params.description) {
throw new Error('Invalid params');
}
if (!Number.isFinite(params.amount) || params.amount === 0) {
throw new Error('Amount must be positive');
}
await db.insert(creditTransaction).values({ await db.insert(creditTransaction).values({
id: crypto.randomUUID(), id: crypto.randomUUID(),
userId: params.userId, userId: params.userId,
@ -58,6 +64,15 @@ export async function addCredits({
paymentId?: string; paymentId?: string;
expireDays?: number; expireDays?: number;
}) { }) {
if (!userId || !type || !description) {
throw new Error('Invalid params');
}
if (!Number.isFinite(amount) || amount <= 0) {
throw new Error('Amount must be positive');
}
if (!Number.isFinite(expireDays) || expireDays <= 0) {
throw new Error('expireDays must be positive');
}
// Process expired credits first // Process expired credits first
await processExpiredCredits(userId); await processExpiredCredits(userId);
// Update user credit balance // Update user credit balance
@ -111,6 +126,12 @@ export async function consumeCredits({
amount: number; amount: number;
description: string; description: string;
}) { }) {
if (!userId || !description) {
throw new Error('Invalid params');
}
if (!Number.isFinite(amount) || amount <= 0) {
throw new Error('Amount must be positive');
}
// Process expired credits first // Process expired credits first
await processExpiredCredits(userId); await processExpiredCredits(userId);
// Check balance // Check balance