✨ (billing) Implement custom plan

apps/builder/pages/api/stripe/custom-plan-checkout.ts

@@ -0,0 +1,61 @@
+import { withSentry } from '@sentry/nextjs'
+import { Plan } from 'db'
+import prisma from 'libs/prisma'
+import { NextApiRequest, NextApiResponse } from 'next'
+import { getAuthenticatedUser } from 'services/api/utils'
+import Stripe from 'stripe'
+import { methodNotAllowed, notAuthenticated } from 'utils/api'
+
+const handler = async (req: NextApiRequest, res: NextApiResponse) => {
+  const user = await getAuthenticatedUser(req)
+  if (!user) return notAuthenticated(res)
+  if (req.method === 'GET') {
+    const session = await createCheckoutSession(user.id)
+    if (!session?.url) return res.redirect('/typebots')
+    return res.redirect(session.url)
+  }
+
+  return methodNotAllowed(res)
+}
+
+const createCheckoutSession = async (userId: string) => {
+  if (!process.env.STRIPE_SECRET_KEY)
+    throw Error('STRIPE_SECRET_KEY var is missing')
+  const stripe = new Stripe(process.env.STRIPE_SECRET_KEY, {
+    apiVersion: '2022-08-01',
+  })
+
+  const claimableCustomPlan = await prisma.claimableCustomPlan.findFirst({
+    where: { workspace: { members: { some: { userId } } } },
+  })
+
+  if (!claimableCustomPlan) return null
+
+  return stripe.checkout.sessions.create({
+    success_url: `${process.env.NEXTAUTH_URL}/typebots?stripe=${Plan.CUSTOM}&success=true`,
+    cancel_url: `${process.env.NEXTAUTH_URL}/typebots?stripe=cancel`,
+    mode: 'subscription',
+    metadata: {
+      claimableCustomPlanId: claimableCustomPlan.id,
+    },
+    currency: claimableCustomPlan.currency,
+    automatic_tax: { enabled: true },
+    line_items: [
+      {
+        price_data: {
+          currency: claimableCustomPlan.currency,
+          tax_behavior: 'exclusive',
+          recurring: { interval: 'month' },
+          product_data: {
+            name: claimableCustomPlan.name,
+            description: claimableCustomPlan.description ?? undefined,
+          },
+          unit_amount: claimableCustomPlan.price * 100,
+        },
+        quantity: 1,
+      },
+    ],
+  })
+}
+
+export default withSentry(handler)

apps/builder/pages/api/stripe/webhook.ts

@@ -40,31 +40,58 @@ const webhookHandler = async (req: NextApiRequest, res: NextApiResponse) => {
       switch (event.type) {
         case 'checkout.session.completed': {
           const session = event.data.object as Stripe.Checkout.Session
-          const { workspaceId, plan, additionalChats, additionalStorage } =
-            session.metadata as unknown as {
-              plan: 'STARTER' | 'PRO'
-              additionalChats: string
-              additionalStorage: string
-              workspaceId: string
-            }
+          const metadata = session.metadata as unknown as
+            | {
+                plan: 'STARTER' | 'PRO'
+                additionalChats: string
+                additionalStorage: string
+                workspaceId: string
+              }
+            | { claimableCustomPlanId: string }
+          if ('plan' in metadata) {
+            const { workspaceId, plan, additionalChats, additionalStorage } =
+              metadata
+            if (!workspaceId || !plan || !additionalChats || !additionalStorage)
+              return res
+                .status(500)
+                .send({ message: `Couldn't retrieve valid metadata` })
+            await prisma.workspace.update({
+              where: { id: workspaceId },
+              data: {
+                plan: plan,
+                stripeId: session.customer as string,
+                additionalChatsIndex: parseInt(additionalChats),
+                additionalStorageIndex: parseInt(additionalStorage),
+                chatsLimitFirstEmailSentAt: null,
+                chatsLimitSecondEmailSentAt: null,
+                storageLimitFirstEmailSentAt: null,
+                storageLimitSecondEmailSentAt: null,
+              },
+            })
+          } else {
+            const { claimableCustomPlanId } = metadata
+            if (!claimableCustomPlanId)
+              return res
+                .status(500)
+                .send({ message: `Couldn't retrieve valid metadata` })
+            const { workspaceId, chatsLimit, seatsLimit, storageLimit } =
+              await prisma.claimableCustomPlan.update({
+                where: { id: claimableCustomPlanId },
+                data: { claimedAt: new Date() },
+              })
+
+            await prisma.workspace.update({
+              where: { id: workspaceId },
+              data: {
+                plan: Plan.CUSTOM,
+                stripeId: session.customer as string,
+                customChatsLimit: chatsLimit,
+                customStorageLimit: storageLimit,
+                customSeatsLimit: seatsLimit,
+              },
+            })
+          }
 
-          if (!workspaceId || !plan || !additionalChats || !additionalStorage)
-            return res
-              .status(500)
-              .send({ message: `Couldn't retrieve valid metadata` })
-          await prisma.workspace.update({
-            where: { id: workspaceId },
-            data: {
-              plan: plan,
-              stripeId: session.customer as string,
-              additionalChatsIndex: parseInt(additionalChats),
-              additionalStorageIndex: parseInt(additionalStorage),
-              chatsLimitFirstEmailSentAt: null,
-              chatsLimitSecondEmailSentAt: null,
-              storageLimitFirstEmailSentAt: null,
-              storageLimitSecondEmailSentAt: null,
-            },
-          })
           return res.status(200).send({ message: 'workspace upgraded in DB' })
         }
         case 'customer.subscription.deleted': {