add basic Stripe integration

Author: vkarpov15

netlify/functions/github.js

@@ -5,6 +5,7 @@ const connect = require('../../src/db');
 const extrovert = require('extrovert');
 const githubOAuth = require('../../src/integrations/githubOAuth');
 const mongoose = require('mongoose');
+const stripe = require('../../src/integrations/stripe');
 
 const GithubOAuthParams = new Archetype({
   code: {
@@ -66,11 +67,16 @@ module.exports = extrovert.toNetlifyFunction(async function github(params) {
       await invitation.save();
 
       roles = invitation.roles;
+
+      if (workspace.stripeSubscriptionId && !user.isFreeUser) {
+        const users = await User.find({ _id: { $in: workspace.members.map(member => member.userId) }, isFreeUser: { $ne: true } });
+        const seats = users.length;
+        await stripe.updateSubscriptionSeats(workspace.stripeSubscriptionId, seats);
+      }
     }
   } else {
     roles = member.roles;
   }
 
-
   return { user, accessToken, roles };
 });

netlify/functions/removeFromWorkspace.js

@@ -0,0 +1,5 @@
+'use strict';
+
+const extrovert = require('extrovert');
+
+module.exports = extrovert.toNetlifyFunction(require('../../src/actions/removeFromWorkspace'));

package.json

@@ -12,6 +12,7 @@
     "extrovert": "0.0.26",
     "mongoose": "8.x",
     "ramda": "0.28.0",
+    "stripe": "9.9.0",
     "time-commando": "1.0.1"
   },
   "devDependencies": {

src/actions/removeFromWorkspace.js

@@ -0,0 +1,50 @@
+'use strict';
+
+const Archetype = require('archetype');
+const connect = require('../../src/db');
+const mongoose = require('mongoose');
+
+const RemoveFromWorkspaceParams = new Archetype({
+  authorization: {
+    $type: 'string',
+    $required: true
+  },
+  workspaceId: {
+    $type: mongoose.Types.ObjectId,
+    $required: true
+  },
+  userId: {
+    $type: mongoose.Types.ObjectId,
+    $required: true
+  }
+}).compile('InviteToWorkspaceParams');
+
+module.exports = async function removeFromWorkspace(params) {
+  const db = await connect();
+  const { AccessToken, Workspace } = db.models;
+
+  const { authorization, workspaceId, userId } = new RemoveFromWorkspaceParams(params);
+
+  // Verify access token
+  const accessToken = await AccessToken.findById(authorization).orFail(new Error('Invalid or expired access token'));
+  if (accessToken.expiresAt < new Date()) {
+    throw new Error('Access token has expired');
+  }
+  const initiatedByUserId = accessToken._id;
+
+  const workspace = await Workspace.findById(workspaceId).orFail(new Error('Workspace not found'));
+  const initiatedByUserRoles = workspace.members.find(member => member.userId.toString() === initiatedByUserId.toString())?.roles;
+  if (initiatedByUserRoles == null || (!initiatedByUserRoles.includes('admin') && !initiatedByUserRoles.includes('owner'))) {
+    throw new Error('Forbidden');
+  }
+
+  const memberIndex = workspace.members.findIndex(member => member.userId.toString() === userId.toString());
+  if (memberIndex === -1) {
+    throw new Error('Member not found in the workspace');
+  }
+
+  workspace.members.splice(memberIndex, 1);
+  await workspace.save();
+
+  return { workspace };
+};

src/db/workspace.js

@@ -35,18 +35,14 @@ const workspaceSchema = new mongoose.Schema({
     required: true
   },
   stripeCustomerId: {
-    type: String,
-    unique: true,
-    sparse: true
+    type: String
   },
   stripeSubscriptionId: {
-    type: String,
-    unique: true,
-    sparse: true
+    type: String
   },
   subscriptionTier: {
     type: String,
-    enum: ['pro']
+    enum: ['free', 'pro']
   }
 }, { timestamps: true, id: false });
 

src/integrations/stripe.js

@@ -0,0 +1,29 @@
+'use strict';
+
+const assert = require('assert');
+const stripe = require('stripe')(process.env.STRIPE_SECRET_KEY);
+
+exports.listSubscriptions = async function listSubscriptions() {
+  const subscriptions = await stripe.subscriptions.list();
+  return subscriptions;
+};
+
+
+exports.getSubscription = async function getSubscription(subscriptionId) {
+  const subscription = await stripe.subscriptions.retrieve(subscriptionId);
+  return subscription;
+};
+
+exports.updateSubscriptionSeats = async function updateSubscriptionSeats(subscriptionId, newSeatCount) {
+  assert(Number.isInteger(newSeatCount) && newSeatCount >= 0, 'Seat count must be a non-negative integer.');
+
+  const subscription = await stripe.subscriptions.retrieve(subscriptionId);
+  const subscriptionItemId = subscription.items.data[0].id;
+
+  const updatedSubscription = await stripe.subscriptionItems.update(
+    subscriptionItemId,
+    { quantity: newSeatCount }
+  );
+
+  return updatedSubscription;
+};