🛂 Add new yearly plans and graduated pricing

BREAKING CHANGE: Stripe environment variables have changed. New ones are required. Check out the new Stripe configuration in the
docs.

Closes #457

apps/builder/src/features/billing/api/createCheckoutSession.ts

@@ -33,6 +33,7 @@ export const createCheckoutSession = authenticatedProcedure
           value: z.string(),
         })
         .optional(),
+      isYearly: z.boolean(),
     })
   )
   .output(
@@ -52,14 +53,11 @@ export const createCheckoutSession = authenticatedProcedure
         returnUrl,
         additionalChats,
         additionalStorage,
+        isYearly,
       },
       ctx: { user },
     }) => {
-      if (
-        !process.env.STRIPE_SECRET_KEY ||
-        !process.env.STRIPE_ADDITIONAL_CHATS_PRICE_ID ||
-        !process.env.STRIPE_ADDITIONAL_STORAGE_PRICE_ID
-      )
+      if (!process.env.STRIPE_SECRET_KEY)
         throw new TRPCError({
           code: 'INTERNAL_SERVER_ERROR',
           message: 'Stripe environment variables are missing',
@@ -120,7 +118,8 @@ export const createCheckoutSession = authenticatedProcedure
         line_items: parseSubscriptionItems(
           plan,
           additionalChats,
-          additionalStorage
+          additionalStorage,
+          isYearly
         ),
       })
 

apps/builder/src/features/billing/api/getSubscription.ts

@@ -5,6 +5,7 @@ import { WorkspaceRole } from '@typebot.io/prisma'
 import Stripe from 'stripe'
 import { z } from 'zod'
 import { subscriptionSchema } from '@typebot.io/schemas/features/billing/subscription'
+import { priceIds } from '@typebot.io/lib/pricing'
 
 export const getSubscription = authenticatedProcedure
   .meta({
@@ -23,15 +24,11 @@ export const getSubscription = authenticatedProcedure
   )
   .output(
     z.object({
-      subscription: subscriptionSchema,
+      subscription: subscriptionSchema.or(z.null()),
     })
   )
   .query(async ({ input: { workspaceId }, ctx: { user } }) => {
-    if (
-      !process.env.STRIPE_SECRET_KEY ||
-      !process.env.STRIPE_ADDITIONAL_CHATS_PRICE_ID ||
-      !process.env.STRIPE_ADDITIONAL_STORAGE_PRICE_ID
-    )
+    if (!process.env.STRIPE_SECRET_KEY)
       throw new TRPCError({
         code: 'INTERNAL_SERVER_ERROR',
         message: 'Stripe environment variables are missing',
@@ -43,10 +40,9 @@ export const getSubscription = authenticatedProcedure
       },
     })
     if (!workspace?.stripeId)
-      throw new TRPCError({
-        code: 'NOT_FOUND',
-        message: 'Workspace not found',
-      })
+      return {
+        subscription: null,
+      }
     const stripe = new Stripe(process.env.STRIPE_SECRET_KEY, {
       apiVersion: '2022-11-15',
     })
@@ -58,24 +54,34 @@ export const getSubscription = authenticatedProcedure
     const subscription = subscriptions?.data.shift()
 
     if (!subscription)
-      throw new TRPCError({
-        code: 'NOT_FOUND',
-        message: 'Subscription not found',
-      })
+      return {
+        subscription: null,
+      }
 
     return {
       subscription: {
-        additionalChatsIndex:
-          subscription?.items.data.find(
-            (item) =>
-              item.price.id === process.env.STRIPE_ADDITIONAL_CHATS_PRICE_ID
-          )?.quantity ?? 0,
-        additionalStorageIndex:
-          subscription.items.data.find(
-            (item) =>
-              item.price.id === process.env.STRIPE_ADDITIONAL_STORAGE_PRICE_ID
-          )?.quantity ?? 0,
+        isYearly: subscription.items.data.some((item) => {
+          return (
+            priceIds.STARTER.chats.yearly === item.price.id ||
+            priceIds.STARTER.storage.yearly === item.price.id ||
+            priceIds.PRO.chats.yearly === item.price.id ||
+            priceIds.PRO.storage.yearly === item.price.id
+          )
+        }),
         currency: subscription.currency as 'usd' | 'eur',
+        cancelDate: subscription.cancel_at
+          ? new Date(subscription.cancel_at * 1000)
+          : undefined,
       },
     }
   })
+
+export const chatPriceIds = [priceIds.STARTER.chats.monthly]
+  .concat(priceIds.STARTER.chats.yearly)
+  .concat(priceIds.PRO.chats.monthly)
+  .concat(priceIds.PRO.chats.yearly)
+
+export const storagePriceIds = [priceIds.STARTER.storage.monthly]
+  .concat(priceIds.STARTER.storage.yearly)
+  .concat(priceIds.PRO.storage.monthly)
+  .concat(priceIds.PRO.storage.yearly)

apps/builder/src/features/billing/api/router.ts

@@ -1,5 +1,4 @@
 import { router } from '@/helpers/server/trpc'
-import { cancelSubscription } from './cancelSubscription'
 import { createCheckoutSession } from './createCheckoutSession'
 import { getBillingPortalUrl } from './getBillingPortalUrl'
 import { getSubscription } from './getSubscription'
@@ -10,7 +9,6 @@ import { updateSubscription } from './updateSubscription'
 export const billingRouter = router({
   getBillingPortalUrl,
   listInvoices,
-  cancelSubscription,
   createCheckoutSession,
   updateSubscription,
   getSubscription,

apps/builder/src/features/billing/api/updateSubscription.ts

@@ -7,6 +7,12 @@ import { workspaceSchema } from '@typebot.io/schemas'
 import Stripe from 'stripe'
 import { isDefined } from '@typebot.io/lib'
 import { z } from 'zod'
+import {
+  getChatsLimit,
+  getStorageLimit,
+  priceIds,
+} from '@typebot.io/lib/pricing'
+import { chatPriceIds, storagePriceIds } from './getSubscription'
 
 export const updateSubscription = authenticatedProcedure
   .meta({
@@ -25,6 +31,7 @@ export const updateSubscription = authenticatedProcedure
       additionalChats: z.number(),
       additionalStorage: z.number(),
       currency: z.enum(['usd', 'eur']),
+      isYearly: z.boolean(),
     })
   )
   .output(
@@ -40,14 +47,11 @@ export const updateSubscription = authenticatedProcedure
         additionalChats,
         additionalStorage,
         currency,
+        isYearly,
       },
       ctx: { user },
     }) => {
-      if (
-        !process.env.STRIPE_SECRET_KEY ||
-        !process.env.STRIPE_ADDITIONAL_CHATS_PRICE_ID ||
-        !process.env.STRIPE_ADDITIONAL_STORAGE_PRICE_ID
-      )
+      if (!process.env.STRIPE_SECRET_KEY)
         throw new TRPCError({
           code: 'INTERNAL_SERVER_ERROR',
           message: 'Stripe environment variables are missing',
@@ -70,42 +74,48 @@ export const updateSubscription = authenticatedProcedure
         customer: workspace.stripeId,
       })
       const subscription = data[0] as Stripe.Subscription | undefined
-      const currentStarterPlanItemId = subscription?.items.data.find(
-        (item) => item.price.id === process.env.STRIPE_STARTER_PRICE_ID
-      )?.id
-      const currentProPlanItemId = subscription?.items.data.find(
-        (item) => item.price.id === process.env.STRIPE_PRO_PRICE_ID
+      const currentPlanItemId = subscription?.items.data.find((item) =>
+        [
+          process.env.STRIPE_STARTER_PRODUCT_ID,
+          process.env.STRIPE_PRO_PRODUCT_ID,
+        ].includes(item.price.product.toString())
       )?.id
       const currentAdditionalChatsItemId = subscription?.items.data.find(
-        (item) => item.price.id === process.env.STRIPE_ADDITIONAL_CHATS_PRICE_ID
+        (item) => chatPriceIds.includes(item.price.id)
       )?.id
       const currentAdditionalStorageItemId = subscription?.items.data.find(
-        (item) =>
-          item.price.id === process.env.STRIPE_ADDITIONAL_STORAGE_PRICE_ID
+        (item) => storagePriceIds.includes(item.price.id)
       )?.id
+      const frequency = isYearly ? 'yearly' : 'monthly'
+
       const items = [
         {
-          id: currentStarterPlanItemId ?? currentProPlanItemId,
-          price:
-            plan === Plan.STARTER
-              ? process.env.STRIPE_STARTER_PRICE_ID
-              : process.env.STRIPE_PRO_PRICE_ID,
+          id: currentPlanItemId,
+          price: priceIds[plan].base[frequency],
           quantity: 1,
         },
         additionalChats === 0 && !currentAdditionalChatsItemId
           ? undefined
           : {
               id: currentAdditionalChatsItemId,
-              price: process.env.STRIPE_ADDITIONAL_CHATS_PRICE_ID,
-              quantity: additionalChats,
+              price: priceIds[plan].chats[frequency],
+              quantity: getChatsLimit({
+                plan,
+                additionalChatsIndex: additionalChats,
+                customChatsLimit: null,
+              }),
               deleted: subscription ? additionalChats === 0 : undefined,
             },
         additionalStorage === 0 && !currentAdditionalStorageItemId
           ? undefined
           : {
               id: currentAdditionalStorageItemId,
-              price: process.env.STRIPE_ADDITIONAL_STORAGE_PRICE_ID,
-              quantity: additionalStorage,
+              price: priceIds[plan].storage[frequency],
+              quantity: getStorageLimit({
+                plan,
+                additionalStorageIndex: additionalStorage,
+                customStorageLimit: null,
+              }),
               deleted: subscription ? additionalStorage === 0 : undefined,
             },
       ].filter(isDefined)
@@ -126,6 +136,9 @@ export const updateSubscription = authenticatedProcedure
           items,
           currency,
           default_payment_method: paymentMethods[0].id,
+          automatic_tax: {
+            enabled: true,
+          },
         })
       }