Added subscription update middleware

packages/members-api/index.js

@@ -114,7 +114,8 @@ module.exports = function MembersApi({
     const middleware = {
         sendMagicLink: Router(),
         createCheckoutSession: Router(),
-        handleStripeWebhook: Router()
+        handleStripeWebhook: Router(),
+        updateSubscription: Router({mergeParams: true})
     };
 
     middleware.sendMagicLink.use(body.json(), async function (req, res) {
@@ -231,6 +232,57 @@ module.exports = function MembersApi({
         }
     });
 
+    middleware.updateSubscription.use(ensureStripe, body.json(), async function (req, res) {
+        const identity = req.body.identity;
+        const cancelAtPeriodEnd = req.body.cancel_at_period_end;
+        const subscriptionId = req.params.id;
+
+        let member;
+
+        try {
+            if (!identity) {
+                throw new common.errors.BadRequestError({
+                    message: 'Cancel membership failed! Could not find member'
+                });
+            }
+
+            const claims = await decodeToken(identity);
+            const email = claims.sub;
+            member = email ? await users.get({email}) : null;
+
+            if (!member) {
+                throw new common.errors.BadRequestError({
+                    message: 'Cancel membership failed! Could not find member'
+                });
+            }
+        } catch (err) {
+            res.writeHead(401);
+            return res.end('Unauthorized');
+        }
+
+        // Don't allow removing subscriptions that don't belong to the member
+        const subscription = member.stripe.subscriptions.find(sub => sub.id === subscriptionId);
+
+        if (!subscription) {
+            res.writeHead(403);
+            return res.end('No permission');
+        }
+
+        if (cancelAtPeriodEnd === undefined) {
+            throw new common.errors.BadRequestError({
+                message: 'Canceling membership failed!',
+                help: 'Request should contain boolean "cancel" field.'
+            });
+        }
+
+        subscription.cancel_at_period_end = !!(cancelAtPeriodEnd);
+
+        await stripe.updateSubscriptionFromClient(subscription);
+
+        res.writeHead(204);
+        res.end();
+    });
+
     const getPublicConfig = function () {
         return Promise.resolve({
             publicKey,

packages/members-api/lib/common.js

@@ -14,5 +14,9 @@ module.exports = {
                 Object.assign(loggerInterface, Object.create(newLogger));
             }
         });
+    },
+
+    get errors() {
+        return require('ghost-ignition').errors;
     }
 };

packages/members-api/lib/stripe/index.js

@@ -1,5 +1,5 @@
 const debug = require('ghost-ignition').debug('stripe');
-const {retrieve, list, create, del} = require('./api/stripeRequests');
+const {retrieve, list, create, update, del} = require('./api/stripeRequests');
 const api = require('./api');
 
 const STRIPE_API_VERSION = '2019-09-09';
@@ -135,6 +135,15 @@ module.exports = class StripePaymentProcessor {
         return true;
     }
 
+    async updateSubscriptionFromClient(subscription) {
+        const updatedSubscription = await update(this._stripe, 'subscriptions', subscription.id, {
+            cancel_at_period_end: subscription.cancel_at_period_end
+        });
+        await this._updateSubscription(updatedSubscription);
+
+        return updatedSubscription;
+    }
+
     async getSubscriptions(member) {
         const metadata = await this.storage.get(member);
 
@@ -162,6 +171,7 @@ module.exports = class StripePaymentProcessor {
                 status: subscription.status,
                 start_date: subscription.start_date,
                 default_payment_card_last4: subscription.default_payment_card_last4,
+                cancel_at_period_end: subscription.cancel_at_period_end,
                 current_period_end: subscription.current_period_end
             };
         });
@@ -235,6 +245,7 @@ module.exports = class StripePaymentProcessor {
 
                 subscription_id: subscription.id,
                 status: subscription.status,
+                cancel_at_period_end: subscription.cancel_at_period_end,
                 current_period_end: new Date(subscription.current_period_end * 1000),
                 start_date: new Date(subscription.start_date * 1000),
                 default_payment_card_last4: payment && payment.card && payment.card.last4 || null,