Set access to null upon expiration

elie222 · Jan 4, 2025 · 009a0d3 · 009a0d3
1 parent ca09c9d
commit 009a0d3
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 4 deletions.
diff --git a/apps/web/app/api/lemon-squeezy/webhook/route.ts b/apps/web/app/api/lemon-squeezy/webhook/route.ts
@@ -94,7 +94,7 @@ export const POST = withError(async (request: Request) => {
     return await subscriptionPlanChanged({ payload, userId });
   }
 
-  // cancellation
+  // cancelled or expired
   if (payload.data.attributes.ends_at) {
     return await subscriptionCancelled({
       payload,
@@ -414,6 +414,7 @@ async function subscriptionCancelled({
     premiumId,
     variantId,
     lemonSqueezyEndsAt: new Date(endsAt),
+    expired: payload.data.attributes.status === "expired",
   });
 
   if (!updatedPremium) return NextResponse.json({ ok: true });

diff --git a/apps/web/utils/actions/premium.ts b/apps/web/utils/actions/premium.ts
@@ -334,6 +334,7 @@ export const changePremiumStatusAction = withActionInstrumentation(
         await cancelPremium({
           premiumId: userToUpgrade.premiumId,
           lemonSqueezyEndsAt: new Date(),
+          expired: true,
         });
       } else {
         return { error: "User not premium." };

diff --git a/apps/web/utils/premium/server.ts b/apps/web/utils/premium/server.ts
@@ -75,10 +75,12 @@ export async function cancelPremium({
   premiumId,
   lemonSqueezyEndsAt,
   variantId,
+  expired,
 }: {
   premiumId: string;
   lemonSqueezyEndsAt: Date;
   variantId?: number;
+  expired: boolean;
 }) {
   if (variantId) {
     // Check if the premium exists for the given variant
@@ -95,9 +97,13 @@ export async function cancelPremium({
     where: { id: premiumId },
     data: {
       lemonSqueezyRenewsAt: lemonSqueezyEndsAt,
-      bulkUnsubscribeAccess: null,
-      aiAutomationAccess: null,
-      coldEmailBlockerAccess: null,
+      ...(expired
+        ? {
+            bulkUnsubscribeAccess: null,
+            aiAutomationAccess: null,
+            coldEmailBlockerAccess: null,
+          }
+        : {}),
     },
     select: {
       users: {