[refactor] [fix] simplify the processPurchaseInfo function + Bug fixes

Author: swashata

src/app/credits/credits.tsx

@@ -1,10 +1,10 @@
 'use client';
 
-import { Subscribe } from '@/react-starter/components/subscribe';
+import { Topup } from '@/react-starter/components/topup';
 import { formatNumber } from '@/react-starter/utils/formatter';
 
-export default function Credits(props: { credits?: number; hasSubscription: boolean }) {
-    const { credits, hasSubscription } = props;
+export default function Credits(props: { credits?: number }) {
+    const { credits } = props;
 
     const info = (
         <div className="text-center">
@@ -13,5 +13,5 @@ export default function Credits(props: { credits?: number; hasSubscription: bool
         </div>
     );
 
-    return <Subscribe>{info}</Subscribe>;
+    return <Topup>{info}</Topup>;
 }

src/app/credits/page.tsx

@@ -1,6 +1,6 @@
 import AppMain, { AppContent } from '@/components/app-main';
 import { auth } from '@/lib/auth';
-import { getCredits, getUserEntitlement } from '@/lib/user-entitlement';
+import { getCredits } from '@/lib/user-entitlement';
 import { headers } from 'next/headers';
 import { redirect } from 'next/navigation';
 import { ErrorBoundary } from '@/components/error';
@@ -17,7 +17,6 @@ export default async function CreditsPage() {
         redirect('/login');
     }
 
-    const entitlement = await getUserEntitlement(session.user.id);
     const credits = await getCredits(session.user.id);
 
     const checkout = await freemius.checkout.create({
@@ -30,7 +29,7 @@ export default async function CreditsPage() {
             <AppContent>
                 <ErrorBoundary>
                     <AppCheckoutProvider checkout={checkout.serialize()}>
-                        <Credits credits={credits} hasSubscription={!!entitlement} />
+                        <Credits credits={credits} />
                     </AppCheckoutProvider>
                 </ErrorBoundary>
             </AppContent>

src/lib/user-entitlement.ts

@@ -17,15 +17,43 @@ import { freemius } from './freemius';
 // #region Freemius SDK Supporting Functions for User Entitlements
 
 export async function processPurchaseInfo(fsPurchase: PurchaseInfo): Promise<void> {
-    // Todo: fill me
+    const user = await getUserByEmail(fsPurchase.email);
+
+    if (!user) {
+        return;
+    }
+
+    const credit = await getCreditsForUserPurchase(user, fsPurchase);
+
+    // Save purchase info in our DB
+    await prisma.userFsEntitlement.upsert({
+        where: { fsLicenseId: fsPurchase.licenseId },
+        create: fsPurchase.toEntitlementRecord({ userId: user.id }),
+        update: fsPurchase.toEntitlementRecord(),
+    });
+
+    if (credit > 0) {
+        await addCredits(user.id, credit);
+    }
 }
 
 export async function getUserEntitlement(userId: string): Promise<UserFsEntitlement | null> {
-    // Todo: fill me
+    const entitlements = await prisma.userFsEntitlement.findMany({
+        where: { userId, type: 'subscription' },
+    });
+
+    return freemius.entitlement.getActive(entitlements);
 }
 
 export const getFsUser: UserRetriever = async () => {
-    // Todo: fill me
+    const session = await auth.api.getSession({
+        headers: await headers(),
+    });
+
+    const entitlement = session ? await getUserEntitlement(session.user.id) : null;
+    const email = session?.user.email ?? undefined;
+
+    return freemius.entitlement.getFsUser(entitlement, email);
 };
 
 function getEntitledCredits(fsPurchase: PurchaseInfo): number {
@@ -46,13 +74,6 @@ async function getCreditsForUserPurchase(user: User, fsPurchase: PurchaseInfo):
         credits = getEntitledCredits(fsPurchase);
     }
 
-    // Save purchase info in our DB
-    await prisma.userFsEntitlement.upsert({
-        where: { fsLicenseId: fsPurchase.licenseId },
-        create: fsPurchase.toEntitlementRecord({ userId: user.id }),
-        update: fsPurchase.toEntitlementRecord(),
-    });
-
     return credits;
 }
 

src/react-starter/components/payments-section.tsx

@@ -52,7 +52,7 @@ export function PaymentsSection(props: { payments: PortalPayment[]; unit: Sellin
 
                         <div className="fs-saas-starter-payments-section__plan col-span-2 lg:col-span-1">
                             {payment.planTitle}{' '}
-                            {payment.quota ? (
+                            {payment.quota && payment.quota !== 1 ? (
                                 <span className="text-disabled-foreground text-xs font-semibold">
                                     {locale.portal.payments.pricingTitle(
                                         formatNumber(payment.quota ?? 0, locale.code),