From b5fed7b01dc48acc02ea5b9339e1e2d78d288207 Mon Sep 17 00:00:00 2001
From: liuyi <forehalo@gmail.com>
Date: Fri, 22 Nov 2024 11:57:38 +0800
Subject: [PATCH] fix(server): ignore invalid subscription variant (#8892)

---
 .../backend/server/src/plugins/payment/resolver.ts | 14 +++++++++++++-
 1 file changed, 13 insertions(+), 1 deletion(-)

diff --git a/packages/backend/server/src/plugins/payment/resolver.ts b/packages/backend/server/src/plugins/payment/resolver.ts
index db5d983a8f..f8771677e3 100644
--- a/packages/backend/server/src/plugins/payment/resolver.ts
+++ b/packages/backend/server/src/plugins/payment/resolver.ts
@@ -370,7 +370,7 @@ export class UserSubscriptionResolver {
       };
     }
 
-    return this.db.userSubscription.findUnique({
+    const subscription = await this.db.userSubscription.findUnique({
       where: {
         userId_plan: {
           userId: user.id,
@@ -379,6 +379,18 @@ export class UserSubscriptionResolver {
         status: SubscriptionStatus.Active,
       },
     });
+
+    if (
+      subscription &&
+      subscription.variant &&
+      ![SubscriptionVariant.EA, SubscriptionVariant.Onetime].includes(
+        subscription.variant as SubscriptionVariant
+      )
+    ) {
+      subscription.variant = null;
+    }
+
+    return subscription;
   }
 
   @ResolveField(() => [UserSubscriptionType])