Update subscriptions.py

subscriptions.py

@@ -5,6 +5,21 @@ SUPABASE_URL = os.getenv("SUPABASE_URL")
 SUPABASE_KEY = os.getenv("SUPABASE_SERVICE_ROLE_KEY")
 supabase: Client = create_client(SUPABASE_URL, SUPABASE_KEY)
 
+
+def normalize_plan_key(plan_name: str | None) -> str:
+    if not plan_name:
+        return "free"
+    normalized = "".join(ch for ch in str(plan_name).lower() if ch.isalpha())
+    if "professional" in normalized:
+        return "professional"
+    if "creator" in normalized:
+        return "creator"
+    if "pro" in normalized:
+        return "pro"
+    if "light" in normalized:
+        return "light"
+    return "free"
+
 async def fetch_subscription(jwt: str):
     auth_res = supabase.auth.get_user(jwt)
     if auth_res.user is None:
@@ -24,7 +39,8 @@ async def fetch_subscription(jwt: str):
         return {
             "email": email,
             "signed_up": user.created_at.isoformat() if hasattr(user.created_at, "isoformat") else user.created_at,
-            "subscription": None
+            "subscription": None,
+            "plan_key": "free",
         }
 
     customer = cust_res.data[0]
@@ -42,10 +58,12 @@ async def fetch_subscription(jwt: str):
         return {
             "email": email,
             "signed_up": user.created_at.isoformat() if hasattr(user.created_at, "isoformat") else user.created_at,
-            "subscription": None
+            "subscription": None,
+            "plan_key": "free",
         }
 
     subscriptions = []
+    preferred_plan_key = "free"
 
     for s in sub_res.data:
         price_id = (
@@ -80,16 +98,21 @@ async def fetch_subscription(jwt: str):
                     if product_res.data:
                         product_name = product_res.data[0].get("name")
 
+        plan_key = normalize_plan_key(product_name or nickname)
         subscriptions.append({
             "status": s["status"],
             "subscription_id": s["id"],
             "current_period_end": s["current_period_end"],
             "product_name": product_name,
-            "nickname": nickname
+            "nickname": nickname,
+            "plan_key": plan_key,
         })
+        if s.get("status") in ("active", "trialing"):
+            preferred_plan_key = plan_key
 
     return {
         "email": email,
         "signed_up": user.created_at.isoformat() if hasattr(user.created_at, "isoformat") else user.created_at,
-        "subscription": subscriptions
+        "subscription": subscriptions,
+        "plan_key": preferred_plan_key,
     }