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

refactor: rename functions to remove 'IfNeed' suffix for clarity and consistency

Browse Source
This commit is contained in:
javayhu 2025-07-13 00:37:01 +08:00
parent 7af313868c
commit 52aeb2d61c
3 changed files with 11 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
src/credits/credits.ts
View File

@ -397,12 +397,12 @@ export async function addRegisterGiftCredits(userId: string) {
* Add free monthly credits
* @param userId - User ID
*/
export async function addMonthlyFreeCreditsIfNeed(userId: string) {
export async function addMonthlyFreeCredits(userId: string) {
const freePlan = Object.values(websiteConfig.price.plans).find(
(plan) => plan.isFree
);
if (!freePlan) {
console.log('addMonthlyFreeCreditsIfNeed, no free plan found');
console.log('addMonthlyFreeCredits, no free plan found');
return;
}
if (
@ -411,7 +411,7 @@ export async function addMonthlyFreeCreditsIfNeed(userId: string) {
!freePlan.credits?.amount
) {
console.log(
'addMonthlyFreeCreditsIfNeed, plan disabled or credits disabled',
'addMonthlyFreeCredits, plan disabled or credits disabled',
freePlan.id
);
return;
@ -448,7 +448,7 @@ export async function addMonthlyFreeCreditsIfNeed(userId: string) {
});
console.log(
`addMonthlyFreeCreditsIfNeed, ${credits} credits for user ${userId}, date: ${now.getFullYear()}-${now.getMonth() + 1}`
`addMonthlyFreeCredits, ${credits} credits for user ${userId}, date: ${now.getFullYear()}-${now.getMonth() + 1}`
);
}
}
@ -496,7 +496,7 @@ export async function addSubscriptionRenewalCredits(
* Add lifetime monthly credits
* @param userId - User ID
*/
export async function addLifetimeMonthlyCreditsIfNeed(userId: string) {
export async function addLifetimeMonthlyCredits(userId: string) {
const lifetimePlan = Object.values(websiteConfig.price.plans).find(
(plan) => plan.isLifetime
);
@ -601,12 +601,12 @@ export async function distributeCreditsToAllUsers() {
if (pricePlan?.isLifetime) {
// Lifetime user - add monthly credits
await addLifetimeMonthlyCreditsIfNeed(userRecord.userId);
await addLifetimeMonthlyCredits(userRecord.userId);
}
// Note: Subscription renewals are handled by Stripe webhooks, not here
} else {
// User has no active subscription - add free monthly credits if enabled
await addMonthlyFreeCreditsIfNeed(userRecord.userId);
await addMonthlyFreeCredits(userRecord.userId);
}
processedCount++;

4
src/lib/auth.ts
View File

@ -1,6 +1,6 @@
import { websiteConfig } from '@/config/website';
import {
addMonthlyFreeCreditsIfNeed,
addMonthlyFreeCredits,
addRegisterGiftCredits,
} from '@/credits/credits';
import { getDb } from '@/db/index';
@ -201,7 +201,7 @@ async function onCreateUser(user: User) {
freePlan?.credits?.amount > 0
) {
try {
await addMonthlyFreeCreditsIfNeed(user.id);
await addMonthlyFreeCredits(user.id);
const credits = freePlan.credits.amount;
console.log(
`added free monthly credits for user ${user.id}, credits: ${credits}`

4
src/payment/provider/stripe.ts
View File

@ -2,7 +2,7 @@ import { randomUUID } from 'crypto';
import { websiteConfig } from '@/config/website';
import {
addCredits,
addLifetimeMonthlyCreditsIfNeed,
addLifetimeMonthlyCredits,
addSubscriptionRenewalCredits,
} from '@/credits/credits';
import { getCreditPackageById } from '@/credits/server';
@ -796,7 +796,7 @@ export class StripeProvider implements PaymentProvider {
websiteConfig.price?.plans || {}
).find((plan) => plan.isLifetime && plan.credits?.enable);
if (lifetimePlan?.prices?.some((p) => p.priceId === priceId)) {
await addLifetimeMonthlyCreditsIfNeed(userId);
await addLifetimeMonthlyCredits(userId);
}
}