Merge pull request #12 from WorldBrain/feature/pioneer-addon-variable-donation

Pioneer add-on, enables beta feature use

Author: cdharris

firebase/functions/src/external/memex-common

@@ -34,11 +34,20 @@ export const getCheckoutLink = functions.https.onCall(
         const checkoutOptions = {
             subscription: { plan_id: data.planId },
             customer: getUser(context),
-            "redirect_url": undefined,
+            redirect_url: undefined,
+            addons: {},
         }
 
-        if (data["redirect_url"]) {
-            checkoutOptions["redirect_url"] = data["redirect_url"]
+        if (data.redirect_url) {
+            checkoutOptions.redirect_url = data.redirect_url
+        }
+
+        if (data?.pioneerDonationAmount) {
+            checkoutOptions.addons = [{
+                "id": "pioneer",
+                unit_price: Math.max(data.pioneerDonationAmount * 100,100),
+                quantity: 1,
+            }]
         }
 
         const result = await chargebee.hosted_page

firebase/functions/src/subscriptions/index.ts

@@ -75,11 +75,21 @@ export const refreshUserSubscriptionStatus = async (userId: string, { getSubscri
                 // N.B. In case a user has more than one subscription to the same plan,
                 // (e.g. newly configured plan or an old trial) make sure that the furthest expiry date is set.
                 if (existingSubscription == null || existingSubscription.expiry < expiry) {
+
+                    const donationAddOnObject = entry.subscription?.addons?.find(
+                        (addOn: any) => addOn.id === 'pioneer'
+                    )
+                    const donation = (donationAddOnObject) ? {
+                        donation: donationAddOnObject.unit_price * donationAddOnObject.amount
+                    } : {}
+
                     // Set subscription specific expiry and status
                     claims.subscriptions[subPlanId] = {
                         expiry,
                         status: entry.subscription.status,
+                        ...donation
                     }
+
                     // Update overall subscription status
                     claims.subscriptionStatus = entry.subscription.status
                     claims.subscriptionExpiry = expiry
@@ -91,6 +101,7 @@ export const refreshUserSubscriptionStatus = async (userId: string, { getSubscri
 
 
     setFeaturesFromSubscriptions(claims);
+    setFeaturesIfDonated(claims);
 
     // N.B. Claims are always reset, not additive
     // console.log(`setCustomUserClaims(${userId},${JSON.stringify(claims)})`)
@@ -112,6 +123,15 @@ const setFeaturesFromSubscriptions = (claims: Claims) => {
     }
 }
 
+const setFeaturesIfDonated = (claims: Claims) => {
+    for (const subscriptionKey of Object.keys(claims.subscriptions)) {
+        const subscription = claims.subscriptions[subscriptionKey as UserPlan]
+        if (subscription && (subscription?.donation ?? 0) > 0){
+            claims.features.beta = { expiry: subscription.expiry }
+        }
+    }
+}
+
 export const subscriptionToFeatures = new Map<UserPlan, UserFeature[]>([
     ["pro-yearly", ['backup', 'sync']],
     ["pro-monthly", ['backup', 'sync']],